Update app.py

Add get_songs tool

app.py

@@ -49,6 +49,19 @@ def get_player_info(player_name:str)-> dict:
     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')
+    artist_name = artist_name.lower()
+    df['artist(s)_name'] = df['artist(s)_name'].str.lower()
+    songs = 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
@@ -98,7 +111,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, get_current_time_in_timezone, get_player_info], ## add your tools here (don't remove final answer)
+    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,