Handle yt-dlp bot check

whichframe.py

@@ -17,15 +17,26 @@ CACHED_DATA_PATH = "cached_data/"
 device = "cuda" if torch.cuda.is_available() else "cpu"
 model, preprocess = openai_clip.load("ViT-B/32", device=device)
 
+def export_cookies():
+    cookie_path = "youtube.com_cookies.txt"
+    if not os.path.exists(cookie_path):
+        with open(cookie_path, "w") as f:
+            f.write("# Netscape HTTP Cookie File\n")
+            f.write("# https://curl.haxx.se/rfc/cookie_spec.html\n")
+            f.write("# This is a generated file!  Do not edit.\n\n")
+            f.write(".youtube.com\tTRUE\t/\tFALSE\t2147483647\tCONSENT\tYES+cb.20210328-17-p0.en+FX+{}\n")
+    return cookie_path
+
 def fetch_video(url):
     try:
+        cookie_path = export_cookies()
         ydl_opts = {
             '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
+            'cookies': cookie_path,
+            'extract_flat': False,
+            'no_check_certificates': True
         }
         with yt_dlp.YoutubeDL(ydl_opts) as ydl:
             info = ydl.extract_info(url, download=False)