app.py

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
)

# Add example usage instructions to the Gradio UI description
description = """
### Currency and Time Agent
This app allows you to:
1. Get the current exchange rate for a currency pair (e.g., USD to JPY).
2. Get the current time in a specific timezone (e.g., America/New_York).

#### Example Queries:
- "What is the current exchange rate for USD to JPY?"
- "Get the EUR to GBP rate."
- "What is the current time in America/New_York?"
- "Get the time in Asia/Tokyo."
"""

# Launch the Gradio UI
GradioUI(agent, description=description).launch()