music recommendation app

app.py

@@ -1,6 +1,6 @@
 from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool
 from duckduckgo_search import DDGS
-from typing import List
+from typing import Optional
 import time
 import requests
 import pytz
@@ -15,50 +15,52 @@ from Gradio_UI import GradioUI
 
 
 @tool
-def search_internet(query: str, max_results: int = 3) -> str:
-    """A tool that searches the internet for information about a given query.
+def get_similar_songs(song_title: str, artist_name: str, num_recommendations: int = 5) -> str:
+    """A tool that suggests similar songs based on a given song and artist.
     
     Args:
-        query: The search query (e.g., 'latest AI news', 'climate change updates')
-        max_results: Maximum number of results to return (default: 3)
+        song_title: Name of the song
+        artist_name: Name of the artist
+        num_recommendations: Number of similar songs to return (default: 5)
     """
+    # LastFM API endpoint and parameters
+    API_KEY = "d17717e179b91347018a13452956c8a0"  # Replace with your LastFM API key
+    BASE_URL = "http://ws.audioscrobbler.com/2.0/"
+    
     try:
-        # Initialize the DuckDuckGo search
-        with DDGS() as ddgs:
-            # Perform the search
-            results = list(ddgs.text(query, max_results=max_results))
-            
-            if not results:
-                return f"No results found for '{query}'"
-            
-            # Format the results
-            formatted_results = f"Search results for '{query}':\n\n"
+        # Get similar tracks
+        params = {
+            "method": "track.getsimilar",
+            "artist": artist_name,
+            "track": song_title,
+            "api_key": API_KEY,
+            "format": "json",
+            "limit": num_recommendations
+        }
+        
+        response = requests.get(BASE_URL, params=params)
+        data = response.json()
+        
+        # Check if we got valid results
+        if "similartracks" not in data or "track" not in data["similartracks"]:
+            return f"No similar songs found for '{song_title}' by {artist_name}. Please check the song and artist names."
+        
+        # Format the recommendations
+        similar_tracks = data["similartracks"]["track"]
+        
+        result = f"Similar songs to '{song_title}' by {artist_name}:\n\n"
+        
+        for i, track in enumerate(similar_tracks, 1):
+            result += f"{i}. \"{track['name']}\" by {track['artist']['name']}\n"
+            result += f"   Match score: {float(track['match'])*100:.1f}%\n"
+            if 'playcount' in track:
+                result += f"   Playcount: {track['playcount']:,}\n"
+            result += "\n"
             
-            for i, result in enumerate(results, 1):
-                formatted_results += f"{i}. {result['title']}\n"
-                formatted_results += f"   {result['link']}\n"
-                formatted_results += f"   {result['body']}\n\n"
-                
-            return formatted_results
-    
-    except Exception as e:
-        return f"Error performing search: {str(e)}"
+        return result
 
-
-@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)}"
+        return f"Error finding similar songs: {str(e)}"
 
 
 final_answer = FinalAnswerTool()
@@ -82,7 +84,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, search_internet, get_current_time_in_timezone,], ## add your tools here (don't remove final answer)
+    tools=[final_answer, get_similar_songs], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,