Update app.py

app.py

@@ -63,23 +63,24 @@ class DownloadRequest(BaseModel):
 
 
 
-# Download a Spotify track and return a download URL
 
+
+# Download a Spotify track and return a download URL
 @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))
+        for item in downloads_dir.iterdir():
+            if item.is_file():
+                item.unlink()
+            elif item.is_dir():
+                shutil.rmtree(item)
                 
         # Download the track using spotloader
         logger.info(f"Downloading Spotify track: {track_id}")
         spo.download_track(
             link_track=f"https://open.spotify.com/track/{track_id}",
-            output_dir="downloads",
+            output_dir=str(downloads_dir),
             quality_download="VERY_HIGH",
             recursive_quality=False,
             recursive_download=False,
@@ -87,27 +88,25 @@ def download_spotify_track(track_id: str):
             method_save=1
         )
         
-
         file_extension = "ogg"
         # Recursively search for the file in the downloads directory
         filepath = None
-        for root, dirs, files in os.walk("downloads"):
+        for root, dirs, files in os.walk(downloads_dir):
             for file in files:
                 if file.endswith(f'.{file_extension}'):
-                    filepath = os.path.join(root, file)
+                    filepath = Path(root) / file
                     break
             if filepath:
                 break
 
         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")
+        
+        file_size = filepath.stat().st_size
+        logger.info(f"Downloaded file size: {file_size} bytes")
 
         # Return the download URL
-        relative_path = quote(str(os.path.relpath(filepath, "downloads")))
-        # Remove spaces from the relative path
+        relative_path = quote(str(filepath.relative_to(downloads_dir)))
         download_url = f"{BASE_URL}/downloads/{relative_path}"
         logger.info(f"Download successful: {download_url}")
         return {"download_url": download_url}
@@ -116,6 +115,7 @@ def download_spotify_track(track_id: str):
         raise HTTPException(status_code=500, detail=str(e))
 
 
+
 @app.get("/")
 def read_root():
     return {"message": "running"}