clean 2

agents/video_analyzer_agent.py

@@ -22,6 +22,37 @@ from youtube_transcript_api import YouTubeTranscriptApi, TranscriptsDisabled, No
 logger = logging.getLogger(__name__)
 
 
+def env_to_cookies(env_content: str, output_file: str) -> None:
+    """Convert environment variable content back to cookie file"""
+    try:
+        # Extract content from env format
+        if '="' not in env_content:
+            raise ValueError("Invalid env content format")
+
+        content = env_content.split('="', 1)[1].strip('"')
+
+        # Replace escaped newlines with actual newlines
+        cookie_content = content.replace('\\n', '\n')
+
+        # Write to cookie file
+        with open(output_file, 'w') as f:
+            f.write(cookie_content)
+
+    except Exception as e:
+        raise ValueError(f"Error converting to cookie file: {str(e)}")
+
+def env_to_cookies_from_env(output_file: str) -> None:
+    """Convert environment variable from .env file to cookie file"""
+    try:
+        env_content = os.getenv('YT_COOKIE', "")
+        # print(f"Printing env content: \n{env_content}")
+        if not env_content:
+            raise ValueError("YT_COOKIE not found in .env file")
+
+        env_to_cookies(f'YT_COOKIE="{env_content}"', output_file)
+    except Exception as e:
+        raise ValueError(f"Error converting to cookie file: {str(e)}")
+
 # ---------------------------------------------------------------------------
 # Prompt loader
 # ---------------------------------------------------------------------------
@@ -106,17 +137,20 @@ def download_video_and_analyze(video_url: str) -> str:
     llm_model_name = os.getenv("VIDEO_ANALYZER_LLM_MODEL", "models/gemini-1.5-pro")
     gemini_api_key = os.getenv("GEMINI_API_KEY")
 
-    cookie_txt = "yt_cookie.txt"
-
     ydl_opts = {
         'format': 'best',
         'outtmpl': os.path.join("downloaded_videos", 'temp_video.%(ext)s'),
         'quiet': True,
         'extract_flat': True,
         'ignoreerrors': True,
-        "cookiefile": cookie_txt
     }
 
+    cookiefile = "cookies.txt"
+    env_to_cookies_from_env(cookiefile)
+
+    # Add cookies
+    ydl_opts["cookiefile"] = cookiefile  # create_temp_cookie_file()
+
     with yt_dlp.YoutubeDL(ydl_opts) as ydl_download:
         ydl_download.download(video_url)
 

get_cookie.py

@@ -34,3 +34,26 @@ def export_youtube_cookies_netscape(domain: str = "youtube.com") -> str:
 
     tmp.flush()
     return tmp.name
+
+
+def cookies_to_env(cookie_file_path: str) -> str:
+    """Convert cookie file content to environment variable format"""
+    try:
+        with open(cookie_file_path, 'r') as f:
+            lines = f.readlines()
+
+        # Keep header comments
+        header = [line.strip() for line in lines if line.startswith('#')]
+        # Get cookie content (non-comment lines)
+        cookies = [line.strip() for line in lines if line.strip() and not line.startswith('#')]
+
+        # Join with escaped newlines
+        content = '\\n'.join(header + [''] + cookies)  # Empty line after headers
+
+        # Create env file content
+        return f'FIREFOX_COOKIES="{content}"'
+
+    except Exception as e:
+        raise ValueError(f"Error converting cookie file: {str(e)}")
+
+print(cookies_to_env(export_youtube_cookies_netscape("youtube.com")))

yt_cookie.txt

@@ -1,16 +0,0 @@
-# Netscape HTTP Cookie File
-# This file is generated by yt-dlp.  Do not edit.
-
-.youtube.com	TRUE	/	TRUE	1761617545	__Secure-ROLLOUT_TOKEN	CK2CtoGE9_qCKBDVmd6PqaqLAxjs7cKEmYGNAw%3D%3D
-.youtube.com	TRUE	/	TRUE	1761622693	VISITOR_INFO1_LIVE	XJWWeg-61Jo
-.youtube.com	TRUE	/	TRUE	1780502573	__Secure-3PAPISID	Es8lNi2tKH_w6xhk/A5yqk-H0g0K329X1o
-.youtube.com	TRUE	/	TRUE	1780502573	__Secure-3PSID	g.a000wQiyYVADhJ0IDxSDoFRi6cUtMUs77vKiv67cKlhfBRyt8TWh6wP59ja7Ccu2lEPStG0E9gACgYKAT0SARASFQHGX2MirUvQtL-jYTLLnMocXq-eVxoVAUF8yKq2SkbS4FUiTUqswnomlXXS0076
-.youtube.com	TRUE	/	TRUE	1761622693	VISITOR_PRIVACY_METADATA	CgJCRRIhEh0SGwsMDg8QERITFBUWFxgZGhscHR4fICEiIyQlJiBt
-.youtube.com	TRUE	/	FALSE	0	PREF	f4=4000000&tz=UTC&f7=100&f6=40000400&f5=30000&hl=en
-.youtube.com	TRUE	/	TRUE	1777602578	__Secure-1PSIDTS	sidts-CjIBjplskKsM_03lNIkS3RrNBUeXvyspivYPoOliq4TT_dcTTR60_KSnwv-tWi7MPFIArRAA
-.youtube.com	TRUE	/	TRUE	1777602578	__Secure-3PSIDTS	sidts-CjIBjplskKsM_03lNIkS3RrNBUeXvyspivYPoOliq4TT_dcTTR60_KSnwv-tWi7MPFIArRAA
-.youtube.com	TRUE	/	TRUE	1777602578	__Secure-3PSIDCC	AKEyXzWAno5sygvBmiCVOYca7OcomSey2MkLYLei77boI7N8znStQNIMPKLzIesnRhDOs_tFdC4T
-.youtube.com	TRUE	/	TRUE	0	SOCS	CAI
-.youtube.com	TRUE	/	TRUE	1746072493	GPS	1
-.youtube.com	TRUE	/	TRUE	1809142693	__Secure-YT_TVFAS	t=483201&s=2
-.youtube.com	TRUE	/	TRUE	1761622693	DEVICE_INFO	ChxOelE1T1RNeE5qVXlOREF5TVRVeE1ERTJOdz09EKXZy8AGGKXZy8AG