Update app.py

app.py

@@ -102,18 +102,19 @@ class DownloadResponse(BaseModel):
 
 
 # Function to wait until the file is fully downloaded
-def wait_for_file_to_complete(file_path, initial_size, max_wait_time=30):
+def wait_for_file_to_complete(file_path, initial_size, max_wait_time=60):
     wait_time = 0
     while True:
         current_size = os.path.getsize(file_path)
         if current_size == initial_size:
             wait_time += 1
             if wait_time >= max_wait_time:
+                logger.info(f"Timeout reached waiting for file to complete: {file_path}")
                 break
         else:
             wait_time = 0
         initial_size = current_size
-        time.sleep(1)
+        time.sleep(2)  # Wait 2 seconds before checking again
 
 @app.post("/download", response_model=DownloadResponse)
 async def download_file(request: DownloadRequest):
@@ -159,6 +160,10 @@ async def download_file(request: DownloadRequest):
         file_size = os.path.getsize(os.path.join(download_subdir, downloaded_file))
         logger.info(f"Final file size: {file_size} bytes")
 
+        # If the size is still incorrect, raise an error
+        if file_size < 7_000_000:  # Assuming 7 MB is the minimum expected file size
+            raise Exception(f"File seems incomplete. Final size: {file_size} bytes.")
+
         # Generate the download URL and URL encode the filename
         space_url = os.getenv("SPACE_URL", "https://tecuts-vob.hf.space")
         encoded_filename = quote(downloaded_file)
@@ -187,8 +192,6 @@ async def download_file(request: DownloadRequest):
         )
 
 
-
-
 @app.get("/test")
 async def test():
     """Test endpoint to verify setup"""