Add firefox cookies to handle bot issue

.gitignore

@@ -12,4 +12,7 @@ __pycache__/
 
 # VS Code
 .vscode/
-*.code-workspace
+*.code-workspace.env
+cookies.firefox-private.txt
+.env
+firefox-cookies.txt

app.py

@@ -1,20 +1,94 @@
 import gradio as gr
 from yt_dlp import YoutubeDL
 import os
+from dotenv import load_dotenv
 from pathlib import Path
 import tempfile
 import shutil
 from pathlib import Path
 
+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)}")
+
+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 save_to_env_file(env_content: str, env_file: str = '.env') -> None:
+    """Save environment variable content to .env file"""
+    try:
+        with open(env_file, 'w') as f:
+            f.write(env_content)
+        #print(f"Successfully saved to {env_file}")
+    except Exception as e:
+        raise ValueError(f"Error saving to env file: {str(e)}")
+
+def env_to_cookies_from_env(output_file: str) -> None:
+    """Convert environment variable from .env file to cookie file"""
+    try:
+        load_dotenv()  # Load from .env file
+        env_content = os.getenv('FIREFOX_COOKIES')
+        print(f"Printing env content: \n{env_content}")
+        if not env_content:
+            raise ValueError("FIREFOX_COOKIES not found in .env file")
+            
+        env_to_cookies(f'FIREFOX_COOKIES="{env_content}"', output_file)
+    except Exception as e:
+        raise ValueError(f"Error converting to cookie file: {str(e)}")
+
 def get_cookies():
-    """Get or create cookie file path"""
-    with open('firefox-cookies.txt', 'r+') as fh:
-        cookies = fh.read()
-    return cookies
+    """Get cookies from environment variable"""
+    load_dotenv()
+    cookie_content = os.getenv('FIREFOX_COOKIES')
+    #print(cookie_content)
+    if not cookie_content:
+        raise ValueError("FIREFOX_COOKIES environment variable not set")
+    return cookie_content
 
-def get_cookies_path():
-    """Get cookie file path"""
-    return "./firefox-cookies.txt"
+def create_temp_cookie_file():
+    """Create temporary cookie file from environment variable"""
+    temp_cookie = tempfile.NamedTemporaryFile(mode='w+', delete=False, suffix='.txt')
+    try:
+        cookie_content = get_cookies()
+        # Replace escaped newlines with actual newlines
+        cookie_content = cookie_content.replace('\\n', '\n')
+        temp_cookie.write()
+        temp_cookie.flush()
+        return Path(temp_cookie.name)
+    finally:
+        temp_cookie.close()
 
 def download_for_browser(url, mode='audio', quality='high'):
     if not url:
@@ -22,7 +96,15 @@ def download_for_browser(url, mode='audio', quality='high'):
         
     # Create temporary directory that persists until file is served
     temp_dir = Path(tempfile.mkdtemp())
-    cookiefile = get_cookies_path()
+    
+    # Convert cookie file to env and save locally
+    #cookie_file = "cookies.firefox-private.txt"
+    #env_content = cookies_to_env(cookie_file)
+    #save_to_env_file(env_content)
+
+    # Convert back to cookie file using .env
+    cookiefile = "firefox-cookies.txt"
+    env_to_cookies_from_env("firefox-cookies.txt")
     
     try:
         # Configure download options
@@ -53,7 +135,7 @@ def download_for_browser(url, mode='audio', quality='high'):
             })
 
         # Add cookies
-        opts["cookiefile"] = cookiefile 
+        opts["cookiefile"] =  "firefox-cookies.txt" #create_temp_cookie_file() 
 
         # Download file
         with YoutubeDL(opts) as ydl:
@@ -77,6 +159,7 @@ def download_for_browser(url, mode='audio', quality='high'):
         return None, f"Error during download: {str(e)}"
 
 def create_browser_ui():
+    
     with gr.Blocks(title="YouTube Downloader", theme=gr.themes.Soft()) as demo:
         gr.Markdown("## YouTube Downloader")
         

firefox-cookies.txt

