from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool
import datetime
import requests
import pytz
import yaml
from tools.final_answer import FinalAnswerTool

from Gradio_UI import GradioUI

# Below is an example of a tool that does nothing. Amaze us with your creativity !

@tool
def generate_motivational_quote(mood: str, goal: str) -> str:
    """A tool that generates a custom motivational quote based on the user's mood and goal.
    Args:
        mood: The user's current mood (e.g., 'happy', 'stressed', 'excited').
        goal: The user's current goal or focus (e.g., 'finish a project', 'learn something new').
    """
    import random

    # Define mood-based phrases
    mood_phrases = {
        "happy": ["Keep shining!", "Your positivity is contagious!", "The world is brighter with you in it!"],
        "stressed": ["Take a deep breath.", "You're stronger than you think.", "This too shall pass."],
        "excited": ["The sky's the limit!", "Your enthusiasm is inspiring!", "Go chase your dreams!"],
        "tired": ["Rest, but don't quit.", "You're making progress, even if it feels slow.", "Tomorrow is a new day."],
        "focused": ["You're in the zone!", "Stay sharp, you're unstoppable!", "Great things are coming your way!"],
    }

    # Define goal-based phrases
    goal_phrases = {
        "finish a project": ["You're almost there!", "The finish line is in sight!", "One step at a time."],
        "learn something new": ["Every expert was once a beginner.", "Knowledge is power!", "Keep curious!"],
        "start a new habit": ["Small steps lead to big changes.", "Consistency is key!", "You're building a better you!"],
        "exercise": ["Your body will thank you!", "Sweat now, shine later!", "You're getting stronger every day!"],
        "relax": ["You deserve this break.", "Recharge and come back stronger.", "Peace begins with a moment of rest."],
    }

    # Default fallback phrases
    default_phrases = [
        "You're capable of amazing things!",
        "Every day is a new opportunity.",
        "Believe in yourself!",
    ]

    # Get mood and goal phrases
    mood_quote = random.choice(mood_phrases.get(mood.lower(), default_phrases))
    goal_quote = random.choice(goal_phrases.get(goal.lower(), default_phrases))

    # Combine into a motivational quote
    return f"🌟 {mood_quote} 🌟 {goal_quote}"

@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='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], ## 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()