from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool
import datetime
import requests
import pytz
import yaml
import pandas as pd
from tools.final_answer import FinalAnswerTool
# from nba_api.stats.static import players
# from nba_api.stats.endpoints import commonplayerinfo


from Gradio_UI import GradioUI


def get_player(name):
    player = players.find_players_by_full_name(name)
    if player:
        return player[0]
    return None

@tool
def get_player_info(player_name:str)-> dict:
    """
    A tool that gets details about an NBA player using the NBA API.
    the details that are returned in the dictionary are:
    Name, Height, Weight, Birth Date, College, Country, Draft Year, Draft Round, Draft Number, Years Pro, Current Team
    Args:
        player_name: the full name of the NBA player (e.g. Kobe Bryant)
    """
    # Fetch player bio details
    player_id = get_player(player_name)['id']
    player_info = commonplayerinfo.CommonPlayerInfo(player_id=player_id)
    data = player_info.get_dict()
    info = data['resultSets'][0]['rowSet'][0]  # The first (and only) row

    # Dictionary mapping for extracted values
    player_details = {
        "Name": info[3],  # Full name
        "Height": info[11],  # Height in feet-inches
        "Weight": info[12],  # Weight in pounds
        "Birth Date": info[7].split('T')[0],  # Date of birth
        "College": info[8],  # College name
        "Country": info[9],  # Country of birth
        "Draft Year": info[29],  # Year drafted
        "Draft Round": info[30],  # Draft round
        "Draft Number": info[31],  # Pick number
        "Years Pro": info[13],  # Years in the NBA
        "Team": info[20] if info[20] != '' else 'NA'   # Current team
    }
    return player_details


@tool
def get_songs_by_artist(artist_name:str)-> list:
    """
    A tool that gets a name of an artist, read a file with song details and return all the artist's songs according to the file.
    Args:
        artist_name: the name of the artist
    """
    df = pd.read_csv('spotify-2023.csv', encoding='latin1')
    artist_name = artist_name.lower()
    df['artist(s)_name'] = df['artist(s)_name'].str.lower()
    songs = df[df.apply(lambda row: artist_name in row['artist(s)_name'].lower(), axis=1)]['track_name'].tolist()
    return songs


# Below is an example of a tool that does nothing. Amaze us with your creativity !
@tool
def my_custom_tool(arg1:str, arg2:int)-> str: #it's import to specify the return type
    #Keep this format for the description / args / args description but feel free to modify the tool
    """A tool that does nothing yet 
    Args:
        arg1: the first argument
        arg2: the second argument
    """
    return "What magic will you build ?"

@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()

# 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, get_current_time_in_timezone, get_songs_by_artist], ## 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()