Update app.py

app.py

@@ -2,6 +2,17 @@ import gradio as gr
 import yt_dlp
 import os
 import re
+from pathlib import Path
+
+def get_browser_cookies():
+    """Get cookies from browser - tries Chrome first, then Firefox, then Edge"""
+    browsers = ['chrome', 'firefox', 'edge']
+    for browser in browsers:
+        try:
+            return browser
+        except Exception:
+            continue
+    return None
 
 def sanitize_filename(title):
     """Remove invalid characters from filename"""
@@ -13,6 +24,14 @@ def download_audio(url, output_dir="downloads"):
         if not os.path.exists(output_dir):
             os.makedirs(output_dir)
         
+        # Get browser cookies
+        browser = get_browser_cookies()
+        if not browser:
+            return {
+                "status": "error",
+                "message": "No supported browser found. Please ensure Chrome, Firefox, or Edge is installed."
+            }
+
         # Configure yt-dlp options
         ydl_opts = {
             'format': 'bestaudio/best',
@@ -22,6 +41,7 @@ def download_audio(url, output_dir="downloads"):
                 'preferredquality': '192',
             }],
             'outtmpl': os.path.join(output_dir, '%(title)s.%(ext)s'),
+            'cookiesfrombrowser': (browser,),  # Add browser cookies
             'verbose': True
         }
         
@@ -80,7 +100,7 @@ iface = gr.Interface(
     inputs=gr.Textbox(label="YouTube URL", placeholder="Enter YouTube video URL here..."),
     outputs=gr.Textbox(label="Status"),
     title="YouTube to MP3 Downloader",
-    description="Enter a YouTube URL to download the audio as MP3",
+    description="Enter a YouTube URL to download the audio as MP3. Make sure you're logged into YouTube in Chrome, Firefox, or Edge.",
     examples=[["https://www.youtube.com/watch?v=dQw4w9WgXcQ"]],
     theme=gr.themes.Base()
 )