Update app.py

app.py

@@ -68,6 +68,13 @@ class DownloadRequest(BaseModel):
 @app.post("/spot-track/{track_id}")
 def download_spotify_track(track_id: str):
     try:
+        # 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 spotloader
         logger.info(f"Downloading Spotify track: {track_id}")
         spo.download_track(
@@ -79,28 +86,37 @@ def download_spotify_track(track_id: str):
             not_interface=False,
             method_save=1
         )
+        except Exception as e:
+            logger.error(f"Error downloading file: {e}")
+            raise HTTPException(status_code=500, detail="File download failed")
 
         # 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:
-            raise HTTPException(status_code=500, detail="File download failed")
+        file_extension = "ogg"
+        # Recursively search for the file in the downloads directory
+        filepath = None
+        for root, dirs, files in os.walk("downloads"):
+            for file in files:
+                if file.endswith(f'.{file_extension}'):
+                    filepath = os.path.join(root, file)
+                    break
+            if filepath:
+                break
 
-        # Use the first matching file
-        filepath = downloaded_files[0]
-        filename = filepath.name
+        if not filepath:
+            raise HTTPException(status_code=500, detail=f"{file_extension} file not found after download")
+        if filepath:
+            file_size = os.path.getsize(filepath)
+            logger.info(f"Downloaded file size: {file_size} bytes")
 
         # Return the download URL
-        download_url = f"{BASE_URL}/downloads/{quote(str(filepath.relative_to('downloads')))}"
+        relative_path = quote(str(os.path.relpath(filepath, "downloads")))
+        # Remove spaces from the relative path
+        download_url = f"{BASE_URL}/downloads/{relative_path}"
         logger.info(f"Download successful: {download_url}")
         return {"download_url": download_url}
     except Exception as e:
-        logger.error(f"Error downloading Spotify track: {e}")
-        raise HTTPException(status_code=500, detail="Failed to download Spotify track. Please try again later.")
+        logger.error(f"Error downloading track: {e}")
+        raise HTTPException(status_code=500, detail=str(e))
 
 
 @app.get("/")