Update app.py

app.py

@@ -1,85 +1,106 @@
 import gradio as gr
-import yt_dlp
-import os
-import tempfile
-from glob import glob
+import subprocess
 
-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')
+# 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
 
-    # 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
+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"'
+    ]
     if embed_metadata:
-        ydl_opts['addmetadata'] = True
-        ydl_opts['postprocessors'].append({'key': 'FFmpegMetadata'})
+        options.append('--embed-metadata')
     if embed_thumbnail:
-        ydl_opts['writethumbnail'] = True
-        ydl_opts['postprocessors'].append({'key': 'EmbedThumbnail'})
-    if write_description:
-        ydl_opts['writedescription'] = True
+        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
 
-    # Process URLs
-    url_list = urls.strip().split()
-    with yt_dlp.YoutubeDL(ydl_opts) as ydl:
-        ydl.download(url_list)
+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
 
-    # Collect output files
-    downloaded_files = glob(os.path.join(temp_dir, '*'))
-    # Return the list of file paths for download
-    return downloaded_files
+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)
 
-def main():
-    with gr.Blocks() as demo:
-        gr.Markdown("# YT-DLP Gradio Interface")
+# Gradio interface components
+with gr.Blocks() as yt_dlp_app:
+    gr.Markdown("## YT-DLP Command Helper")
+    with gr.Tab("General Commands"):
         with gr.Row():
-            url_input = gr.Textbox(label="Video URL(s)", placeholder="Enter one or more URLs separated by space or newline")
+            url = gr.Textbox(label="Video URL")
+            output_folder = gr.Textbox(label="Output Folder", value="P:/Local Music")
         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)
+            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"):
         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")
+            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)
 
-        def show_custom_bitrate(audio_quality):
-            if audio_quality == 'Custom':
-                return gr.update(visible=True)
-            else:
-                return gr.update(visible=False)
+        format_button.click(list_formats, inputs=[format_url], outputs=format_output)
 
-        audio_quality.change(show_custom_bitrate, inputs=audio_quality, outputs=custom_bitrate)
+    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)
 
-        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()
+        custom_button.click(
+            download_custom, 
+            inputs=[custom_urls, output_folder, command_template], 
+            outputs=custom_output
+        )
 
-if __name__ == "__main__":
-    main()
+yt_dlp_app.launch()