Handle yt-dlp bot check

whichframe.py

@@ -1,13 +1,10 @@
 import streamlit as st
-from pytube import YouTube
-from pytube import extract
 import cv2
 from PIL import Image
 import clip as openai_clip
 import torch
 import math
 from humanfriendly import format_timespan
-from moviepy.video.io.VideoFileClip import VideoFileClip
 import numpy as np
 import time
 import os
@@ -22,15 +19,35 @@ 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]',
             'quiet': True,
-            'no_warnings': True
+            'no_warnings': True,
         }
         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
+            if duration >= 300:
                 st.error("Please find a YouTube video shorter than 5 minutes.")
                 st.stop()
             video_url = info['url']