Update app.py

app.py

@@ -11,6 +11,8 @@ from fastapi.staticfiles import StaticFiles
 from dotenv import load_dotenv
 from pydantic import BaseModel
 from urllib.parse import quote
+from pathlib import Path
+
 
 # Set up logging
 logging.basicConfig(level=logging.INFO)
@@ -77,24 +79,19 @@ def download_spotify_track(track_id: str):
             method_save=1
         )
 
-        # Search for the downloaded file
-        downloaded_files = os.listdir("downloads")
+        # Recursively search for .mp3 files in the downloads directory
+        downloaded_files = list(Path("downloads").rglob("*.mp3"))
         logger.info(f"Files in downloads directory: {downloaded_files}")
 
-        # Fetch track metadata from Spotify (if needed)
-        # For now, assume the file is named like "Artist - Track Title.mp3"
-        # Look for files that match the pattern
-        matching_files = [file for file in downloaded_files if file.endswith(".mp3")]
-
-        if not matching_files:
+        if not downloaded_files:
             raise HTTPException(status_code=500, detail="File download failed")
 
         # Use the first matching file
-        filename = matching_files[0]
-        filepath = os.path.join("downloads", filename)
+        filepath = downloaded_files[0]
+        filename = filepath.name
 
         # Return the download URL
-        download_url = f"/downloads/{quote(filename)}"
+        download_url = f"/downloads/{quote(str(filepath.relative_to('downloads')))}"
         logger.info(f"Download successful: {download_url}")
         return {"download_url": download_url}
     except Exception as e: