Handle yt-dlp bot check

whichframe.py

@@ -19,35 +19,18 @@ model, preprocess = openai_clip.load("ViT-B/32", device=device)
 
 def fetch_video(url):
     try:
-        browsers = ['chrome', 'firefox', 'edge', 'safari', 'opera']
-        for browser in browsers:
-            try:
-                ydl_opts = {
-                    'format': 'bestvideo[height<=360][ext=mp4]',
-                    'quiet': True,
-                    'no_warnings': True,
-                    'cookiesfrombrowser': (browser,),
-                }
-                with yt_dlp.YoutubeDL(ydl_opts) as ydl:
-                    info = ydl.extract_info(url, download=False)
-                    duration = info.get('duration', 0)
-                    if duration >= 300:  # 5 minutes
-                        st.error("Please find a YouTube video shorter than 5 minutes.")
-                        st.stop()
-                    video_url = info['url']
-                    return None, video_url
-            except Exception:
-                continue
-        
         ydl_opts = {
-            'format': 'bestvideo[height<=360][ext=mp4]',
+            'format': 'bestvideo[height<=360][ext=mp4]/best[height<=360]',
             'quiet': True,
             'no_warnings': True,
+            'extract_flat': True,
+            'no_check_certificates': True,
+            'force_generic_extractor': True
         }
         with yt_dlp.YoutubeDL(ydl_opts) as ydl:
             info = ydl.extract_info(url, download=False)
             duration = info.get('duration', 0)
-            if duration >= 300:
+            if duration >= 300:  # 5 minutes
                 st.error("Please find a YouTube video shorter than 5 minutes.")
                 st.stop()
             video_url = info['url']