Update app.py

app.py

@@ -1,3 +1,4 @@
+import shutil
 from fastapi import FastAPI, HTTPException
 from deezspot.deezloader import DeeLogin
 import requests
@@ -64,6 +65,13 @@ def download_track(track_id: str, quality: str = "MP3_320"):
         artist_name = track_info.get("artist", {}).get("name", "unknown")
         expected_filename = f"{artist_name} - {track_title}.mp3".replace("/", "_")  # Sanitize filename
 
+        # Clear the downloads directory
+        for root, dirs, files in os.walk("downloads"):
+            for file in files:
+                os.remove(os.path.join(root, file))
+            for dir in dirs:
+                shutil.rmtree(os.path.join(root, dir))
+
         # Download the track using deezspot
         logger.info(f"Downloading track: {expected_filename}")
         dl.download_trackdee(
@@ -74,21 +82,13 @@ def download_track(track_id: str, quality: str = "MP3_320"):
             recursive_download=False
         )
 
-        # Search for the downloaded file in the downloads directory
+        # Get the downloaded file
         downloaded_files = os.listdir("downloads")
-        logger.warning(f"Files in downloads directory: {downloaded_files}")
-
-        # Try to find the file with a more flexible approach
-        found_filepath = None
-        for file in downloaded_files:
-            if (track_title.lower() in file.lower() or artist_name.lower() in file.lower()) and file.lower().endswith('.mp3'):
-                found_filepath = os.path.join("downloads", file)
-                logger.info(f"Found matching file: {found_filepath}")
-                break
-
-        if not found_filepath:
+        if len(downloaded_files) == 0:
             raise HTTPException(status_code=500, detail="File download failed")
 
+        found_filepath = os.path.join("downloads", downloaded_files[0])
+
         # Return the download URL
         download_url = f"/downloads/{os.path.basename(found_filepath)}"
         logger.info(f"Download successful: {download_url}")