Update app.py

app.py

@@ -13,6 +13,7 @@ import mimetypes
 import urllib.parse
 import subprocess
 import json
+from tqdm import tqdm
 
 # Configure logging
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
@@ -100,25 +101,31 @@ def download_file(url):
         response = session.get(url, stream=True, allow_redirects=True)
         response.raise_for_status()
         
-        # Check if we can get the filename from Content-Disposition header
-        content_disposition = response.headers.get('Content-Disposition')
-        if content_disposition:
-            filename = re.findall("filename=(.+)", content_disposition)[0].strip('"')
-        else:
-            # If not, use a default name with .mp4 extension
-            filename = 'downloaded_video.mp4'
+        # Get the total file size
+        total_size = int(response.headers.get('content-length', 0))
+        
+        # Use a default name with .mp4 extension
+        filename = 'downloaded_video.mp4'
 
         # Save the content to a temporary file with .mp4 extension
         with tempfile.NamedTemporaryFile(delete=False, suffix='.mp4') as temp_file:
+            progress_bar = tqdm(total=total_size, unit='iB', unit_scale=True, desc=filename)
             for chunk in response.iter_content(chunk_size=8192):
-                if chunk:
-                    temp_file.write(chunk)
+                size = temp_file.write(chunk)
+                progress_bar.update(size)
+            progress_bar.close()
             temp_file_path = temp_file.name
 
+        # Check if the downloaded file size matches the expected size
+        actual_size = os.path.getsize(temp_file_path)
+        if total_size != 0 and actual_size != total_size:
+            logger.error(f"Downloaded file size ({actual_size} bytes) does not match expected size ({total_size} bytes)")
+            raise Exception("Incomplete download")
+
     logger.info(f"File downloaded and saved as: {temp_file_path}")
+    logger.info(f"File size: {actual_size} bytes")
     return temp_file_path
 
-
 def get_file_info(file_path):
     try:
         result = subprocess.run(['ffprobe', '-v', 'quiet', '-print_format', 'json', '-show_format', '-show_streams', file_path],