Update run.py

run.py

@@ -27,17 +27,17 @@ def download_youtube(url, file_format):
             if file_format == "audio":
                 file_path = os.path.splitext(file_path)[0] + ".mp3"
 
-        return file_path
+        return file_path, None
     except Exception as e:
-        return str(e)
+        return None, str(e)
 
-def show_output(file_path):
-    if not file_path or not os.path.exists(file_path):
-        return "Error downloading file.", None, None
+def show_output(file_path, error):
+    if error:
+        return error, None, None
     if file_path.endswith(".mp3"):
-        return None, gr.Audio(file_path, label="Audio Output"), None
+        return None, gr.Audio.update(value=file_path, visible=True), gr.Video.update(visible=False)
     else:
-        return None, None, gr.Video(file_path, label="Video Output")
+        return None, gr.Audio.update(visible=False), gr.Video.update(value=file_path, visible=True)
 
 with gr.Blocks() as demo:
     with gr.Row():
@@ -50,8 +50,8 @@ with gr.Blocks() as demo:
             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=[error_output, audio_output, video_output]
+    download_button.click(download_youtube, inputs=[url_input, format_input], outputs=[error_output, audio_output, video_output]).then(
+        show_output, inputs=[error_output, audio_output], outputs=[error_output, audio_output, video_output]
     )
 
 demo.launch()