Update app.py

app.py

@@ -1,182 +1,158 @@
 import gradio as gr
 import subprocess
 import os
+import tempfile
 
-def run_ytdlp_command(command_type, url_or_id, quality, bitrate, start_time, end_time, playlist_range, output_dir="downloads", archive_file="", sub_lang="en", additional_urls="", speed_limit=""):
-    """Runs the yt-dlp command based on the selected command type and parameters."""
-
-    # sanitize output_dir to prevent arbitrary code execution
-    output_dir = os.path.abspath(output_dir)
+# Function to run yt-dlp commands and return the file path
+def run_yt_dlp(command, url, bitrate=None, start_time=None, end_time=None, playlist_items=None, archive_file=None, cookies_file=None):
+    # Temporary directory to store the downloaded files
+    temp_dir = tempfile.mkdtemp()
+    
+    # Base command for yt-dlp
+    base_command = ["yt-dlp", "-f", "bestaudio", "--extract-audio", "--audio-format", "mp3", "--audio-quality", "0", "-o", f"{temp_dir}/%(title)s.%(ext)s"]
+    
+    if cookies_file:
+        base_command.extend(["--cookies", cookies_file])
+    
+    if command == "list_formats":
+        base_command = ["yt-dlp", "-F", url]
+    elif command == "download_mp3_max_quality":
+        base_command.extend(["--postprocessor-args", "-b:a 320k", url])
+    elif command == "basic_high_quality_mp3":
+        base_command.append(url)
+    elif command == "custom_bitrate":
+        base_command.extend(["--postprocessor-args", f"-b:a {bitrate}k", url])
+    elif command == "add_metadata_tags":
+        base_command.extend(["--embed-metadata", url])
+    elif command == "download_playlist":
+        base_command.append(url)
+    elif command == "download_playlist_range":
+        base_command.extend(["--playlist-items", playlist_items, url])
+    elif command == "download_hq_soundcloud":
+        base_command.extend(["-o", f"{temp_dir}/%(uploader)s - %(title)s.%(ext)s", url])
+    elif command == "embed_album_art":
+        base_command.extend(["--embed-thumbnail", url])
+    elif command == "download_opus":
+        base_command.extend(["--audio-format", "opus", url])
+    elif command == "skip_downloaded_files":
+        base_command.extend(["--download-archive", archive_file, url])
+    elif command == "download_multiple_files":
+        base_command.extend(url.split())
+    elif command == "download_split_chapters":
+        base_command.extend(["--split-chapters", url])
+    elif command == "extract_portion":
+        base_command.extend(["-f", "bestvideo+bestaudio", "--external-downloader", "ffmpeg", "--external-downloader-args", f"ffmpeg_i:-ss {start_time} -to {end_time}", url])
+    elif command == "update_metadata":
+        base_command.extend(["--download-archive", archive_file, "--skip-download", "--embed-metadata", "--embed-thumbnail", url])
+    elif command == "auto_rename":
+        base_command.extend(["--force-overwrites", url])
+    elif command == "download_url_list":
+        base_command.extend(["-a", url])
+    elif command == "limit_speed":
+        base_command.extend(["--rate-limit", "1M", url])
+    elif command == "download_vimeo":
+        base_command.extend(["-o", f"{temp_dir}/Vimeo/%(uploader)s - %(title)s.%(ext)s", url])
+    elif command == "download_facebook":
+        base_command.extend(["-o", f"{temp_dir}/Facebook/%(uploader)s - %(title)s.%(ext)s", url])
+    elif command == "download_instagram":
+        base_command.extend(["-o", f"{temp_dir}/Instagram/%(uploader)s - %(title)s.%(ext)s", url])
+    elif command == "download_twitter":
+        base_command.extend(["-o", f"{temp_dir}/Twitter/%(uploader)s - %(title)s.%(ext)s", url])
+    elif command == "download_tiktok":
+        base_command.extend(["-o", f"{temp_dir}/TikTok/%(uploader)s - %(title)s.%(ext)s", url])
+    elif command == "download_reddit":
+        base_command.extend(["-o", f"{temp_dir}/Reddit/%(uploader)s - %(title)s.%(ext)s", url])
+    elif command == "download_dailymotion":
+        base_command.extend(["-o", f"{temp_dir}/Dailymotion/%(uploader)s - %(title)s.%(ext)s", url])
+    elif command == "download_transcript_text":
+        base_command.extend(["--write-auto-subs", "--sub-lang", "en", "--skip-download", url])
+    elif command == "download_transcript_markdown":
+        base_command.extend(["--write-auto-subs", "--sub-lang", "en", "--skip-download", "--convert-subs", "srt", "-o", f"{temp_dir}/%(title)s.md", url])
+    elif command == "download_video_info":
+        base_command.extend(["--write-info-json", "--skip-download", "-o", f"{temp_dir}/%(title)s.json", url])
+    elif command == "list_formats_no_download":
+        base_command = ["yt-dlp", "-F", url]
+    elif command == "download_age_restricted":
+        base_command.extend(["--username", "YOUR_YOUTUBE_EMAIL", "--password", "YOUR_PASSWORD", "-f", "bestvideo+bestaudio", url])
+    
+    try:
+        result = subprocess.run(base_command, capture_output=True, text=True)
+        if result.returncode != 0:
+            return None, result.stderr
+        else:
+            # Find the downloaded file
+            downloaded_files = [f for f in os.listdir(temp_dir) if os.path.isfile(os.path.join(temp_dir, f))]
+            if downloaded_files:
+                file_path = os.path.join(temp_dir, downloaded_files[0])
+                return file_path, result.stdout
+            else:
+                return None, "No file downloaded."
+    except Exception as e:
+        return None, str(e)
 
