Update app.py

app.py

@@ -1,5 +1,5 @@
 # app.py
-from fastapi import FastAPI, HTTPException
+from fastapi import FastAPI, HTTPException, Response
 from fastapi.staticfiles import StaticFiles
 from pydantic import BaseModel
 import subprocess
@@ -65,15 +65,15 @@ app = FastAPI(
     version="1.0.0"
 )
 
+
 # Create downloads directory if it doesn't exist
 DOWNLOADS_DIR = "downloads"
 os.makedirs(DOWNLOADS_DIR, exist_ok=True)
 
-
 # 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)
+    def __init__(self, directory: str):
+        super().__init__(directory=directory)
         
     async def get_response(self, path: str, scope):
         response = await super().get_response(path, scope)
@@ -94,6 +94,7 @@ 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:
@@ -147,7 +148,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}")