Update app.py

app.py

@@ -29,6 +29,7 @@ from fastapi.staticfiles import StaticFiles
 from collections import defaultdict
 from starlette.responses import JSONResponse
 import logging
+import gc
 
 tmp_dir = tempfile.gettempdir()
 BASE_URL = "https://chrunos-multi.hf.space"
@@ -219,6 +220,8 @@ async def download_high_quality_video(request: Request):
     # Increment the user's request count
     user_info["count"] += 1
 
+    gc.collect()
+
     return {"url": download_url}
 
 
@@ -275,6 +278,7 @@ async def download_high_quality_video(request: Request):
     downloaded_file = downloaded_files[0]
     encoded_filename = urllib.parse.quote(downloaded_file.name)
     download_url = f"{BASE_URL}/file/{encoded_filename}"
+    gc.collect()
     return {"url": download_url}
 
 
@@ -312,6 +316,7 @@ async def download_audio(request: Request):
     downloaded_file = downloaded_files[0]
     encoded_filename = urllib.parse.quote(downloaded_file.name)
     download_url = f"{BASE_URL}/file/{encoded_filename}"
+    gc.collect()
     return {"url": download_url}
 
 # Configure logging
@@ -373,6 +378,7 @@ async def search_and_download_song(request: Request):
 
     # Log just before returning the response
     logging.info("Preparing to send response back to the client")
+    gc.collect()
     return JSONResponse(content={"url": download_url}, status_code=200)
 
 # Mount the static files directory