Update app.py

app.py

@@ -18,7 +18,7 @@ class YouTubeDownloader:
         )
         return youtube_regex.match(url) is not None
     
-    def get_video_info(self, url, progress=gr.Progress()):
+    def get_video_info(self, url, progress=gr.Progress(),cookiefile=None)):
         """Get detailed video information without downloading"""
         if not url or not url.strip():
             return None, "❌ Please enter a YouTube URL"
@@ -34,6 +34,8 @@ class YouTubeDownloader:
                 'noplaylist': True,
                 'extract_flat': False,
             }
+            if cookiefile:
+                ydl_opts['cookiefile'] = cookiefile
             
             with yt_dlp.YoutubeDL(ydl_opts) as ydl:
                 try:
@@ -73,7 +75,7 @@ class YouTubeDownloader:
         except Exception as e:
             return None, f"❌ An unexpected error occurred: {str(e)}"
 
-    def download_video(self, url, progress=gr.Progress()):
+    def download_video(self, url, progress=gr.Progress(),cookiefile=None):
         """Download YouTube video using yt-dlp"""
         if not url or not url.strip():
             return None, "❌ Please enter a YouTube URL"
@@ -90,7 +92,9 @@ class YouTubeDownloader:
                 'outtmpl': os.path.join(self.download_dir, '%(title)s.%(ext)s'),
                 'noplaylist': True,  # Download only single video, not playlist
             }
-            
+            if cookiefile:
+                ydl_opts['cookiefile'] = cookiefile
+                
             progress(0.3, desc="Fetching video information...")
             
             with yt_dlp.YoutubeDL(ydl_opts) as ydl:
@@ -269,6 +273,13 @@ def create_interface():
             lines=1,
             max_lines=1
         )
+
+        cookies_input = gr.File(
+            label="Upload cookies.txt (optional for age-restricted/protected videos)",
+            type="filepath",
+            file_types=[".txt"],
+            visible=True
+        )
         
         download_btn = gr.Button(
             "Download Video",
@@ -307,12 +318,15 @@ def create_interface():
         def handle_download(url):
             if not url or not url.strip():
                 return "Please enter a YouTube URL", gr.File(visible=False), gr.Button(visible=False), None
+
+           # Set cookies path if provided
+        cookiefile = cookies_path if cookies_path and os.path.exists(cookies_path) else None 
             
             # First get video info and store it
-            video_info, info_message = downloader.get_video_info(url)
+            video_info, info_message = downloader.get_video_info(url,cookiefile=cookiefile)
             
             # Then download the video
-            file_path, download_message = downloader.download_video(url)
+            file_path, download_message = downloader.download_video(url,cookiefile=cookiefile)
             
             if file_path and video_info:
                 success_message = f"{download_message}\n\nVideo downloaded successfully! Click 'Show Analysis Results' to see detailed information."
@@ -347,7 +361,7 @@ def create_interface():
         # Button click events
         download_btn.click(
             fn=handle_download,
-            inputs=[url_input],
+            inputs=[url_input,cookies_input],
             outputs=[status_output, file_output, analysis_btn, video_info_state],
             show_progress="full"
         )
@@ -361,7 +375,7 @@ def create_interface():
         # Allow Enter key to trigger download
         url_input.submit(
             fn=handle_download,
-            inputs=[url_input],
+            inputs=[url_input,cookies_input],
             outputs=[status_output, file_output, analysis_btn, video_info_state],
             show_progress="full"
         )