-    # Create output directory if it doesn't exist
-    os.makedirs(output_dir, exist_ok=True)
+# Gradio Interface
+with gr.Blocks() as demo:
+    gr.Markdown("# yt-dlp Command Executor")
 
-    # Basic command structure
-    base_command = ["yt-dlp", "-o", f"{output_dir}/%(title)s.%(ext)s"]
+    with gr.Row():
+        command = gr.Dropdown(choices=[
+            "list_formats",
+            "download_mp3_max_quality",
+            "basic_high_quality_mp3",
+            "custom_bitrate",
+            "add_metadata_tags",
+            "download_playlist",
+            "download_playlist_range",
+            "download_hq_soundcloud",
+            "embed_album_art",
+            "download_opus",
+            "skip_downloaded_files",
+            "download_multiple_files",
+            "download_split_chapters",
+            "extract_portion",
+            "update_metadata",
+            "auto_rename",
+            "download_url_list",
+            "limit_speed",
+            "download_vimeo",
+            "download_facebook",
+            "download_instagram",
+            "download_twitter",
+            "download_tiktok",
+            "download_reddit",
+            "download_dailymotion",
+            "download_transcript_text",
+            "download_transcript_markdown",
+            "download_video_info",
+            "list_formats_no_download",
+            "download_age_restricted"
+        ], label="Command")
 
-    # Add archive file if provided
-    if archive_file:
-        base_command.extend(["--download-archive", archive_file])
+        url = gr.Textbox(label="URL", placeholder="Enter the URL here")
 
