Update app.py

app.py

@@ -10,34 +10,37 @@ from Gradio_UI import GradioUI
 
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
 
-# Initialize the Genius API client with access token
-genius = lyricsgenius.Genius("2kWJ8Wy72HT3T8qjkh__gGB3fAbCLtTbC68j1i9tvOM9DF9ikSr95wm-hOQLZq9_")
 
+@tool
+def get_song_lyrics(song_name: str, artist_name: str) -> str:
+    """Fetch lyrics from Genius based on song title and artist name."""
+    genius_token = os.getenv("Genius")
+    if not genius_token:
+        return "Error: Genius API token not found."
 
-# Setting some preferences
-genius.remove_section_headers = True  # Removes section headers like [Chorus], [Verse], etc.
-genius.excluded_terms = ["(Remix)", "(Live)"]  # Excludes unwanted terms from search results
+    # Initialize Genius API client
+    genius = lyricsgenius.Genius(genius_token)
 
-@tool
-def get_song_lyrics(song_name:str, artist_name:str)-> 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 fetches lyrics from Genius based on the song title and artist name.
-    Args:
-        song_name: A string representing a valid name of a song.
-        artist_name: A string representing a valid name of a song.
-    """
+    # Add headers to avoid 403 errors
+    headers = {
+        "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/58.0.3029.110 Safari/537.36"
+    }
+    genius._headers = headers  # Set the headers in the API client
+
+    # Optional settings
+    genius.remove_section_headers = True  # Removes [Chorus], [Verse] headers
+    genius.excluded_terms = ["(Remix)", "(Live)"]
 
     try:
         # Search for the song by title and artist
         song = genius.search_song(song_name, artist_name)
         
         if song:
-            # Return the song lyrics if found
-            return f"The lyrics of the {song_name} by {artist_name} is ... {song.lyrics}"
+            return f"The lyrics of the song '{song_name}' by {artist_name} are ...\n{song.lyrics[:300]}..."
         else:
             return "Song not found."
     except Exception as e:
-        return f"Error fetching song lyrics of the {song_name} by {artist_name}: {str(e)}"
+        return f"Error fetching song lyrics of '{song_name}' by {artist_name}: {str(e)}"