Update app.py

app.py

@@ -1,154 +1,182 @@
 import gradio as gr
 import subprocess
 import os
-import tempfile
 
-# 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):
-    # 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 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)
+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."""
 
-# Gradio Interface
-with gr.Blocks() as demo:
-    gr.Markdown("# yt-dlp Command Executor")
+    # sanitize output_dir to prevent arbitrary code execution
+    output_dir = os.path.abspath(output_dir)
 
-    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")
+    # Create output directory if it doesn't exist
+    os.makedirs(output_dir, exist_ok=True)
 
-        url = gr.Textbox(label="URL", placeholder="Enter the URL here")
+    # Basic command structure
+    base_command = ["yt-dlp", "-o", f"{output_dir}/%(title)s.%(ext)s"]
 
-    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")
+    # Add archive file if provided
+    if archive_file:
+        base_command.extend(["--download-archive", archive_file])
 
-    execute_button = gr.Button("Execute Command")
-    file_output = gr.File(label="Downloaded File")
-    output = gr.Textbox(label="Output", lines=10)
+    # 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."
 
-    def execute_and_download(command, url, bitrate, start_time, end_time, playlist_items, archive_file):
-        file_path, result = run_yt_dlp(command, url, bitrate, start_time, end_time, playlist_items, archive_file)
-        if file_path:
-            return file_path, result
+    try:
+        # Execute command and capture output
+        process = subprocess.Popen(command, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
+        stdout, stderr = process.communicate()
+
+        # Decode output to string
+        stdout = stdout.decode("utf-8")
+        stderr = stderr.decode("utf-8")
+
+        if process.returncode != 0:
+            return f"Error executing command:\n{stderr}"
         else:
-            return None, result
+            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)
 
-    execute_button.click(
-        fn=execute_and_download,
-        inputs=[command, url, bitrate, start_time, end_time, playlist_items, archive_file],
-        outputs=[file_output, output]
+    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
     )
 
-demo.launch()
+if __name__ == "__main__":
+    demo.launch()