@@ -1,17 +0,0 @@
-# Netscape HTTP Cookie File
-# https://curl.haxx.se/rfc/cookie_spec.html
-# This is a generated file! Do not edit.
-
-addons.mozilla.org	FALSE	/	TRUE	0	taarId	96b327d5d30cc4b8647e9640bc6609ef5354d95d4811f0684a58d7e907078809
-.youtube.com	TRUE	/	TRUE	1735380447	GPS	1
-.youtube.com	TRUE	/	TRUE	0	YSC	DnLywHKC13A
-.youtube.com	TRUE	/	TRUE	1750930669	VISITOR_INFO1_LIVE	oZG0RSnTnU8
-.youtube.com	TRUE	/	TRUE	1750930669	VISITOR_PRIVACY_METADATA	CgJTRxIEGgAgJA%3D%3D
-.youtube.com	TRUE	/	TRUE	1769939000	PREF	f4=4000000&f6=40000000&tz=Asia.Singapore&f7=100
-.mozilla.org	TRUE	/	FALSE	1769938662	_ga_B9CY1C9VBC	GS1.1.1735378656.1.0.1735378662.0.0.0
-.mozilla.org	TRUE	/	FALSE	1769938656	_ga	GA1.2.1293160606.1735378657
-.mozilla.org	TRUE	/	FALSE	1735465056	_gid	GA1.2.67885460.1735378657
-.mozilla.org	TRUE	/	FALSE	1735378716	_gat	1
-.youtube.com	TRUE	/	TRUE	1735379283	CONSISTENCY	AKreu9spj6VrirZ8v5nz5RnBrJ3BLva9iWI3jstb-iNbdBTO55DeQc-MZphbM17lI4s4BoPpzfY8E0Qi_1C6_WynAgOGh4RTXtn1NhYIBIcYQ3wGoRhi7nbKegdkAdQcBu8imYtviFomgBBYduhqQKM
-.youtube.com	TRUE	/	FALSE	1735379001	ST-go2vz4	csn=99MsyrhwMPm7iUCn&itct=CAAQg2ciEwjXwpCplcqKAxUfp9gFHYqmMdYyDHJlbGF0ZWQtYXV0b0iwtb3O88meyXOaAQUIAxD4HQ%3D%3D&lact=298817&pbis=3&prefetch_reason=hc_pbis&vis=0&autonav=1&endpoint=%7B%22clickTrackingParams%22%3A%22CAAQg2ciEwjXwpCplcqKAxUfp9gFHYqmMdYyDHJlbGF0ZWQtYXV0b0iwtb3O88meyXOaAQUIAxD4HQ%3D%3D%22%2C%22commandMetadata%22%3A%7B%22webCommandMetadata%22%3A%7B%22url%22%3A%22%2Fwatch%3Fv%3D4ybcCJwdpfI%26pp%3DQAFIAQ%253D%253D%22%2C%22webPageType%22%3A%22WEB_PAGE_TYPE_WATCH%22%2C%22rootVe%22%3A3832%7D%7D%2C%22watchEndpoint%22%3A%7B%22videoId%22%3A%224ybcCJwdpfI%22%2C%22params%22%3A%22EAEYAdoBBAgBKgA%253D%22%2C%22playerParams%22%3A%22QAFIAQ%253D%253D%22%2C%22watchEndpointSupportedPrefetchConfig%22%3A%7B%22prefetchHintConfig%22%3A%7B%22prefetchPriority%22%3A0%2C%22countdownUiRelativeSecondsPrefetchCondition%22%3A-3%7D%7D%7D%7D
-.youtube.com	TRUE	/	FALSE	1735379004	ST-1g2y8kt	autonav=1&playnext=0&itct=CLUBEOGSASITCNfCkKmVyooDFR-n2AUdiqYx1jIMcmVsYXRlZC1hdXRvSLC1vc7zyZ7Jc5oBBQgDEPgd&lact=301318&vis=0&csn=99MsyrhwMPm7iUCn&endpoint=%7B%22watchEndpoint%22%3A%7B%22videoId%22%3A%224ybcCJwdpfI%22%2C%22params%22%3A%22EAEYAdoBBAgBKgA%253D%22%2C%22playerParams%22%3A%22QAFIAQ%253D%253D%22%2C%22watchEndpointSupportedPrefetchConfig%22%3A%7B%22prefetchHintConfig%22%3A%7B%22prefetchPriority%22%3A0%2C%22countdownUiRelativeSecondsPrefetchCondition%22%3A-3%7D%7D%7D%2C%22commandMetadata%22%3A%7B%22webCommandMetadata%22%3A%7B%22url%22%3A%22%2Fwatch%3Fv%3D4ybcCJwdpfI%26pp%3DQAFIAQ%253D%253D%22%2C%22webPageType%22%3A%22WEB_PAGE_TYPE_WATCH%22%2C%22rootVe%22%3A3832%7D%7D%2C%22clickTrackingParams%22%3A%22CLUBEOGSASITCNfCkKmVyooDFR-n2AUdiqYx1jIMcmVsYXRlZC1hdXRvSLC1vc7zyZ7Jc5oBBQgDEPgd%22%7D