Update app.py

app.py

@@ -82,21 +82,22 @@ def download_track(track_id: str, quality: str = "MP3_320"):
             recursive_download=False
         )
 
-        # Get the downloaded file
-        downloaded_files = os.listdir("downloads")
-        if len(downloaded_files) == 0:
-            raise HTTPException(status_code=500, detail="File download failed")
-
-        # Find the MP3 file
-        mp3_files = [file for file in downloaded_files if file.endswith('.mp3')]
-        if len(mp3_files) == 0:
-            raise HTTPException(status_code=500, detail="MP3 file not found after download")
+        # Recursively search for the MP3 file in the downloads directory
+        mp3_filepath = None
+        for root, dirs, files in os.walk("downloads"):
+            for file in files:
+                if file.endswith('.mp3'):
+                    mp3_filepath = os.path.join(root, file)
+                    break
+            if mp3_filepath:
+                break
 
-        found_file = mp3_files[0]
-        found_filepath = os.path.join("downloads", found_file)
+        if not mp3_filepath:
+            raise HTTPException(status_code=500, detail="MP3 file not found after download")
 
         # Return the download URL
-        download_url = f"/downloads/{os.path.basename(found_file)}"
+        relative_path = os.path.relpath(mp3_filepath, "downloads")
+        download_url = f"/downloads/{relative_path}"
         logger.info(f"Download successful: {download_url}")
         return {"download_url": download_url}
     except Exception as e: