test with pygame

app.py

@@ -1,7 +1,7 @@
 from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool
 import datetime
 import requests
-import vlc
+import pygame
 import time
 import pytz
 import yaml
@@ -34,7 +34,28 @@ def get_current_time_in_timezone(timezone: str) -> str:
         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)}"
+        
+def play_audio_from_url(url):
+    # Initialize the pygame mixer
+    pygame.mixer.init()
+
+    # Stream the audio content from the URL
+    response = requests.get(url, stream=True)
+    response.raise_for_status()
+
+    # Load the audio data into a BytesIO object
+    audio_data = BytesIO(response.content)
 
+    # Load the audio data into pygame
+    pygame.mixer.music.load(audio_data, 'mp3')
+
+    # Play the audio
+    pygame.mixer.music.play()
+
+    # Keep the script running while the audio is playing
+    while pygame.mixer.music.get_busy():
+        pygame.time.Clock().tick(10)
+        
 @tool
 def play_preview_deezer(track: str, artist:str) -> None:
     """Play a preview of a track from Deezer.
@@ -64,12 +85,10 @@ def play_preview_deezer(track: str, artist:str) -> None:
             print(f"Artist: {first_track['artist']['name']}")
             print(f"Album: {first_track['album']['title']}")
             print(f"link: {first_track['link']}")
-            # Play the track using VLC
+            # Play the track
             print("Playing preview: ", first_track['preview'])
-            player = vlc.MediaPlayer(first_track['preview'])
-            player.play()
-            # wait time for vlc to load
-            time.sleep(30)
+            play_audio_from_url(first_track['preview'])
+            time.sleep(2)
         else:
             print("No tracks found.")
     else: