Update app.py

app.py

@@ -11,6 +11,7 @@ import tempfile
 from pathlib import Path
 from dotenv import load_dotenv
 from urllib.parse import quote
+import time
 
 # Set up logging
 logging.basicConfig(level=logging.INFO)
@@ -69,6 +70,8 @@ app = FastAPI(
 
 
 
+
+
 # Create downloads directory if it doesn't exist
 DOWNLOADS_DIR = "downloads"
 TEMP_DIR = "temp"
@@ -149,13 +152,16 @@ async def download_file(request: DownloadRequest):
         downloaded_file = files[0]
         logger.info(f"Downloaded file: {downloaded_file}")
 
+        # Ensure the file is fully written
+        time.sleep(1)  # Sleep for 1 second
+
         # Get the file size
         file_size = os.path.getsize(downloaded_file)
 
-        # Check for encryption or corruption issues
+        # Check for potential issues in file handling
         if process.stderr:
             raise Exception(f"Error in downloaded file: {process.stderr}")
-        
+
         # Move the completed download to the final directory
         final_download_subdir = os.path.join(DOWNLOADS_DIR, timestamp)
         os.makedirs(final_download_subdir, exist_ok=True)