from smolagents import CodeAgent, tool, OpenAIServerModel, FinalAnswerTool
import datetime
import pytz
import yfinance as yf
import os
from Gradio_UI import GradioUI

# Fetch the API key from the environment variable (if needed)
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)}"

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

# Load tools
final_answer = FinalAnswerTool()
currency_tool = get_currency_rate
time_tool = get_current_time_in_timezone

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

# Define example usage instructions
example_usage = """
### 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.
"""


# Launch the Gradio UI with the example usage
CustomGradioUI(agent).launch()