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 !

# Create a single instance of the DDG tool (so it can be reused)
ddg = DuckDuckGoSearchTool()

@tool
def web_search(query: str) -> str:
    """Search the public web using DuckDuckGo.
    Use this tool whenever the user asks for information found on the internet
    (news, definitions, comparisons, current info, Hugging Face updates, etc.)
    Args:
        query: The search query.
    """
    return ddg(query)

@tool
def calculator(expression: str) -> str:
    """Evaluate a basic math expression safely.
    Args:
        expression: Math expression like "12*3 + 4".
    """
    try:
        allowed = "0123456789+-*/(). %"
        if any(ch not in allowed for ch in expression):
            return "Error: expression contains invalid characters."
        result = eval(expression, {"__builtins__": {}}, {})
        return str(result)
    except Exception as e:
        return f"Error evaluating expression: {e}"

@tool
def get_exchange_rate(base: str, target: str) -> str:
    """Get latest exchange rate from base currency to target currency.
    Args:
        base: Base currency code like 'USD'.
        target: Target currency code like 'ARS'.
    """
    try:
        url = f"https://open.er-api.com/v6/latest/{base.upper()}"
        data = requests.get(url, timeout=10).json()
        if data.get("result") != "success":
            return f"Error: API response not successful: {data}"
        rate = data["rates"].get(target.upper())
        if rate is None:
            return f"Error: target currency {target} not found."
        return f"1 {base.upper()} = {rate} {target.upper()}"
    except Exception as e:
        return f"Error fetching exchange rate: {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='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=[
        web_search,
        get_current_time_in_timezone,
        image_generation_tool,
        get_exchange_rate,
        calculator,
        final_answer
    ],
    max_steps=6,
    verbosity_level=1,
    grammar=None,
    planning_interval=None,
    name=None,
    description=None,
    prompt_templates=prompt_templates
)


GradioUI(agent).launch()