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 get_converted_speed(unit:str, speed:float)-> float: #it's import to specify the return type
    #Keep this format for the description / args / args description but feel free to modify the tool
    """A tool that converts the speed from kilometers per hour to miles per hour. Or from miles per hour to kilometers per hour.
    Args:
        unit: A string representing the unit speed is supplied in. Can be either "kph" or "mph".
        speed: A float representing the speed.
    """
    factor = 0.621371;

    if unit == "kph":
        converted_speed = speed * factor
    elif unit == "mph": 
        converted_speed = speed / factor
    else:
        raise ValueError(f"Invalid unit: {unit}. Please use 'kph' or 'mph'.")

    return converted_speed

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, get_converted_speed], ## 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()