from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, load_tool, tool
import datetime
import requests
import pytz
import yaml
from tools.final_answer import FinalAnswerTool
from datetime import datetime, timedelta
import json
from Gradio_UI import GradioUI
import os
from dotenv import load_dotenv
import openai

# Load environment variables from .env file
load_dotenv()

# Access the API key
openai.api_key = os.getenv('OPENAI_API_KEY')

# Below is the new deadline calculator tool

@tool
def chatgpt_conversation(prompt: str) -> str:
    """A tool that interacts with the OpenAI API to simulate a conversation with a dynamic prompt.
    
    Args:
        prompt: The message to ChatGPT.
    Returns:
        The assistant's response to the prompt.
    """
    response = openai.chat.completions.create(
        model="gpt-4o-mini",
        messages=[
            {"role": "user", "content": prompt}
        ],
    )
    return response.choices[0].message.content

@tool
def marketing_expert_review(post: str) -> str:
    """A tool that reviews and enhances social media posts to maximize impressions, leveraging expertise in marketing, neuromarketing, sales, and social media strategies.
    
    Args:
        post: The social media post to be reviewed and enhanced.
    """
    prompt = f"""
    You are an expert in marketing, neuromarketing, sales, and social media post reviewing with over 15 years of experience optimizing content for maximum reach and engagement. Your expertise includes understanding audience psychology, crafting compelling narratives, and applying proven strategies to boost impressions on platforms like Instagram, Facebook, LinkedIn, and Twitter. Your task is to review and enhance the following social media post to make it as engaging, attractive, and impactful as possible, ensuring it captures attention and drives the highest possible impressions.

    **Context**: Social media platforms thrive on posts that are emotionally resonant, visually suggestive (even in text), and encourage interaction (likes, shares, comments). Successful posts often include a hook in the first few words, a clear value proposition, and a call-to-action.

    **Post to enhance**: "{post}"

    **Format**: Provide the enhanced post directly as a single block of text, ready to be posted. Keep it concise yet powerful, suitable for broad social media use unless a specific platform is mentioned in the instructions.

    **References**: For guidance, consider posts like: "Just dropped our new summer collection – bold colors, big vibes! Which shade’s your vibe? Drop it below! 🌟 #SummerStyle" (hook + engagement + hashtag) or "Struggling with productivity? This 5-min trick changed my life – and it’ll change yours too. Link in bio!" (problem-solution + curiosity).

    Enhanced post:
    """
    try:
        response = openai.chat.completions.create(
            model="gpt-4o-mini",
            messages=[
                {"role": "user", "content": prompt}
            ],
        )
        return response.choices[0].message.content
    except Exception as e:
        return f"Error generating enhanced post: {str(e)}"

@tool
def twitter_post_enhancer(post: str) -> str:
    """A tool that enhances short, basic social media posts into engaging Twitter (X) posts, leveraging expertise in social media engagement and content writing.
    
    Args:
        post: The social media post to be enhanced for Twitter.
    """
    prompt = f"""
    You are an expert in social media engagement and content writing, with a specialization in crafting posts for Twitter (X). You have a deep understanding of Twitter’s unique culture, where concise, punchy, and emotionally engaging content thrives within a 280-character limit. Your task is to take the following short, basic social media post and enhance it to make it more engaging, attractive, and appealing for a general Twitter audience. Focus on:
    - Adding relevant emojis to boost visual appeal and emotion.
    - Rephrasing text for clarity, impact, and a casual, conversational tone.
    - Making the post interesting and shareable while keeping it concise and easy to understand.

    **Context**: Twitter users scroll fast—posts need a strong hook, relatable or surprising elements, and often a question or bold statement to spark interaction. Trending topics, humor, and timely references can amplify reach.

    **Post to enhance**: "{post}"

    **Format**: Return the enhanced post in this structure:
    <title>
    <br>
    <Brief summary.>

    IMPORTANT: NEVER USE HASHTAGS!, use emojis.

    **References**: Draw inspiration from posts like:
    - "AI just solved a 10-yr mystery in 48 hrs! 🧠💥 Scientists are shook. What’s next for tech?"
    <br>
    "Google’s co-scientist cracked a superbug puzzle faster than a decade of research."
    - "New year, new hacks! 🚀✨ What’s your top productivity tip? Spill it below!"
    <br>
    "Quick tricks to kickstart your 2023 goals."

    Enhanced post:
    """
    try:
        response = openai.chat.completions.create(
            model="gpt-4o-mini",
            messages=[
                {"role": "user", "content": prompt}
            ],
        )
        return response.choices[0].message.content
    except Exception as e:
        return f"Error generating enhanced Twitter post: {str(e)}"

@tool
def get_current_time_in_timezone(timezone: str) -> str:
    """A tool that fetches the current local time in a specified timezone.
    Args:
        timezone: A string representing a valid timezone (e.g., 'America/New_York').
    """
    try:
        # Create timezone object
        tz = pytz.timezone(timezone)
        # Get current time in that timezone
        local_time = datetime.datetime.now(tz).strftime("%Y-%m-%d %H:%M:%S")
        return f"The current local time in {timezone} is: {local_time}"
    except Exception as e:
        return f"Error fetching time for timezone '{timezone}': {str(e)}"


final_answer = FinalAnswerTool()

# If the agent does not answer, the model is overloaded, please use another model or the following Hugging Face Endpoint that also contains qwen2.5 coder:
# model_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud' 

model = HfApiModel(
max_tokens=2096,
temperature=0.5,
model_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud', #model_id='Qwen/Qwen2.5-Coder-32B-Instruct',# it is possible that this model may be overloaded
custom_role_conversions=None,
)


# Import tool from Hub
image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)

with open("prompts.yaml", 'r') as stream:
    prompt_templates = yaml.safe_load(stream)
    
agent = CodeAgent(
    model=model,
    tools=[final_answer, twitter_post_enhancer, marketing_expert_review], ## add your tools here (don't remove final answer)
    max_steps=6,
    verbosity_level=1,
    grammar=None,
    planning_interval=None,
    name=None,
    description=None,
    prompt_templates=prompt_templates
)


GradioUI(agent).launch()