Update app.py

app.py

@@ -17,38 +17,51 @@ class MediaDownloader:
     def __init__(self):
         self.temp_dir = tempfile.mkdtemp()
         self.create_directories()
+        self.setup_yt_dlp()
 
-    def create_directories(self):
-        self.image_path = os.path.join(self.temp_dir, "images")
-        self.video_path = os.path.join(self.temp_dir, "videos")
-        os.makedirs(self.image_path, exist_ok=True)
-        os.makedirs(self.video_path, exist_ok=True)
-
-    def cleanup(self):
-        shutil.rmtree(self.temp_dir)
-
-    def download_video(self, url):
-        output_template = os.path.join(self.video_path, '%(title)s.%(ext)s')
-        ydl_opts = {
+    def setup_yt_dlp(self):
+        self.ydl_opts = {
             'format': 'best',
-            'outtmpl': output_template,
             'ignoreerrors': True,
             'no_warnings': True,
             'quiet': True,
             'extract_flat': False,
             'http_headers': {
-                'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36'
+                'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36',
+                'Accept': 'text/html,application/xhtml+xml,application/xml;q=0.9,*/*;q=0.8',
+                'Accept-Language': 'en-us,en;q=0.5',
+                'Sec-Fetch-Mode': 'navigate'
             }
         }
 
+    def create_directories(self):
+        self.image_path = os.path.join(self.temp_dir, "images")
+        self.video_path = os.path.join(self.temp_dir, "videos")
+        os.makedirs(self.image_path, exist_ok=True)
+        os.makedirs(self.video_path, exist_ok=True)
+
+    def cleanup(self):
         try:
-            with yt_dlp.YoutubeDL(ydl_opts) as ydl:
+            shutil.rmtree(self.temp_dir)
+        except Exception as e:
+            print(f"Cleanup error: {str(e)}")
+
+    def download_video(self, url):
+        self.ydl_opts['outtmpl'] = os.path.join(self.video_path, '%(title)s.%(ext)s')
+        try:
+            with yt_dlp.YoutubeDL(self.ydl_opts) as ydl:
+                try:
+                    ydl.update()  # Update yt-dlp before downloading
+                except Exception as e:
+                    print(f"Update error: {str(e)}")
+                
                 info = ydl.extract_info(url, download=True)
                 if info:
                     filename = ydl.prepare_filename(info)
                     return filename if os.path.exists(filename) else None
             return None
-        except:
+        except Exception as e:
+            print(f"Video download error: {str(e)}")
             return None
 
     def download_images(self, url):
@@ -72,7 +85,7 @@ class MediaDownloader:
                 try:
                     response = requests.get(img_url, headers={
                         'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36'
-                    })
+                    }, timeout=10)
                     response.raise_for_status()
 
                     timestamp = datetime.now().strftime("%Y%m%d_%H%M%S_%f")
@@ -97,6 +110,10 @@ class MediaDownloader:
             return None
 
     def download_media(self, url):
+        if 'tiktok.com' in url:
+            # Special handling for TikTok
+            self.ydl_opts['format'] = 'bestvideo+bestaudio/best'
+            
         video_path = self.download_video(url)
         if video_path:
             return [video_path]