app.py

import gradio as gr
import subprocess

def run_command(command):
    try:
        result = subprocess.run(command, shell=True, check=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
        return result.stdout.decode('utf-8')
    except subprocess.CalledProcessError as e:
        return e.stderr.decode('utf-8')

def list_available_formats(url):
    command = f'yt-dlp -F "{url}"'
    return run_command(command)

def download_mp3_max_quality(url):
    command = f'yt-dlp -f bestaudio --extract-audio --audio-format mp3 --audio-quality 0 --postprocessor-args "-b:a 320k" -o "P:/Local Music/%(title)s.%(ext)s" "{url}"'
    return run_command(command)

def basic_high_quality_mp3_download(url):
    command = f'yt-dlp -f bestaudio --extract-audio --audio-format mp3 --audio-quality 0 -o "P:/Local Music/%(title)s.%(ext)s" "{url}"'
    return run_command(command)

def download_custom_bitrate(url, bitrate):
    command = f'yt-dlp -f bestaudio --extract-audio --audio-format mp3 --postprocessor-args "-b:a {bitrate}k" -o "P:/Local Music/%(title)s.%(ext)s" "{url}"'
    return run_command(command)

def download_with_metadata(url):
    command = f'yt-dlp -f bestaudio --extract-audio --audio-format mp3 --audio-quality 0 --embed-metadata -o "P:/Local Music/%(title)s.%(ext)s" "{url}"'
    return run_command(command)

def download_entire_playlist(url):
    command = f'yt-dlp -f bestaudio --extract-audio --audio-format mp3 --audio-quality 0 -o "P:/Local Music/%(playlist_title)s/%(title)s.%(ext)s" "{url}"'
    return run_command(command)

def download_playlist_range(url, start, end):
    command = f'yt-dlp --playlist-items {start}-{end} -f bestaudio --extract-audio --audio-format mp3 --audio-quality 0 -o "P:/Local Music/%(playlist_title)s/%(title)s.%(ext)s" "{url}"'
    return run_command(command)

def download_hq_soundcloud(url):
    command = f'yt-dlp -f bestaudio --extract-audio --audio-format mp3 --audio-quality 0 -o "P:/Local Music/SoundCloud/%(uploader)s - %(title)s.%(ext)s" "{url}"'
    return run_command(command)

def embed_album_art(url):
    command = f'yt-dlp -f bestaudio --extract-audio --audio-format mp3 --embed-thumbnail --audio-quality 0 -o "P:/Local Music/%(title)s.%(ext)s" "{url}"'
    return run_command(command)

def download_specific_format(url, format):
    command = f'yt-dlp -f bestaudio --extract-audio --audio-format {format} -o "P:/Local Music/%(title)s.%(ext)s" "{url}"'
    return run_command(command)

def skip_downloaded_files(url):
    command = f'yt-dlp --download-archive "P:/Local Music/downloaded.txt" -f bestaudio --extract-audio --audio-format mp3 --audio-quality 0 -o "P:/Local Music/%(title)s.%(ext)s" "{url}"'
    return run_command(command)

def download_multiple_files(urls):
    command = f'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" {" ".join(urls)}'
    return run_command(command)

def split_by_chapters(url):
    command = f'yt-dlp -f bestaudio --extract-audio --audio-format mp3 --audio-quality 0 --split-chapters -o "P:/Local Music/%(title)s.%(ext)s" "{url}"'
    return run_command(command)

def extract_portion(url, start_time, end_time):
    command = f'yt-dlp -f bestvideo+bestaudio --external-downloader ffmpeg --external-downloader-args "ffmpeg_i:-ss {start_time} -to {end_time}" -o "P:/Local Music/%(title)s.%(ext)s" "{url}"'
    return run_command(command)

def update_metadata(url):
    command = f'yt-dlp --download-archive "P:/Local Music/downloaded.txt" --skip-download --embed-metadata --embed-thumbnail --write-description "{url}"'
    return run_command(command)

def auto_rename_duplicates(url):
    command = f'yt-dlp -f bestaudio --extract-audio --audio-format mp3 --audio-quality 0 --force-overwrites -o "P:/Local Music/%(title)s.%(ext)s" "{url}"'
    return run_command(command)

def download_from_url_list(file_path):
    command = f'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_command(command)

def limit_download_speed(url, speed):
    command = f'yt-dlp -f bestaudio --extract-audio --audio-format mp3 --audio-quality 0 --rate-limit {speed} -o "P:/Local Music/%(title)s.%(ext)s" "{url}"'
    return run_command(command)

def download_from_vimeo(url):
    command = f'yt-dlp -f bestaudio --extract-audio --audio-format mp3 --audio-quality 0 -o "P:/Local Music/Vimeo/%(uploader)s - %(title)s.%(ext)s" "{url}"'
    return run_command(command)

def download_from_facebook(url):
    command = f'yt-dlp -f bestaudio --extract-audio --audio-format mp3 --audio-quality 0 -o "P:/Local Music/Facebook/%(uploader)s - %(title)s.%(ext)s" "{url}"'
    return run_command(command)

def download_from_instagram(url):
    command = f'yt-dlp -f bestaudio --extract-audio --audio-format mp3 --audio-quality 0 -o "P:/Local Music/Instagram/%(uploader)s - %(title)s.%(ext)s" "{url}"'
    return run_command(command)

def download_from_twitter(url):
    command = f'yt-dlp -f bestaudio --extract-audio --audio-format mp3 --audio-quality 0 -o "P:/Local Music/Twitter/%(uploader)s - %(title)s.%(ext)s" "{url}"'
    return run_command(command)

def download_from_tiktok(url):
    command = f'yt-dlp -f bestaudio --extract-audio --audio-format mp3 --audio-quality 0 -o "P:/Local Music/TikTok/%(uploader)s - %(title)s.%(ext)s" "{url}"'
    return run_command(command)

def download_from_reddit(url):
    command = f'yt-dlp -f bestaudio --extract-audio --audio-format mp3 --audio-quality 0 -o "P:/Local Music/Reddit/%(uploader)s - %(title)s.%(ext)s" "{url}"'
    return run_command(command)

def download_from_dailymotion(url):
    command = f'yt-dlp -f bestaudio --extract-audio --audio-format mp3 --audio-quality 0 -o "P:/Local Music/Dailymotion/%(uploader)s - %(title)s.%(ext)s" "{url}"'
    return run_command(command)

def download_youtube_transcript_text(url):
    command = f'yt-dlp --write-auto-subs --sub-lang en --skip-download -o "P:/Local Music/%(title)s.%(ext)s" "{url}"'
    return run_command(command)

def download_youtube_transcript_markdown(url):
    command = f'yt-dlp --write-auto-subs --sub-lang en --skip-download --convert-subs srt -o "P:/Local Music/%(title)s.md" "{url}"'
    return run_command(command)

def download_video_info_json(url):
    command = f'yt-dlp --write-info-json --skip-download -o "P:/Local Music/%(title)s.json" "{url}"'
    return run_command(command)

def list_available_formats_no_download(url):
    command = f'yt-dlp -F "{url}"'
    return run_command(command)

def download_age_restricted_content(url, email, password):
    command = f'yt-dlp --username "{email}" --password "{password}" -f bestvideo+bestaudio -o "P:/Local Music/%(title)s.%(ext)s" "{url}"'
    return run_command(command)

# Define the Gradio interface
with gr.Blocks() as demo:
    gr.Markdown("## YT-DLP Commands via Gradio")

    with gr.Tab("General Commands"):
        with gr.Row():
            url_input = gr.Textbox(label="URL")
            list_formats_btn = gr.Button("List Available Formats")
            list_formats_btn.click(list_available_formats, inputs=url_input, outputs=gr.Textbox())

        with gr.Row():
            download_mp3_max_btn = gr.Button("Download MP3 at Max Quality")
            download_mp3_max_btn.click(download_mp3_max_quality, inputs=url_input, outputs=gr.Textbox())

        with gr.Row():
            download_mp3_basic_btn = gr.Button("Basic High-Quality MP3 Download")
            download_mp3_basic_btn.click(basic_high_quality_mp3_download, inputs=url_input, outputs=gr.Textbox())

        with gr.Row():
            bitrate_input = gr.Textbox(label="Bitrate (e.g., 192)")
            download_custom_bitrate_btn = gr.Button("Download Custom Bitrate")
            download_custom_bitrate_btn.click(download_custom_bitrate, inputs=[url_input, bitrate_input], outputs=gr.Textbox())

        with gr.Row():
            download_metadata_btn = gr.Button("Download and add Metadata Tags")
            download_metadata_btn.click(download_with_metadata, inputs=url_input, outputs=gr.Textbox())

        with gr.Row():
            playlist_url_input = gr.Textbox(label="Playlist URL")
            download_playlist_btn = gr.Button("Download Entire Playlist")
            download_playlist_btn.click(download_entire_playlist, inputs=playlist_url_input, outputs=gr.Textbox())

        with gr.Row():
            start_input = gr.Textbox(label="Start Index")
            end_input = gr.Textbox(label="End Index")
            download_playlist_range_btn = gr.Button("Download Playlist (Specific Range)")
            download_playlist_range_btn.click(download_playlist_range, inputs=[playlist_url_input, start_input, end_input], outputs=gr.Textbox())

        with gr.Row():
            soundcloud_url_input = gr.Textbox(label="SoundCloud URL")
            download_hq_soundcloud_btn = gr.Button("Download HQ from SoundCloud")
            download_hq_soundcloud_btn.click(download_hq_soundcloud, inputs=soundcloud_url_input, outputs=gr.Textbox())

        with gr.Row():
            embed_album_art_btn = gr.Button("Embed Album Art Automatically")
            embed_album_art_btn.click(embed_album_art, inputs=url_input, outputs=gr.Textbox())

        with gr.Row():
            format_input = gr.Textbox(label="Format (e.g., opus)")
            download_specific_format_btn = gr.Button("Download Specific Formats")
            download_specific_format_btn.click(download_specific_format, inputs=[url_input, format_input], outputs=gr.Textbox())

        with gr.Row():
            skip_downloaded_btn = gr.Button("Skip Already Downloaded Files")
            skip_downloaded_btn.click(skip_downloaded_files, inputs=url_input, outputs=gr.Textbox())

        with gr.Row():
            urls_input = gr.Textbox(label="URLs (space-separated)")
            download_multiple_files_btn = gr.Button("Download Multiple Individual Files")
            download_multiple_files_btn.click(download_multiple_files, inputs=urls_input, outputs=gr.Textbox())

        with gr.Row():
            split_chapters_btn = gr.Button("Download and Automatically Split by Chapters")
            split_chapters_btn.click(split_by_chapters, inputs=url_input, outputs=gr.Textbox())

        with gr.Row():
            start_time_input = gr.Textbox(label="Start Time (e.g., 00:01:00)")
            end_time_input = gr.Textbox(label="End Time (e.g., 00:05:00)")
            extract_portion_btn = gr.Button("Extract Only a Portion of a Video (by Time)")
            extract_portion_btn.click(extract_portion, inputs=[url_input, start_time_input, end_time_input], outputs=gr.Textbox())

        with gr.Row():
            update_metadata_btn = gr.Button("Update Metadata of Existing Files")
            update_metadata_btn.click(update_metadata, inputs=url_input, outputs=gr.Textbox())

        with gr.Row():
            auto_rename_btn = gr.Button("Auto-Rename Duplicate Files")
            auto_rename_btn.click(auto_rename_duplicates, inputs=url_input, outputs=gr.Textbox())

        with gr.Row():
            file_path_input = gr.Textbox(label="File Path")
            download_url_list_btn = gr.Button("Download from a URL List File")
            download_url_list_btn.click(download_from_url_list, inputs=file_path_input, outputs=gr.Textbox())

        with gr.Row():
            speed_input = gr.Textbox(label="Speed (e.g., 1M)")
            limit_download_speed_btn = gr.Button("Download and Limit Download Speed")
            limit_download_speed_btn.click(limit_download_speed, inputs=[url_input, speed_input], outputs=gr.Textbox())

        with gr.Row():
            vimeo_url_input = gr.Textbox(label="Vimeo URL")
            download_vimeo_btn = gr.Button("Download from Vimeo")
            download_vimeo_btn.click(download_from_vimeo, inputs=vimeo_url_input, outputs=gr.Textbox())

        with gr.Row():
            facebook_url_input = gr.Textbox(label="Facebook URL")
            download_facebook_btn = gr.Button("Download from Facebook")
            download_facebook_btn.click(download_from_facebook, inputs=facebook_url_input, outputs=gr.Textbox())

        with gr.Row():
            instagram_url_input = gr.Textbox(label="Instagram URL")
            download_instagram_btn = gr.Button("Download from Instagram")
            download_instagram_btn.click(download_from_instagram, inputs=instagram_url_input, outputs=gr.Textbox())

        with gr.Row():
            twitter_url_input = gr.Textbox(label="Twitter URL")
            download_twitter_btn = gr.Button("Download from Twitter")
            download_twitter_btn.click(download_from_twitter, inputs=twitter_url_input, outputs=gr.Textbox())

        with gr.Row():
            tiktok_url_input = gr.Textbox(label="TikTok URL")
            download_tiktok_btn = gr.Button("Download from TikTok")
            download_tiktok_btn.click(download_from_tiktok, inputs=tiktok_url_input, outputs=gr.Textbox())

        with gr.Row():
            reddit_url_input = gr.Textbox(label="Reddit URL")
            download_reddit_btn = gr.Button("Download from Reddit")
            download_reddit_btn.click(download_from_reddit, inputs=reddit_url_input, outputs=gr.Textbox())

        with gr.Row():
            dailymotion_url_input = gr.Textbox(label="Dailymotion URL")
            download_dailymotion_btn = gr.Button("Download from Dailymotion")
            download_dailymotion_btn.click(download_from_dailymotion, inputs=dailymotion_url_input, outputs=gr.Textbox())

        with gr.Row():
            download_transcript_text_btn = gr.Button("Download YouTube Transcript as Text")
            download_transcript_text_btn.click(download_youtube_transcript_text, inputs=url_input, outputs=gr.Textbox())

        with gr.Row():
            download_transcript_markdown_btn = gr.Button("Download YouTube Transcript as Markdown")
            download_transcript_markdown_btn.click(download_youtube_transcript_markdown, inputs=url_input, outputs=gr.Textbox())

        with gr.Row():
            download_video_info_btn = gr.Button("Download Video Info and Save as JSON")
            download_video_info_btn.click(download_video_info_json, inputs=url_input, outputs=gr.Textbox())

        with gr.Row():
            list_formats_no_download_btn = gr.Button("List Available Formats without Downloading")
            list_formats_no_download_btn.click(list_available_formats_no_download, inputs=url_input, outputs=gr.Textbox())

        with gr.Row():
            email_input = gr.Textbox(label="Email")
            password_input = gr.Textbox(label="Password", type="password")
            age_restricted_url_input = gr.Textbox(label="Age-Restricted Video URL")
            download_age_restricted_btn = gr.Button("Download YouTube Age-Restricted Content")
            download_age_restricted_btn.click(download_age_restricted_content, inputs=[age_restricted_url_input, email_input, password_input], outputs=gr.Textbox())

    with gr.Tab("SoundCloud Commands"):
        with gr.Row():
            soundcloud_list_formats_btn = gr.Button("List Available Formats")
            soundcloud_list_formats_btn.click(list_available_formats, inputs=soundcloud_url_input, outputs=gr.Textbox())

        with gr.Row():
            soundcloud_download_hq_btn = gr.Button("Download High-Quality MP3 from SoundCloud")
            soundcloud_download_hq_btn.click(download_hq_soundcloud, inputs=soundcloud_url_input, outputs=gr.Textbox())

        with gr.Row():
            soundcloud_download_hq_metadata_btn = gr.Button("Download HQ with Metadata from SoundCloud")
            soundcloud_download_hq_metadata_btn.click(download_hq_soundcloud, inputs=soundcloud_url_input, outputs=gr.Textbox())

        with gr.Row():
            soundcloud_playlist_url_input = gr.Textbox(label="SoundCloud Playlist URL")
            soundcloud_download_playlist_btn = gr.Button("Download Entire Playlist from SoundCloud")
            soundcloud_download_playlist_btn.click(download_entire_playlist, inputs=soundcloud_playlist_url_input, outputs=gr.Textbox())

        with gr.Row():
            soundcloud_download_playlist_metadata_btn = gr.Button("Download Entire Playlist from SoundCloud with Metadata")
            soundcloud_download_playlist_metadata_btn.click(download_entire_playlist, inputs=soundcloud_playlist_url_input, outputs=gr.Textbox())

        with gr.Row():
            soundcloud_multiple_playlists_input = gr.Textbox(label="SoundCloud Playlist URLs (space-separated)")
            soundcloud_download_multiple_playlists_btn = gr.Button("Download Multiple Playlists with Metadata")
            soundcloud_download_multiple_playlists_btn.click(download_multiple_files, inputs=soundcloud_multiple_playlists_input, outputs=gr.Textbox())

        with gr.Row():
            soundcloud_playlist_range_start_input = gr.Textbox(label="Start Index")
            soundcloud_playlist_range_end_input = gr.Textbox(label="End Index")
            soundcloud_download_playlist_range_btn = gr.Button("Download Playlist with Specific Range")
            soundcloud_download_playlist_range_btn.click(download_playlist_range, inputs=[soundcloud_playlist_url_input, soundcloud_playlist_range_start_input, soundcloud_playlist_range_end_input], outputs=gr.Textbox())

        with gr.Row():
            soundcloud_embed_album_art_btn = gr.Button("Download and Embed Album Art")
            soundcloud_embed_album_art_btn.click(embed_album_art, inputs=soundcloud_url_input, outputs=gr.Textbox())

        with gr.Row():
            soundcloud_skip_downloaded_btn = gr.Button("Skip Downloading Already Downloaded SoundCloud Files")
            soundcloud_skip_downloaded_btn.click(skip_downloaded_files, inputs=soundcloud_url_input, outputs=gr.Textbox())

        with gr.Row():
            soundcloud_multiple_tracks_input = gr.Textbox(label="SoundCloud Track URLs (space-separated)")
            soundcloud_download_multiple_tracks_btn = gr.Button("Download Multiple Tracks from SoundCloud")
            soundcloud_download_multiple_tracks_btn.click(download_multiple_files, inputs=soundcloud_multiple_tracks_input, outputs=gr.Textbox())

        with gr.Row():
            soundcloud_limit_speed_input = gr.Textbox(label="Speed (e.g., 1M)")
            soundcloud_limit_download_speed_btn = gr.Button("Limit Download Speed for SoundCloud")
            soundcloud_limit_download_speed_btn.click(limit_download_speed, inputs=[soundcloud_url_input, soundcloud_limit_speed_input], outputs=gr.Textbox())

        with gr.Row():
            soundcloud_split_chapters_btn = gr.Button("Download from SoundCloud and Split by Chapters (if available)")
            soundcloud_split_chapters_btn.click(split_by_chapters, inputs=soundcloud_url_input, outputs=gr.Textbox())

        with gr.Row():
            soundcloud_extract_portion_start_input = gr.Textbox(label="Start Time (e.g., 00:01:00)")
            soundcloud_extract_portion_end_input = gr.Textbox(label="End Time (e.g., 00:05:00)")
            soundcloud_extract_portion_btn = gr.Button("Download SoundCloud Track and Extract a Specific Portion")
            soundcloud_extract_portion_btn.click(extract_portion, inputs=[soundcloud_url_input, soundcloud_extract_portion_start_input, soundcloud_extract_portion_end_input], outputs=gr.Textbox())

        with gr.Row():
            soundcloud_update_metadata_btn = gr.Button("Update Metadata of Existing SoundCloud Downloads")
            soundcloud_update_metadata_btn.click(update_metadata, inputs=soundcloud_url_input, outputs=gr.Textbox())

        with gr.Row():
            soundcloud_auto_rename_btn = gr.Button("Auto-Rename Duplicate SoundCloud Files")
            soundcloud_auto_rename_btn.click(auto_rename_duplicates, inputs=soundcloud_url_input, outputs=gr.Textbox())

        with gr.Row():
            soundcloud_url_list_file_input = gr.Textbox(label="File Path")
            soundcloud_download_url_list_btn = gr.Button("Download from SoundCloud URL List File")
            soundcloud_download_url_list_btn.click(download_from_url_list, inputs=soundcloud_url_list_file_input, outputs=gr.Textbox())

demo.launch()