from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool
import datetime
from datetime import date
import requests
import pytz
import yaml
from tools.final_answer import FinalAnswerTool
from tools.visit_webpage import VisitWebpageTool

from Gradio_UI import GradioUI

import os
from huggingface_hub import login
login(token = os.getenv('nangelov_hf_token'))
FOOTBALL_DATA_TOKEN = os.getenv('football_data_token')

@tool
def get_Competition_Code(competition_name: str)-> str:
    """A tool that returns the competition code for a given competition name.
    Args:
        competition_name: A string representing the full name of the competition, e.g., "Premier League" or "Bundesliga".
    """
    competitions = {
        "WC": "FIFA World Cup",
        "CL": "UEFA Champions League",
        "BL1": "Bundesliga",
        "DED": "Eredivisie",
        "BSA": "Campeonato Brasileiro Série A",
        "PD": "Primera Division",
        "FL1": "Ligue 1",
        "ELC": "Championship",
        "PPL": "Primeira Liga",
        "EC": "European Championship",
        "SA": "Serie A",
        "PL": "Premier League"
    }
    
    print(competition_name)
    # Iterate through the dictionary to find the matching competition name.
    for code, name in competitions.items():
        if name.lower() == competition_name.lower():
            print(code)
            return code
    # Return None if no match is found.
    return None

@tool
def search_Competition_Matches_Results_By_Date(competition_code: str, matchday: int)-> dict:
    """A tool that fetch football games results for a given competition from a specified date.
    Args:
        competition_code: A string representing the code of the competition, e.g., "Premier League" or "Bundesliga".
        matchday: An integer representing the matchday for which to retrieve match results format: Integer /[1-4]+[0-9]*/.
    """
   
    # Create the URL using an f-string
    url = f'https://api.football-data.org/v4/competitions/{competition_code}/matches?matchday={matchday}'
    print(url)
    
    # Set up the headers with your API token.
    headers = {
        'X-Auth-Token': FOOTBALL_DATA_TOKEN
    }
    
    # Make the request
    response = requests.get(url, headers=headers)
    
    # Check if the request was successful.
    if response.status_code == 200:
        return response.json()  # Parse and return the JSON response.
    else:
        # Return error details as a dict
        return {"error": response.status_code, "message": response.text}

@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)}"


final_answer = FinalAnswerTool()
visit_webpage = VisitWebpageTool()

# 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,
)


# Import 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, visit_webpage, search_Competition_Matches_Results_By_Date, get_current_time_in_timezone, get_Competition_Code], ## add your tools here (don't remove final answer)
    max_steps=6,
    verbosity_level=1,
    grammar=None,
    planning_interval=None,
    name=None,
    description=None,
    prompt_templates=prompt_templates
)


GradioUI(agent).launch()