Update app.py

app.py

@@ -1,106 +1,85 @@
 import gradio as gr
-import subprocess
+import yt_dlp
+import os
+import tempfile
+from glob import glob
 
-# Define functions for each command
-def list_formats(url):
-    command = f'yt-dlp -F "{url}"'
-    result = subprocess.run(command, shell=True, capture_output=True, text=True)
-    return result.stdout
+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')
 
-def download_mp3(url, output_folder="P:/Local Music", audio_quality="0", bitrate="320k", embed_metadata=False, embed_thumbnail=False):
-    options = [
-        f'yt-dlp -f bestaudio --extract-audio --audio-format mp3 --audio-quality {audio_quality}',
-        f'--postprocessor-args "-b:a {bitrate}"',
-        f'-o "{output_folder}/%(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:
-        options.append('--embed-metadata')
+        ydl_opts['addmetadata'] = True
+        ydl_opts['postprocessors'].append({'key': 'FFmpegMetadata'})
     if embed_thumbnail:
-        options.append('--embed-thumbnail')
-    options.append(f'"{url}"')
-    command = " ".join(options)
-    result = subprocess.run(command, shell=True, capture_output=True, text=True)
-    return result.stdout
+        ydl_opts['writethumbnail'] = True
+        ydl_opts['postprocessors'].append({'key': 'EmbedThumbnail'})
+    if write_description:
+        ydl_opts['writedescription'] = True
 
-def download_playlist(url, output_folder="P:/Local Music", range_items="", embed_metadata=False):
-    options = [
-        f'yt-dlp -f bestaudio --extract-audio --audio-format mp3 --audio-quality 0',
-        f'-o "{output_folder}/%(playlist_title)s/%(title)s.%(ext)s"'
-    ]
-    if range_items:
-        options.insert(1, f'--playlist-items {range_items}')
-    if embed_metadata:
-        options.append('--embed-metadata')
-    options.append(f'"{url}"')
-    command = " ".join(options)
-    result = subprocess.run(command, shell=True, capture_output=True, text=True)
-    return result.stdout
+    # Process URLs
+    url_list = urls.strip().split()
+    with yt_dlp.YoutubeDL(ydl_opts) as ydl:
+        ydl.download(url_list)
 
-def download_custom(url_list, output_folder="P:/Local Music", command_template=""):
-    results = []
-    for url in url_list.splitlines():
-        command = command_template.replace("VIDEO_URL", url).replace("P:/Local Music", output_folder)
-        result = subprocess.run(command, shell=True, capture_output=True, text=True)
-        results.append(result.stdout)
-    return "\n".join(results)
+    # Collect output files
+    downloaded_files = glob(os.path.join(temp_dir, '*'))
+    # Return the list of file paths for download
+    return downloaded_files
 
-# Gradio interface components
-with gr.Blocks() as yt_dlp_app:
-    gr.Markdown("## YT-DLP Command Helper")
-    with gr.Tab("General Commands"):
+def main():
+    with gr.Blocks() as demo:
+        gr.Markdown("# YT-DLP Gradio Interface")
         with gr.Row():
-            url = gr.Textbox(label="Video URL")
-            output_folder = gr.Textbox(label="Output Folder", value="P:/Local Music")
+            url_input = gr.Textbox(label="Video URL(s)", placeholder="Enter one or more URLs separated by space or newline")
         with gr.Row():
-            quality = gr.Dropdown(["0", "1", "2", "3"], label="Audio Quality", value="0")
-            bitrate = gr.Textbox(label="Custom Bitrate", value="320k")
-        embed_metadata = gr.Checkbox(label="Embed Metadata")
-        embed_thumbnail = gr.Checkbox(label="Embed Thumbnail")
-        download_button = gr.Button("Download MP3")
-        download_output = gr.Textbox(label="Command Output", lines=10)
-
-        download_button.click(
-            download_mp3, 
-            inputs=[url, output_folder, quality, bitrate, embed_metadata, embed_thumbnail], 
-            outputs=download_output
-        )
-
-    with gr.Tab("List Formats"):
+            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():
-            format_url = gr.Textbox(label="URL to List Formats")
-        format_button = gr.Button("List Available Formats")
-        format_output = gr.Textbox(label="Command Output", lines=10)
+            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")
 
-        format_button.click(list_formats, inputs=[format_url], outputs=format_output)
+        def show_custom_bitrate(audio_quality):
+            if audio_quality == 'Custom':
+                return gr.update(visible=True)
+            else:
+                return gr.update(visible=False)
 
-    with gr.Tab("Playlists"):
-        with gr.Row():
-            playlist_url = gr.Textbox(label="Playlist URL")
-            range_items = gr.Textbox(label="Playlist Range (e.g., 1-5)", placeholder="Optional")
-        embed_metadata_playlist = gr.Checkbox(label="Embed Metadata")
-        playlist_button = gr.Button("Download Playlist")
-        playlist_output = gr.Textbox(label="Command Output", lines=10)
-
-        playlist_button.click(
-            download_playlist,
-            inputs=[playlist_url, output_folder, range_items, embed_metadata_playlist],
-            outputs=playlist_output
-        )
-
-    with gr.Tab("Custom Commands"):
-        with gr.Row():
-            custom_urls = gr.Textbox(label="URLs (one per line)")
-            command_template = gr.Textbox(
-                label="Custom Command Template",
-                placeholder="Example: yt-dlp -f bestaudio --extract-audio --audio-format mp3 -o 'P:/Local Music/%(title)s.%(ext)s' VIDEO_URL"
-            )
-        custom_button = gr.Button("Run Custom Commands")
-        custom_output = gr.Textbox(label="Command Output", lines=10)
+        audio_quality.change(show_custom_bitrate, inputs=audio_quality, outputs=custom_bitrate)
 
-        custom_button.click(
-            download_custom, 
-            inputs=[custom_urls, output_folder, command_template], 
-            outputs=custom_output
-        )
+        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()
 
-yt_dlp_app.launch()
+if __name__ == "__main__":
+    main()