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

API_KEY = os.getenv("KEY")


@tool
def get_currency_rate(currency_pair: str) -> str:
    """Fetches the current exchange rate for a given currency pair (e.g., 'USD-EUR').
    Args:
        currency_pair: The currency pair to query, formatted as 'BASE-CURRENCY'.
    """
    try:
        # Use a free API like exchangerate-api.com
        base, target = currency_pair.split("-")
        response = requests.get(f"https://api.exchangerate-api.com/v4/latest/{base}")
        data = response.json()
        rate = data["rates"].get(target)
        if rate:
            return f"Current {currency_pair} rate: {rate}"
        else:
            return f"Invalid currency pair: {currency_pair}"
    except Exception as e:
        return f"Error fetching rate: {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)}"

model = OpenAIServerModel(
    model_id="deepseek-chat",
    api_base="https://api.deepseek.com",
    api_key=API_KEY
)

# Load your custom tool and others
currency_tool = get_currency_rate
final_answer = FinalAnswerTool()

agent = CodeAgent(
    model=model,
    tools=[currency_tool, final_answer],  # Include your tool here
    max_steps=3,
    verbosity_level=1
)

GradioUI(agent).launch()