added cookies

app.py

@@ -1,54 +1,64 @@
 import gradio as gr
 import yt_dlp
 import os
-import platform
-from pathlib import Path
-
-# Determine the default download directory based on the user's OS
-def get_default_download_path():
-    if platform.system() == 'Windows':
-        return str(Path.home() / "Downloads")  # Windows default Downloads folder
-    elif platform.system() == 'Darwin':
-        return str(Path.home() / "Downloads")  # macOS default Downloads folder
-    else:
-        return str(Path.home() / "Downloads")  # Linux default Downloads folder
-
-# Define the progress hook function that updates the progress bar
-def progress_hook(d, progress):
-    if d['status'] == 'downloading':
-        p = d['downloaded_bytes'] / d['total_bytes'] * 100
-        progress.update(p)  # Update Gradio progress bar with percentage
+import tempfile
 
-# Define the download function
-def download_video(url, progress=gr.Progress()):
+def download_video(url, cookies, progress=gr.Progress()):
     try:
-        download_dir = get_default_download_path()
-
-        # Set options for yt-dlp to get the best video/audio quality and save as .mp4
-        ydl_opts = {
-            'format': 'bestvideo+bestaudio/best',  # Best video and audio stream combination
-            'outtmpl': os.path.join(download_dir, '%(title)s.%(ext)s'),  # Save video in the Downloads folder
-            'merge_output_format': 'mp4',  # Force merge output to .mp4
-            'progress_hooks': [lambda d: progress_hook(d, progress)],  # Hook for progress updates
-        }
-
-        # Create the downloader with options
-        with yt_dlp.YoutubeDL(ydl_opts) as ydl:
-            info_dict = ydl.extract_info(url, download=True)
-            video_path = os.path.join(download_dir, f"{info_dict['title']}.mp4")
-            return video_path  # Return the file path for download
+        # Get the path to the cookies file if provided
+        cookies_path = cookies.get('name') if cookies else None
+
+        with tempfile.TemporaryDirectory() as temp_dir:
+            # Configure yt-dlp options
+            ydl_opts = {
+                'format': 'bestvideo+bestaudio/best',
+                'outtmpl': os.path.join(temp_dir, '%(title)s.%(ext)s'),
+                'merge_output_format': 'mp4',
+                'progress_hooks': [lambda d: progress_hook(d, progress)],
+                'cookies': cookies_path,
+                'verbose': True,
+            }
+
+            # Download video using yt-dlp
+            with yt_dlp.YoutubeDL(ydl_opts) as ydl:
+                info_dict = ydl.extract_info(url, download=True)
+
+            # Read downloaded video file into memory
+            video_file = info_dict.get('filename')
+            if not video_file:
+                return "Download Failed: Could not retrieve video file"
 
+            with open(video_file, 'rb') as f:
+                video_bytes = f.read()
+
+            # Return the video bytes for download
+            return {'name': os.path.basename(video_file), 'data': video_bytes}
+
+    except yt_dlp.utils.DownloadError as e:
+        return f"Download Failed: {str(e)}"
     except Exception as e:
-        return f"Error: {str(e)}"
+        return f"Unexpected Error: {str(e)}"
+
+def progress_hook(d, progress):
+    if d['status'] == 'downloading':
+        p = (d['downloaded_bytes'] / d['total_bytes']) * 100 if d['total_bytes'] else 0
+        progress.update(p, desc=f"Downloading ({d['speed'] or '...'} at {d['eta'] or '...'})")
 
-# Create the Gradio interface with a progress bar that updates during download
+# Gradio interface configuration
 iface = gr.Interface(
     fn=download_video,
-    inputs=gr.Textbox(label="Enter the YouTube URL"),
-    outputs=gr.File(label="Download Video"),
+    inputs=[
+        gr.Textbox(label="YouTube Video URL"),
+        gr.File(label="Upload your cookies.txt file (optional)", file_types=[".txt"], optional=True),
+    ],
+    outputs=gr.File(label="Downloaded Video", type="binary", file_count="single"),
     title="YouTube Video Downloader",
-    description="Enter the URL of the YouTube video you want to download."
+    description="Enter a YouTube video URL and upload your cookies.txt (if required) to download the video.",
+    examples=[
+        ["https://www.youtube.com/watch?v=dQw4w9WgXcQ", None],
+    ],
 )
 
-# Launch the interface with allowed_paths parameter to include Downloads directory
-iface.launch(allowed_paths=["/home/dq/Downloads"])  # Allow the Downloads directory
+# Launch the Gradio interface
+iface.queue()
+iface.launch(share=True)