Update app.py

app.py

@@ -17,6 +17,7 @@ from fastapi import BackgroundTasks
 from collections import defaultdict
 import time
 from datetime import timedelta
+import gc
 
 
 # Set up logging
@@ -218,7 +219,8 @@ def download_track(request: DownloadRequest):
         # 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}
+        gc.collect()
+        return {"download_url": download_url, "requests_remaining": rate_limiter.max_requests - rate_limiter.get_current_count(user_ip)}
     except Exception as e:
         logger.error(f"Error downloading track: {e}")
         raise HTTPException(status_code=500, detail=str(e))