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

# Tool that simulates a "Happiness Score" based on various life factors
@tool
def calculate_happiness(sleep_hours: float, exercise_minutes: int, social_interactions: int, stress_level: int) -> str:
    """Calculates a fun 'Happiness Score' based on lifestyle factors.
    
    Args:
        sleep_hours: Hours of sleep (0-12)
        exercise_minutes: Minutes of exercise per day (0-180)
        social_interactions: Number of meaningful conversations per day (0-20)
        stress_level: Stress level from 1 (low) to 10 (high)

    Returns:
        A string with the calculated happiness score and a fun message.
    """
    # Normalize the inputs into a score (out of 100)
    sleep_score = min(sleep_hours / 8 * 30, 30)  # Max 30 points
    exercise_score = min(exercise_minutes / 60 * 20, 20)  # Max 20 points
    social_score = min(social_interactions / 5 * 20, 20)  # Max 20 points
    stress_penalty = (10 - stress_level) * 3  # Lower stress gives higher score (Max 30 points)

    happiness_score = round(sleep_score + exercise_score + social_score + stress_penalty)

    # Fun messages based on happiness level
    if happiness_score > 80:
        message = "😃 You're living your best life! Keep it up!"
    elif happiness_score > 60:
        message = "😊 You're doing well! A few tweaks can boost your happiness even more."
    elif happiness_score > 40:
        message = "😐 Things are okay, but there's room for improvement."
    else:
        message = "😟 You might need to make some changes. Try adding more sleep or reducing stress."

    return f"Your Happiness Score: {happiness_score}/100. {message}"

@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/',# 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, get_current_time_in_timezone, calculate_happiness], ## add your tools here (don't remove final answer)
    max_steps=6,
    verbosity_level=1,
    grammar=None,
    planning_interval=None,
    name="Happiness Score Calculator",
    description="Creative calculation function that simulates a \"Happiness Score\" based on various life factors.",
    prompt_templates=prompt_templates
)


GradioUI(agent).launch()