Update app.py

app.py

@@ -1,133 +1,394 @@
 import gradio as gr
 import subprocess
-import os
-
-# Function to run yt-dlp commands
-def run_yt_dlp(command, url, output_path, bitrate=None, start_time=None, end_time=None, playlist_items=None, archive_file=None):
-    base_command = ["yt-dlp", "-f", "bestaudio", "--extract-audio", "--audio-format", "mp3", "--audio-quality", "0", "-o", output_path]
-    
-    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", "P:/Local Music/SoundCloud/%(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", "P:/Local Music/Vimeo/%(uploader)s - %(title)s.%(ext)s", url])
-    elif command == "download_facebook":
-        base_command.extend(["-o", "P:/Local Music/Facebook/%(uploader)s - %(title)s.%(ext)s", url])
-    elif command == "download_instagram":
-        base_command.extend(["-o", "P:/Local Music/Instagram/%(uploader)s - %(title)s.%(ext)s", url])
-    elif command == "download_twitter":
-        base_command.extend(["-o", "P:/Local Music/Twitter/%(uploader)s - %(title)s.%(ext)s", url])
-    elif command == "download_tiktok":
-        base_command.extend(["-o", "P:/Local Music/TikTok/%(uploader)s - %(title)s.%(ext)s", url])
-    elif command == "download_reddit":
-        base_command.extend(["-o", "P:/Local Music/Reddit/%(uploader)s - %(title)s.%(ext)s", url])
-    elif command == "download_dailymotion":
-        base_command.extend(["-o", "P:/Local Music/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", "P:/Local Music/%(title)s.md", url])
-    elif command == "download_video_info":
-        base_command.extend(["--write-info-json", "--skip-download", "-o", "P:/Local Music/%(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])
-    
+
+def run_yt_dlp(command, url):
     try:
-        result = subprocess.run(base_command, capture_output=True, text=True)
+        result = subprocess.run(command.split() + [url], capture_output=True, text=True)
         return result.stdout
     except Exception as e:
         return str(e)
 
-# Gradio Interface
-with gr.Blocks() as demo:
-    gr.Markdown("# yt-dlp Command Executor")
-
-    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")
-
-        url = gr.Textbox(label="URL", placeholder="Enter the URL here")
-
-    with gr.Row():
-        output_path = gr.Textbox(label="Output Path", placeholder="Enter the output path here")
-        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")
-
-    execute_button = gr.Button("Execute Command")
-    output = gr.Textbox(label="Output", lines=10)
-
-    execute_button.click(
-        fn=run_yt_dlp,
-        inputs=[command, url, output_path, bitrate, start_time, end_time, playlist_items, archive_file],
-        outputs=output
-    )
-
-demo.launch()
+def download_mp3(url, quality, embed_metadata, embed_thumbnail, split_chapters, bitrate):
+    command = [
+        "yt-dlp",
+        "-f", "bestaudio",
+        "--extract-audio",
+        "--audio-format", "mp3",
+        "--audio-quality", quality,
+        "-o", "P:/Local Music/%(title)s.%(ext)s"
+    ]
+    if embed_metadata:
+        command.append("--embed-metadata")
+    if embed_thumbnail:
+        command.append("--embed-thumbnail")
+    if split_chapters:
+        command.append("--split-chapters")
+    if bitrate:
+        command.extend(["--postprocessor-args", f"-b:a {bitrate}k"])
+    return run_yt_dlp(" ".join(command), url)
+
+def list_formats(url):
+    return run_yt_dlp("yt-dlp -F", url)
+
+def download_playlist(url, start, end):
+    command = [
+        "yt-dlp",
+        "--playlist-items", f"{start}-{end}",
+        "-f", "bestaudio",
+        "--extract-audio",
+        "--audio-format", "mp3",
+        "--audio-quality", "0",
+        "-o", "P:/Local Music/%(playlist_title)s/%(title)s.%(ext)s"
+    ]
+    return run_yt_dlp(" ".join(command), url)
+
+def download_from_platform(url, platform):
+    command = [
+        "yt-dlp",
+        "-f", "bestaudio",
+        "--extract-audio",
+        "--audio-format", "mp3",
+        "--audio-quality", "0",
+        "-o", f"P:/Local Music/{platform}/%(uploader)s - %(title)s.%(ext)s"
+    ]
+    return run_yt_dlp(" ".join(command), url)
+
+def download_transcript(url, format):
+    command = [
+        "yt-dlp",
+        "--write-auto-subs",
+        "--sub-lang", "en",
+        "--skip-download",
+        "-o", f"P:/Local Music/%(title)s.{format}"
+    ]
+    return run_yt_dlp(" ".join(command), url)
+
+def download_video_info(url):
+    command = [
+        "yt-dlp",
+        "--write-info-json",
+        "--skip-download",
+        "-o", "P:/Local Music/%(title)s.json"
+    ]
+    return run_yt_dlp(" ".join(command), url)
+
+def download_age_restricted(url, username, password):
+    command = [
+        "yt-dlp",
+        "--username", username,
+        "--password", password,
+        "-f", "bestvideo+bestaudio",
+        "-o", "P:/Local Music/%(title)s.%(ext)s"
+    ]
+    return run_yt_dlp(" ".join(command), url)
+
+def download_from_url_list(file_path):
+    command = [
+        "yt-dlp",
+        "-a", file_path,
+        "-f", "bestaudio",
+        "--extract-audio",
+        "--audio-format", "mp3",
+        "--audio-quality", "0",
+        "-o", "P:/Local Music/%(title)s.%(ext)s"
+    ]
+    return run_yt_dlp(" ".join(command), "")
+
+def download_with_speed_limit(url, speed_limit):
+    command = [
+        "yt-dlp",
+        "-f", "bestaudio",
+        "--extract-audio",
+        "--audio-format", "mp3",
+        "--audio-quality", "0",
+        "--rate-limit", f"{speed_limit}M",
+        "-o", "P:/Local Music/%(title)s.%(ext)s"
+    ]
+    return run_yt_dlp(" ".join(command), url)
+
+def download_specific_portion(url, start_time, end_time):
+    command = [
+        "yt-dlp",
+        "-f", "bestvideo+bestaudio",
+        "--external-downloader", "ffmpeg",
+        "--external-downloader-args", f"ffmpeg_i:-ss {start_time} -to {end_time}",
+        "-o", "P:/Local Music/%(title)s.%(ext)s"
+    ]
+    return run_yt_dlp(" ".join(command), url)
+
+def update_metadata(url):
+    command = [
+        "yt-dlp",
+        "--download-archive", "P:/Local Music/downloaded.txt",
+        "--skip-download",
+        "--embed-metadata",
+        "--embed-thumbnail",
+        "--write-description"
+    ]
+    return run_yt_dlp(" ".join(command), url)
+
+def auto_rename_duplicates(url):
+    command = [
+        "yt-dlp",
+        "-f", "bestaudio",
+        "--extract-audio",
+        "--audio-format", "mp3",
+        "--audio-quality", "0",
+        "--force-overwrites",
+        "-o", "P:/Local Music/%(title)s.%(ext)s"
+    ]
+    return run_yt_dlp(" ".join(command), url)
+
+def download_multiple_files(urls):
+    command = [
+        "yt-dlp",
+        "-f", "bestaudio",
+        "--extract-audio",
+        "--audio-format", "mp3",
+        "--audio-quality", "0",
+        "--add-metadata",
+        "--embed-thumbnail",
+        "--write-description",
+        "--postprocessor-args", "-b:a 320k",
+        "-o", "P:/Local Music/%(title)s.%(ext)s"
+    ]
+    return run_yt_dlp(" ".join(command), " ".join(urls))
+
+def download_opus(url):
+    command = [
+        "yt-dlp",
+        "-f", "bestaudio",
+        "--extract-audio",
+        "--audio-format", "opus",
+        "-o", "P:/Local Music/%(title)s.%(ext)s"
+    ]
+    return run_yt_dlp(" ".join(command), url)
+
+def download_soundcloud(url, quality, embed_metadata, embed_thumbnail, split_chapters, bitrate):
+    command = [
+        "yt-dlp",
+        "-f", "bestaudio",
+        "--extract-audio",
+        "--audio-format", "mp3",
+        "--audio-quality", quality,
+        "-o", "P:/Local Music/SoundCloud/%(uploader)s - %(title)s.%(ext)s"
+    ]
+    if embed_metadata:
+        command.append("--embed-metadata")
+    if embed_thumbnail:
+        command.append("--embed-thumbnail")
+    if split_chapters:
+        command.append("--split-chapters")
+    if bitrate:
+        command.extend(["--postprocessor-args", f"-b:a {bitrate}k"])
+    return run_yt_dlp(" ".join(command), url)
+
+def download_soundcloud_playlist(url, start, end):
+    command = [
+        "yt-dlp",
+        "--playlist-items", f"{start}-{end}",
+        "-f", "bestaudio",
+        "--extract-audio",
+        "--audio-format", "mp3",
+        "--audio-quality", "0",
+        "-o", "P:/Local Music/SoundCloud/%(playlist_title)s/%(title)s.%(ext)s"
+    ]
+    return run_yt_dlp(" ".join(command), url)
+
+def download_soundcloud_from_url_list(file_path):
+    command = [
+        "yt-dlp",
+        "-a", file_path,
+        "-f", "bestaudio",
+        "--extract-audio",
+        "--audio-format", "mp3",
+        "--audio-quality", "0",
+        "-o", "P:/Local Music/SoundCloud/%(uploader)s - %(title)s.%(ext)s"
+    ]
+    return run_yt_dlp(" ".join(command), "")
+
+def download_soundcloud_with_speed_limit(url, speed_limit):
+    command = [
+        "yt-dlp",
+        "-f", "bestaudio",
+        "--extract-audio",
+        "--audio-format", "mp3",
+        "--audio-quality", "0",
+        "--rate-limit", f"{speed_limit}M",
+        "-o", "P:/Local Music/SoundCloud/%(uploader)s - %(title)s.%(ext)s"
+    ]
+    return run_yt_dlp(" ".join(command), url)
+
+def download_soundcloud_specific_portion(url, start_time, end_time):
+    command = [
+        "yt-dlp",
+        "-f", "bestaudio",
+        "--external-downloader", "ffmpeg",
+        "--external-downloader-args", f"ffmpeg_i:-ss {start_time} -to {end_time}",
+        "-o", "P:/Local Music/SoundCloud/%(uploader)s - %(title)s.%(ext)s"
+    ]
+    return run_yt_dlp(" ".join(command), url)
+
+def update_soundcloud_metadata(url):
+    command = [
+        "yt-dlp",
+        "--download-archive", "P:/Local Music/downloaded_soundcloud.txt",
+        "--skip-download",
+        "--embed-metadata",
+        "--embed-thumbnail"
+    ]
+    return run_yt_dlp(" ".join(command), url)
+
+def auto_rename_soundcloud_duplicates(url):
+    command = [
+        "yt-dlp",
+        "-f", "bestaudio",
+        "--extract-audio",
+        "--audio-format", "mp3",
+        "--audio-quality", "0",
+        "--force-overwrites",
+        "-o", "P:/Local Music/SoundCloud/%(uploader)s - %(title)s.%(ext)s"
+    ]
+    return run_yt_dlp(" ".join(command), url)
+
+def download_soundcloud_multiple_files(urls):
+    command = [
+        "yt-dlp",
+        "-f", "bestaudio",
+        "--extract-audio",
+        "--audio-format", "mp3",
+        "--audio-quality", "0",
+        "-o", "P:/Local Music/SoundCloud/%(uploader)s - %(title)s.%(ext)s"
+    ]
+    return run_yt_dlp(" ".join(command), " ".join(urls))
+
+def download_soundcloud_opus(url):
+    command = [
+        "yt-dlp",
+        "-f", "bestaudio",
+        "--extract-audio",
+        "--audio-format", "opus",
+        "-o", "P:/Local Music/SoundCloud/%(uploader)s - %(title)s.%(ext)s"
+    ]
+    return run_yt_dlp(" ".join(command), url)
+
+def download_soundcloud_playlist_with_metadata(url):
+    command = [
+        "yt-dlp",
+        "-f", "bestaudio",
+        "--extract-audio",
+        "--audio-format", "mp3",
+        "--audio-quality", "0",
+        "--embed-metadata",
+        "--add-metadata",
+        "--embed-thumbnail",
+        "-o", "P:/Local Music/%(playlist_title)s/%(title)s.%(ext)s"
+    ]
+    return run_yt_dlp(" ".join(command), url)
+
+def download_soundcloud_multiple_playlists_with_metadata(urls):
+    command = [
+        "yt-dlp",
+        "-f", "bestaudio",
+        "--extract-audio",
+        "--audio-format", "mp3",
+        "--audio-quality", "0",
+        "--embed-metadata",
+        "--add-metadata",
+        "--embed-thumbnail",
+        "-o", "P:/Local Music/SoundCloud/%(playlist_title)s/%(title)s.%(ext)s"
+    ]
+    return run_yt_dlp(" ".join(command), " ".join(urls))
+
+def download_soundcloud_playlist_with_specific_range(url, start, end):
+    command = [
+        "yt-dlp",
+        "--playlist-items", f"{start}-{end}",
+        "-f", "bestaudio",
+        "--extract-audio",
+        "--audio-format", "mp3",
+        "--audio-quality", "0",
+        "-o", "P:/Local Music/SoundCloud/%(playlist_title)s/%(title)s.%(ext)s"
+    ]
+    return run_yt_dlp(" ".join(command), url)
+
+def download_soundcloud_and_embed_album_art(url):
+    command = [
+        "yt-dlp",
+        "-f", "bestaudio",
+        "--extract-audio",
+        "--audio-format", "mp3",
+        "--audio-quality", "0",
+        "--embed-thumbnail",
+        "-o", "P:/Local Music/SoundCloud/%(uploader)s - %(title)s.%(ext)s"
+    ]
+    return run_yt_dlp(" ".join(command), url)
+
+def download_soundcloud_skip_already_downloaded(url):
+    command = [
+        "yt-dlp",
+        "--download-archive", "P:/Local Music/downloaded_soundcloud.txt",
+        "-f", "bestaudio",
+        "--extract-audio",
+        "--audio-format", "mp3",
+        "--audio-quality", "0",
+        "-o", "P:/Local Music/SoundCloud/%(uploader)s - %(title)s.%(ext)s"
+    ]
+    return run_yt_dlp(" ".join(command), url)
+
+def download_soundcloud_multiple_tracks(urls):
+    command = [
+        "yt-dlp",
+        "-f", "bestaudio",
+        "--extract-audio",
+        "--audio-format", "mp3",
+        "--audio-quality", "0",
+        "-o", "P:/Local Music/SoundCloud/%(uploader)s - %(title)s.%(ext)s"
+    ]
+    return run_yt_dlp(" ".join(command), " ".join(urls))
+
+def download_soundcloud_and_split_by_chapters(url):
+    command = [
+        "yt-dlp",
+        "-f", "bestaudio",
+        "--extract-audio",
+        "--audio-format", "mp3",
+        "--audio-quality", "0",
+        "--split-chapters",
+        "-o", "P:/Local Music/SoundCloud/%(uploader)s - %(title)s.%(ext)s"
+    ]
+    return run_yt_dlp(" ".join(command), url)
+
+def download_soundcloud_track_and_extract_specific_portion(url, start_time, end_time):
+    command = [
+        "yt-dlp",
+        "-f", "bestaudio",
+        "--external-downloader", "ffmpeg",
+        "--external-downloader-args", f"ffmpeg_i:-ss {start_time} -to {end_time}",
+        "-o", "P:/Local Music/SoundCloud/%(uploader)s - %(title)s.%(ext)s"
+    ]
+    return run_yt_dlp(" ".join(command), url)
+
+def update_soundcloud_metadata_of_existing_files(url):
+    command = [
+        "yt-dlp",
+        "--download-archive", "P:/Local Music/downloaded_soundcloud.txt",
+        "--skip-download",
+        "--embed-metadata",
+        "--embed-thumbnail"
+    ]
+    return run_yt_dlp(" ".join(command), url)
+
+def auto_rename_soundcloud_duplicate_files(url):
+    command = [
+        "yt-dlp",
+        "-f", "bestaudio",
+        "--extract-audio",
+        "--audio-format", "mp3",
+        "--audio-quality", "0",
+        "--force-overwrites",
+        "-o", "P:/Local Music/SoundCloud/%(uploader)s - %(title)s.%(ext)s"
+    ]
+    return run_yt_dlp(" ".join(command), url)
+
+def download_sound