-    # Construct command based on type
-    if command_type == "List Formats":
-        command = ["yt-dlp",  "-F", url_or_id]
-    elif command_type == "Download MP3 (Max Quality)":
-        command = base_command + ["-f", "bestaudio", "--extract-audio", "--audio-format", "mp3", "--audio-quality", "0", "--postprocessor-args", "-b:a 320k", url_or_id]
-    elif command_type == "Basic MP3 Download":
-        command = base_command + ["-f", "bestaudio", "--extract-audio", "--audio-format", "mp3", "--audio-quality", quality, url_or_id]
-    elif command_type == "Download with Custom Bitrate":
-        command = base_command + ["-f", "bestaudio", "--extract-audio", "--audio-format", "mp3", "--postprocessor-args", f"-b:a {bitrate}k", url_or_id]
-    elif command_type == "Download with Metadata":
-        command = base_command + ["-f", "bestaudio", "--extract-audio", "--audio-format", "mp3", "--audio-quality", "0", "--embed-metadata", url_or_id]
-    elif command_type == "Download Playlist":
-        command = base_command + ["-f", "bestaudio", "--extract-audio", "--audio-format", "mp3", "--audio-quality", "0", "-o", f"{output_dir}/%(playlist_title)s/%(title)s.%(ext)s", url_or_id]
-    elif command_type == "Download Playlist (Range)":
-        command = base_command + ["--playlist-items", playlist_range, "-f", "bestaudio", "--extract-audio", "--audio-format", "mp3", "--audio-quality", "0", "-o", f"{output_dir}/%(playlist_title)s/%(title)s.%(ext)s", url_or_id]
-    elif command_type == "Download with Album Art":
-        command = base_command + ["-f", "bestaudio", "--extract-audio", "--audio-format", "mp3", "--embed-thumbnail", "--audio-quality", "0", url_or_id]
-    elif command_type == "Download Opus":
-        command = base_command + ["-f", "bestaudio", "--extract-audio", "--audio-format", "opus", url_or_id]
-    elif command_type == "Skip Downloaded":
-        command = base_command + ["--download-archive", archive_file, "-f", "bestaudio", "--extract-audio", "--audio-format", "mp3", "--audio-quality", "0", url_or_id]
-    elif command_type == "Download Multiple URLs":
-        if additional_urls:
-            urls = [url_or_id] + additional_urls.split(",")
-            command = base_command + ["-f", "bestaudio", "--extract-audio", "--audio-format", "mp3", "--audio-quality", "0", "--add-metadata", "--embed-thumbnail", "--write-description", "--postprocessor-args", "-b:a 320k", *urls]
-        else:
-            return "Please provide additional URLs separated by commas."
-    elif command_type == "Download with Chapters":
-        command = base_command + ["-f", "bestaudio", "--extract-audio", "--audio-format", "mp3", "--audio-quality", "0", "--split-chapters", url_or_id]
-    elif command_type == "Extract Portion":
-        command = base_command + ["-f", "bestvideo+bestaudio", "--external-downloader", "ffmpeg", "--external-downloader-args", f"ffmpeg_i:-ss {start_time} -to {end_time}", url_or_id]
-    elif command_type == "Update Metadata":
-        command = ["yt-dlp", "--download-archive", archive_file, "--skip-download", "--embed-metadata", "--embed-thumbnail", "--write-description", url_or_id]
-    elif command_type == "Rename Duplicate":
-        command = base_command + ["-f", "bestaudio", "--extract-audio", "--audio-format", "mp3", "--audio-quality", "0", "--force-overwrites", url_or_id]
-    elif command_type == "Download from URL List":
-        command = ["yt-dlp", "-a", url_or_id, "-f", "bestaudio", "--extract-audio", "--audio-format", "mp3", "--audio-quality", "0", "-o", f"{output_dir}/%(title)s.%(ext)s"]
-    elif command_type == "Limit Download Speed":
-        command = base_command + ["-f", "bestaudio", "--extract-audio", "--audio-format", "mp3", "--audio-quality", "0", "--rate-limit", speed_limit, url_or_id]
-    elif command_type == "Download Transcript (Text)":
-        command = ["yt-dlp", "--write-auto-subs", "--sub-lang", sub_lang, "--skip-download", "-o", f"{output_dir}/%(title)s.%(ext)s", url_or_id]
-    elif command_type == "Download Transcript (Markdown)":
-        command = ["yt-dlp", "--write-auto-subs", "--sub-lang", sub_lang, "--skip-download", "--convert-subs", "srt", "-o", f"{output_dir}/%(title)s.md", url_or_id]
-    elif command_type == "Download Video Info (JSON)":
-        command = ["yt-dlp", "--write-info-json", "--skip-download", "-o", f"{output_dir}/%(title)s.json", url_or_id]
-     # SoundCloud Commands
-    elif command_type == "SoundCloud - Download MP3 (HQ)":
-        command = base_command + ["-f", "bestaudio", "--extract-audio", "--audio-format", "mp3", "--audio-quality", "0", "-o", f"{output_dir}/SoundCloud/%(uploader)s - %(title)s.%(ext)s", url_or_id]
-    elif command_type == "SoundCloud - Download with Metadata":
-        command = base_command + ["-f", "bestaudio", "--extract-audio", "--audio-format", "mp3", "--audio-quality", "0", "--embed-metadata", "-o", f"{output_dir}/SoundCloud/%(uploader)s - %(title)s.%(ext)s", url_or_id]
-    elif command_type == "SoundCloud - Download Playlist":
-        command = base_command + ["-f", "bestaudio", "--extract-audio", "--audio-format", "mp3", "--audio-quality", "0", "-o", f"{output_dir}/SoundCloud/%(playlist_title)s/%(title)s.%(ext)s", url_or_id]
-    elif command_type == "SoundCloud - Download Playlist (Range)":
-        command = base_command + ["--playlist-items", playlist_range, "-f", "bestaudio", "--extract-audio", "--audio-format", "mp3", "--audio-quality", "0", "-o", f"{output_dir}/SoundCloud/%(playlist_title)s/%(title)s.%(ext)s", url_or_id]
-    elif command_type == "SoundCloud - Download with Album Art":
-        command = base_command + ["-f", "bestaudio", "--extract-audio", "--audio-format", "mp3", "--audio-quality", "0", "--embed-thumbnail", "-o", f"{output_dir}/SoundCloud/%(uploader)s - %(title)s.%(ext)s", url_or_id]
-    elif command_type == "SoundCloud - Skip Downloaded":
-        command = base_command + ["--download-archive", archive_file, "-f", "bestaudio", "--extract-audio", "--audio-format", "mp3", "--audio-quality", "0", "-o", f"{output_dir}/SoundCloud/%(uploader)s - %(title)s.%(ext)s", url_or_id]
-    elif command_type == "SoundCloud - Download Multiple Tracks":
-        if additional_urls:
-            urls = [url_or_id] + additional_urls.split(",")
-            command = base_command + ["-f", "bestaudio", "--extract-audio", "--audio-format", "mp3", "--audio-quality", "0", "-o", f"{output_dir}/SoundCloud/%(uploader)s - %(title)s.%(ext)s", *urls]
-        else:
-            return "Please provide additional URLs separated by commas."
-    elif command_type == "SoundCloud - Limit Download Speed":
-        command = base_command + ["-f", "bestaudio", "--extract-audio", "--audio-format", "mp3", "--audio-quality", "0", "--rate-limit", speed_limit, "-o", f"{output_dir}/SoundCloud/%(uploader)s - %(title)s.%(ext)s", url_or_id]
-    elif command_type == "SoundCloud - Download with Chapters":
-        command = base_command + ["-f", "bestaudio", "--extract-audio", "--audio-format", "mp3", "--audio-quality", "0", "--split-chapters", "-o", f"{output_dir}/SoundCloud/%(uploader)s - %(title)s.%(ext)s", url_or_id]
-    elif command_type == "SoundCloud - Extract Portion":
-        command = base_command + ["-f", "bestaudio", "--external-downloader", "ffmpeg", "--external-downloader-args", f"ffmpeg_i:-ss {start_time} -to {end_time}", "-o", f"{output_dir}/SoundCloud/%(uploader)s - %(title)s.%(ext)s", url_or_id]
-    elif command_type == "SoundCloud - Update Metadata":
-        command = ["yt-dlp", "--download-archive", archive_file, "--skip-download", "--embed-metadata", "--embed-thumbnail", url_or_id]
-    elif command_type == "SoundCloud - Rename Duplicate":
-        command = base_command + ["-f", "bestaudio", "--extract-audio", "--audio-format", "mp3", "--audio-quality", "0", "--force-overwrites", "-o", f"{output_dir}/SoundCloud/%(uploader)s - %(title)s.%(ext)s", url_or_id]
-    elif command_type == "SoundCloud - Download from URL List":
-        command = ["yt-dlp", "-a", url_or_id, "-f", "bestaudio", "--extract-audio", "--audio-format", "mp3", "--audio-quality", "0", "-o", f"{output_dir}/SoundCloud/%(uploader)s - %(title)s.%(ext)s"]
-    else:
-        return "Invalid command type selected."
-
-    try:
-        # Execute command and capture output
-        process = subprocess.Popen(command, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
-        stdout, stderr = process.communicate()
+    with gr.Row():
+        bitrate = gr.Textbox(label="Bitrate (for custom bitrate)", placeholder="Enter bitrate in kbps, e.g., 192")
+        start_time = gr.Textbox(label="Start Time (HH:MM:SS)", placeholder="Enter start time, e.g., 00:01:00")
+        end_time = gr.Textbox(label="End Time (HH:MM:SS)", placeholder="Enter end time, e.g., 00:05:00")
+        playlist_items = gr.Textbox(label="Playlist Items (e.g., 1-5)", placeholder="Enter range of items, e.g., 1-5")
+        archive_file = gr.Textbox(label="Archive File", placeholder="Enter path to archive file, e.g., P:/Local Music/downloaded.txt")
+        cookies_file = gr.Textbox(label="Cookies File", placeholder="Enter path to cookies file, e.g., P:/cookies.txt")
 
-        # Decode output to string
-        stdout = stdout.decode("utf-8")
-        stderr = stderr.decode("utf-8")
+    execute_button = gr.Button("Execute Command")
+    file_output = gr.File(label="Downloaded File")
+    output = gr.Textbox(label="Output", lines=10)
 
-        if process.returncode != 0:
-            return f"Error executing command:\n{stderr}"
+    def execute_and_download(command, url, bitrate, start_time, end_time, playlist_items, archive_file, cookies_file):
+        file_path, result = run_yt_dlp(command, url, bitrate, start_time, end_time, playlist_items, archive_file, cookies_file)
+        if file_path:
+            return file_path, result
         else:
-            return f"Command executed successfully:\n{stdout}"
-    except Exception as e:
-        return f"An error occurred: {e}"
-
-# Define Gradio interface
-with gr.Blocks() as demo:
-    gr.Markdown("# YT-DLP Gradio Interface")
-    with gr.Row():
-        with gr.Column():
-            command_type = gr.Dropdown(
-                choices=[
-                    "List Formats",
-                    "Download MP3 (Max Quality)",
-                    "Basic MP3 Download",
-                    "Download with Custom Bitrate",
-                    "Download with Metadata",
-                    "Download Playlist",
-                    "Download Playlist (Range)",
-                    "Download with Album Art",
-                    "Download Opus",
-                    "Skip Downloaded",
-                    "Download Multiple URLs",
-                    "Download with Chapters",
-                    "Extract Portion",
-                    "Update Metadata",
-                    "Rename Duplicate",
-                    "Download from URL List",
-                    "Limit Download Speed",
-                    "Download Transcript (Text)",
-                    "Download Transcript (Markdown)",
-                    "Download Video Info (JSON)",
-                    "SoundCloud - Download MP3 (HQ)",
-                    "SoundCloud - Download with Metadata",
-                    "SoundCloud - Download Playlist",
-                    "SoundCloud - Download Playlist (Range)",
-                    "SoundCloud - Download with Album Art",
-                    "SoundCloud - Skip Downloaded",
-                    "SoundCloud - Download Multiple Tracks",
-                    "SoundCloud - Limit Download Speed",
-                    "SoundCloud - Download with Chapters",
-                    "SoundCloud - Extract Portion",
-                    "SoundCloud - Update Metadata",
-                    "SoundCloud - Rename Duplicate",
-                    "SoundCloud - Download from URL List"
-                ],
-                value="Download MP3 (Max Quality)",
-                label="Command Type"
-            )
-            url_or_id = gr.Textbox(label="URL or Video/Playlist ID")
-            quality = gr.Textbox(label="Quality (e.g., 0 for best)", value="0")
-            bitrate = gr.Textbox(label="Bitrate (e.g., 192 for 192k)", value="320")
-            start_time = gr.Textbox(label="Start Time (HH:MM:SS)", value="00:00:00")
-            end_time = gr.Textbox(label="End Time (HH:MM:SS)", value="00:10:00")
-            playlist_range = gr.Textbox(label="Playlist Range (e.g., 1-5)", value="")
-            output_dir = gr.Textbox(label="Output Directory", value="downloads")
-            archive_file = gr.Textbox(label="Archive File Path", value="")
-            sub_lang = gr.Textbox(label="Subtitle Language (e.g., en, pt)", value="en")
-            additional_urls = gr.Textbox(label="Additional URLs (comma-separated)", value="")
-            speed_limit = gr.Textbox(label="Download Speed Limit (e.g., 1M, 512K)", value="")
-            submit_btn = gr.Button("Run Command")
-        with gr.Column():
-            output_text = gr.Textbox(label="Output", lines=10)
+            return None, result
 
-    submit_btn.click(
-        run_ytdlp_command,
-        inputs=[command_type, url_or_id, quality, bitrate, start_time, end_time, playlist_range, output_dir, archive_file, sub_lang, additional_urls, speed_limit],
-        outputs=output_text
+    execute_button.click(
+        fn=execute_and_download,
+        inputs=[command, url, bitrate, start_time, end_time, playlist_items, archive_file, cookies_file],
+        outputs=[file_output, output]
     )
 
-if __name__ == "__main__":
-    demo.launch()
+demo.launch()