Update app.py

app.py

@@ -1,352 +1,85 @@
 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()
+import yt_dlp
+import os
+import tempfile
+from glob import glob
+
+def download_media(urls, output_format, audio_quality, custom_bitrate, embed_metadata, embed_thumbnail, write_description):
+    # Create a temporary directory to store downloads
+    temp_dir = tempfile.mkdtemp()
+    output_template = os.path.join(temp_dir, '%(title)s.%(ext)s')
+
+    # Prepare the options
+    ydl_opts = {
+        'outtmpl': output_template,
+        'format': 'bestaudio/best',
+        'postprocessors': [],
+    }
+
+    # Set format options
+    if output_format in ['mp3', 'opus']:
+        extract_audio_postprocessor = {
+            'key': 'FFmpegExtractAudio',
+            'preferredcodec': output_format,
+        }
+        # Set postprocessor args
+        if audio_quality == 'Best':
+            extract_audio_postprocessor['preferredquality'] = '0'  # Best quality
+        elif audio_quality == 'Custom' and custom_bitrate:
+            extract_audio_postprocessor['preferredquality'] = '5'  # Default value
+            extract_audio_postprocessor['postprocessor_args'] = ['-b:a', custom_bitrate]
+        ydl_opts['postprocessors'].append(extract_audio_postprocessor)
+    else:
+        ydl_opts['format'] = 'bestvideo+bestaudio/best'
+
+    # Additional options
+    if embed_metadata:
+        ydl_opts['addmetadata'] = True
+        ydl_opts['postprocessors'].append({'key': 'FFmpegMetadata'})
+    if embed_thumbnail:
+        ydl_opts['writethumbnail'] = True
+        ydl_opts['postprocessors'].append({'key': 'EmbedThumbnail'})
+    if write_description:
+        ydl_opts['writedescription'] = True
+
+    # Process URLs
+    url_list = urls.strip().split()
+    with yt_dlp.YoutubeDL(ydl_opts) as ydl:
+        ydl.download(url_list)
+
+    # Collect output files
+    downloaded_files = glob(os.path.join(temp_dir, '*'))
+    # Return the list of file paths for download
+    return downloaded_files
+
+def main():
+    with gr.Blocks() as demo:
+        gr.Markdown("# YT-DLP Gradio Interface")
+        with gr.Row():
+            url_input = gr.Textbox(label="Video URL(s)", placeholder="Enter one or more URLs separated by space or newline")
+        with gr.Row():
+            output_format = gr.Dropdown(['mp3', 'opus', 'bestvideo+bestaudio'], label="Output Format", value='mp3')
+            audio_quality = gr.Dropdown(['Best', 'Custom'], label="Audio Quality", value='Best')
+            custom_bitrate = gr.Textbox(label="Custom Bitrate (e.g., 192k)", visible=False)
+        with gr.Row():
+            embed_metadata = gr.Checkbox(label="Embed Metadata")
+            embed_thumbnail = gr.Checkbox(label="Embed Thumbnail")
+            write_description = gr.Checkbox(label="Write Description")
+        start_button = gr.Button("Start Download")
+        output_files = gr.Files(label="Downloaded Files")
+
+        def show_custom_bitrate(audio_quality):
+            if audio_quality == 'Custom':
+                return gr.update(visible=True)
+            else:
+                return gr.update(visible=False)
+
+        audio_quality.change(show_custom_bitrate, inputs=audio_quality, outputs=custom_bitrate)
+
+        start_button.click(download_media,
+                           inputs=[url_input, output_format, audio_quality, custom_bitrate, embed_metadata, embed_thumbnail, write_description],
+                           outputs=output_files)
+    demo.launch()
+
+if __name__ == "__main__":
+    main()