app.py

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 !
@tool
def get_latest_football_matches(competition: str = "PL") -> str:
    """A tool that fetches the latest football match results from the Football-Data.org API.
    Args:
        competition: The competition code to fetch matches for (e.g., 'PL' for Premier League,
                     'CL' for Champions League, 'BL1' for Bundesliga, 'SA' for Serie A,
                     'PD' for La Liga, 'FL1' for Ligue 1). Defaults to 'PL'.
    """
    try:
        url = f"https://api.football-data.org/v4/competitions/{competition}/matches?status=FINISHED"
        headers = {"X-Auth-Token": "6b7b4c2b0b1d4996a8238ae546b3f049"}  
        response = requests.get(url, headers=headers)

        if response.status_code == 200:
            data = response.json()
            matches = data.get("matches", [])[-5:]  # Get last 5 matches
            if not matches:
                return "No recent matches found."
            result = f"Latest {competition} matches:\n"
            for match in matches:
                home = match["homeTeam"]["name"]
                away = match["awayTeam"]["name"]
                home_score = match["score"]["fullTime"]["home"]
                away_score = match["score"]["fullTime"]["away"]
                date = match["utcDate"][:10]
                result += f"  {date}: {home} {home_score} - {away_score} {away}\n"
            return result
        else:
            return f"Error fetching matches: {response.status_code} - {response.text}"
    except Exception as e:
        return f"Error fetching football matches: {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:
        tz = pytz.timezone(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,
)

# Load image generation 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=[final_answer, image_generation_tool, get_latest_football_matches],  ## ✅ tools registered here
    max_steps=6,
    verbosity_level=1,
    grammar=None,
    planning_interval=None,
    name=None,
    description=None,
    prompt_templates=prompt_templates
)

GradioUI(agent).launch()