from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool
import datetime
import requests
import pytz
import yaml
from tools.final_answer import FinalAnswerTool
import time
import os

from Gradio_UI import GradioUI

lastfm_api_root = 'http://ws.audioscrobbler.com/2.0/'
cache = {
    "current_track": None,
    "last_fetched": 0,
    "listening": False,
    "user": None
}

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

@tool
def get_recent_tracks(user: str) -> str:
    """
    Retrieves the most recent track information for a given Last.fm user.

    This function queries the Last.fm API to fetch the latest track that the specified user
    has listened to. To minimize unnecessary API calls, it uses a caching mechanism:
    
    - If the requested user is the same as the cached user and the cached data is less than
      60 seconds old, the function returns the cached track information.
    - Otherwise, it makes a new API request to fetch the latest data and updates the cache.

    The function formats the response based on whether the track is currently being played.
    If the track has an attribute indicating that it is currently playing, the output is prefixed
    with "Currently listening:"; otherwise, it is prefixed with "Recently listened:".

    Environment Variables:
        LASTFM_API_KEY : API key for authenticating with the Last.fm API.

    Args:
        user: The Last.fm username for which to retrieve the recent track information.
                    Must be a non-empty string.

    Returns:
        str: A formatted string with the track information:
             - "Currently listening: <track>" if the track is playing.
             - "Recently listened: <track>" otherwise.
             In case of errors or missing data, an appropriate error message is returned.

    Examples:
        >>> get_recent_tracks("someuser")
        "Currently listening: Track Title by Artist Name"

        >>> get_recent_tracks("anotheruser")
        "Recently listened: Track Title by Artist Name"

    Error Handling:
        - If no user is provided, an error message is returned.
        - If the API endpoint returns an error (e.g., HTTP 404), a corresponding error message is returned.
        - If no track information is available or an unexpected error occurs, the function returns an
          appropriate error message.
          
    Note:
        The cached data is valid for 60 seconds. After this period, or if a different user is requested,
        the function will fetch new data from the API.
    """
    global cache

    if not user:
        return "Couldn't get the recent tracks information because the user is not provided"
    
    current_time = time.time()
    
    # Use cached data if it's for the same user and is not expired (within 60 seconds)
    if (cache['user'] == user and 
        cache['current_track'] is not None and 
        (current_time - cache['last_fetched'] <= 60)):
        
        prompt = (f'Currently listening: {cache["current_track"]}' 
                  if cache['listening'] 
                  else f'Recently listened: {cache["current_track"]}')
        return prompt

    # Otherwise, fetch new data
    try:
        params = {
            "method": "user.getrecenttracks",
            "user": user,
            "api_key": os.getenv('LASTFM_API_KEY'),
            "limit": 1,
            "format": "json"
        }
        response = requests.get(lastfm_api_root, params=params)
        if response.status_code == 200:
            data = response.json()
            track = data['recenttracks']['track'][0]
            artist = track['artist']['#text']
            title = track['name']

            # Update cache
            cache['current_track'] = f'{title} by {artist}'
            cache['last_fetched'] = current_time
            cache['user'] = user
            cache['listening'] = (track.get('@attr', {}).get('nowplaying') == 'true')
        else:
            return f"Failed to fetch data: HTTP {response.status_code}: {response.json()}"
    except IndexError:
        return f'No recent track information available for {user}'
    except Exception as e:
        return f'An error occurred: {e}'

    prompt = (f'currently listening: {cache["current_track"]}' 
              if cache['listening'] 
              else f'recently listened: {cache["current_track"]}')
    return f'{user} {prompt}'


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, image_generation_tool, get_recent_tracks], ## 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()