Update app.py

app.py

@@ -64,6 +64,7 @@ class DownloadRequest(BaseModel):
 
 
 # Download a Spotify track and return a download URL
+
 @app.post("/spot-track/{track_id}")
 def download_spotify_track(track_id: str):
     try:
@@ -78,10 +79,12 @@ def download_spotify_track(track_id: str):
             not_interface=False,
             method_save=1
         )
-         logger.info(f"Download completed for Spotify track: {track_id}")
 
-        # Recursively search for .mp3 files in the downloads directory
-        downloaded_files = list(Path("downloads").rglob("*.*"))
+        # Recursively search for audio files in the downloads directory
+        audio_extensions = [".mp3", ".ogg", ".flac", ".wav", ".m4a"]
+        downloaded_files = []
+        for ext in audio_extensions:
+            downloaded_files.extend(list(Path("downloads").rglob(f"*{ext}")))
         logger.info(f"Files in downloads directory: {downloaded_files}")
 
         if not downloaded_files: