Update app.py

app.py

@@ -71,6 +71,8 @@ os.makedirs(DOWNLOADS_DIR, exist_ok=True)
 # Mount the downloads directory
 app.mount("/files", StaticFiles(directory=DOWNLOADS_DIR), name="files")
 
+
+
 class DownloadRequest(BaseModel):
     url: str
     
@@ -79,14 +81,10 @@ class DownloadResponse(BaseModel):
     message: str
     filename: str = None
     download_url: str = None
+    file_size: int = None  # Include file size in the response
 
 
 
-def verify_file_size(file_path: str, expected_size: int):
-    actual_size = os.path.getsize(file_path)
-    if actual_size < expected_size:
-        raise Exception(f"Downloaded file size mismatch: expected {expected_size} bytes, got {actual_size} bytes")
-
 @app.post("/download", response_model=DownloadResponse)
 async def download_file(request: DownloadRequest):
     try:
@@ -136,8 +134,8 @@ async def download_file(request: DownloadRequest):
         downloaded_file = files[0]
         logger.info(f"Downloaded file: {downloaded_file}")
 
-        # Verify the size of the downloaded file (expected size in bytes, e.g., 7000000 for 7MB)
-        verify_file_size(downloaded_file, expected_size=7000000)
+        # Get the file size
+        file_size = os.path.getsize(downloaded_file)
         
         # Generate the download URL
         space_url = os.getenv("SPACE_URL", "http://localhost:7860")
@@ -151,7 +149,8 @@ async def download_file(request: DownloadRequest):
             success=True,
             message="File downloaded successfully",
             filename=downloaded_file,
-            download_url=download_url
+            download_url=download_url,
+            file_size=file_size  # Include the file size in the response
         )
             
     except subprocess.CalledProcessError as e:
@@ -172,6 +171,7 @@ async def download_file(request: DownloadRequest):
             os.chdir(original_dir)
 
 
+
 @app.get("/test")
 async def test():
     """Test endpoint to verify setup"""