Update app.py

app.py

@@ -69,12 +69,20 @@ app = FastAPI(
 DOWNLOADS_DIR = "downloads"
 os.makedirs(DOWNLOADS_DIR, exist_ok=True)
 
-# Mount the downloads directory
-app.mount("/files", StaticFiles(directory=DOWNLOADS_DIR), name="files")
-
-
 
+# Mount the downloads directory with cache control headers
+class CacheControlStaticFiles(StaticFiles):
+    def __init__(self, directory: str, name: str = None):
+        super().__init__(directory=directory, name=name)
+        
+    async def get_response(self, path: str, scope):
+        response = await super().get_response(path, scope)
+        response.headers["Cache-Control"] = "no-cache, no-store, must-revalidate"
+        response.headers["Pragma"] = "no-cache"
+        response.headers["Expires"] = "0"
+        return response
 
+app.mount("/files", CacheControlStaticFiles(directory=DOWNLOADS_DIR), name="files")
 
 class DownloadRequest(BaseModel):
     url: str
@@ -86,7 +94,6 @@ class DownloadResponse(BaseModel):
     download_url: str = None
     file_size: int = None  # Include file size in the response
 
-
 @app.post("/download", response_model=DownloadResponse)
 async def download_file(request: DownloadRequest):
     try:
@@ -140,7 +147,7 @@ async def download_file(request: DownloadRequest):
         file_size = os.path.getsize(downloaded_file)
         
         # Generate the download URL and URL encode the filename
-        space_url = os.getenv("SPACE_URL", "https://tecuts-vob.hf.space")
+        space_url = os.getenv("SPACE_URL", "https://tecuts-vob.hf.space/")
         encoded_filename = quote(downloaded_file)  # URL encode the filename
         download_url = f"{space_url}/files/{timestamp}/{encoded_filename}"
         logger.info(f"Generated download URL: {download_url}")