from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool,OpenAIServerModel
import datetime
import requests
import pytz
import yaml
from tools.final_answer import FinalAnswerTool
import os
from Gradio_UI import GradioUI
import gradio as gr
import requests
import yfinance as yf

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:
        base, target = currency_pair.split("-")
        # Use yfinance to get the exchange rate
        ticker = f"{base}{target}=X"  # Format for yfinance (e.g., "USDEUR=X")
        data = yf.Ticker(ticker)
        rate = data.history(period="1d")["Close"].iloc[-1]
        return f"Current {currency_pair} rate: {rate}"
    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()

# Add a Markdown section for example usage
examples = """
### Example Usage
1. **Currency Rate**: Ask for the exchange rate of a currency pair.
   - Example: "What is the current exchange rate for USD to JPY?"
   - Example: "Get the EUR to GBP rate."

2. **Current Time**: Ask for the current time in a specific timezone.
   - Example: "What is the current time in America/New_York?"
   - Example: "Get the time in Asia/Tokyo."
"""

# Modify the Gradio UI to include the examples
def launch_with_examples(agent):
    with gr.Blocks() as demo:
        gr.Markdown("# Currency Agent")
        gr.Markdown(examples)
        GradioUI(agent).render()
    demo.launch()

# Launch the app
launch_with_examples(agent)