Update run.py

run.py

@@ -1,12 +1,10 @@
 import gradio as gr
-from yt_dlp import YoutubeDL
+import yt_dlp
 import os
 
-def download_youtube(url, output_format):
+def download_youtube(url, file_format):
     ydl_opts = {}
-    file_path = ""
-    
-    if output_format == "audio":
+    if file_format == "audio":
         ydl_opts = {
             'format': 'bestaudio/best',
             'postprocessors': [{
@@ -16,40 +14,38 @@ def download_youtube(url, output_format):
             }],
             'outtmpl': 'downloads/%(title)s.%(ext)s',
         }
-    elif output_format == "video":
+    else:
         ydl_opts = {
             'format': 'bestvideo+bestaudio/best',
             'outtmpl': 'downloads/%(title)s.%(ext)s',
         }
 
-    with YoutubeDL(ydl_opts) as ydl:
-        result = ydl.extract_info(url, download=True)
-        file_path = ydl.prepare_filename(result)
-
-        if output_format == "audio":
-            file_path = file_path.replace(result['ext'], 'mp3')
+    with yt_dlp.YoutubeDL(ydl_opts) as ydl:
+        info_dict = ydl.extract_info(url, download=True)
+        file_path = ydl.prepare_filename(info_dict)
+        if file_format == "audio":
+            file_path = os.path.splitext(file_path)[0] + ".mp3"
 
     return file_path
 
-def show_media(file_path, output_format):
-    if output_format == "audio":
-        return gr.Audio.update(value=file_path)
-    elif output_format == "video":
-        return gr.Video.update(value=file_path)
-
-def yt_download(url, output_format):
-    file_path = download_youtube(url, output_format)
-    return show_media(file_path, output_format)
+def show_output(file_path):
+    if file_path.endswith(".mp3"):
+        return gr.Audio(file_path, label="Audio Output"), None
+    else:
+        return None, gr.Video(file_path, label="Video Output")
 
 with gr.Blocks() as demo:
-    gr.Markdown("## YouTube Downloader")
-    url_input = gr.Textbox(label="YouTube URL")
-    format_input = gr.Radio(choices=["audio", "video"], label="Format")
-    download_btn = gr.Button("Download")
-    download_btn.click(yt_download, inputs=[url_input, format_input], outputs=[audio_output, video_output])
+    with gr.Row():
+        with gr.Column():
+            url_input = gr.Textbox(label="YouTube URL")
+            format_input = gr.Dropdown(choices=["audio", "video"], label="Format")
+            download_button = gr.Button("Download")
+        with gr.Column():
+            audio_output = gr.Audio(label="Audio Output", visible=False)
+            video_output = gr.Video(label="Video Output", visible=False)
+
+    download_button.click(download_youtube, inputs=[url_input, format_input], outputs=None).then(
+        show_output, inputs=None, outputs=[audio_output, video_output]
+    )
 
-    audio_output = gr.Audio(label="Audio Output")
-    
-    video_output = gr.Video(label="Video Output")
-        
 demo.launch()