ADDED PROGRESS

.gitignore

@@ -0,0 +1 @@
+venv

app.py

@@ -1,31 +1,47 @@
 import gradio as gr
 import yt_dlp
 import os
+import platform
+from pathlib import Path
 
-def download_video(url):
+# 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
+
+# Define the download function
+def download_video(url, 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': 'downloads/%(title)s.%(ext)s',  # Save the video in the 'downloads' folder
+            '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 downloads directory if it doesn't exist
-        if not os.path.exists('downloads'):
-            os.makedirs('downloads')
-
+        # 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('downloads', f"{info_dict['title']}.mp4")
-
-            # Return the path of the downloaded video file for the user to download
-            return video_path
+            video_path = os.path.join(download_dir, f"{info_dict['title']}.mp4")
+            return video_path  # Return the file path for download
 
     except Exception as e:
         return f"Error: {str(e)}"
 
-# Create the Gradio interface
+# Create the Gradio interface with a progress bar that updates during download
 iface = gr.Interface(
     fn=download_video,
     inputs=gr.Textbox(label="Enter the YouTube URL"),
@@ -34,5 +50,5 @@ iface = gr.Interface(
     description="Enter the URL of the YouTube video you want to download."
 )
 
-# Launch the interface
-iface.launch()
+# Launch the interface with allowed_paths parameter to include Downloads directory
+iface.launch(allowed_paths=["/home/dq/Downloads"])  # Allow the Downloads directory