Update app.py

app.py

@@ -27,7 +27,7 @@ def download_media(url, download_video):
         else:
             output_file = file_title.rsplit('.', 1)[0] + '.mp3'
 
-    return output_file, download_video
+    return output_file
 
 def get_output_component(download_video):
     if download_video:
@@ -42,19 +42,22 @@ with gr.Blocks() as demo:
     with gr.Row():
         url_input = gr.Textbox(label="YouTube URL")
         download_video_checkbox = gr.Checkbox(label="Download Video", value=False)
-    with gr.Row():
         download_button = gr.Button("Download")
-
     
-    dynamic_output = gr.State()
+    output_audio = gr.Audio(label="Downloaded Media", visible=False)
+    output_file = gr.File(label="Downloaded Media", visible=False)
 
-    def update_output_component(download_video):
-        return get_output_component(download_video)
+    def handle_download(url, download_video):
+        output_file = download_media(url, download_video)
+        if download_video:
+            return gr.update(value=output_file, visible=True), gr.update(visible=False)
+        else:
+            return gr.update(visible=False), gr.update(value=output_file, visible=True)
     
-    dynamic_output = download_video_checkbox.change(
-        update_output_component, inputs=download_video_checkbox, outputs=dynamic_output
+    download_button.click(
+        handle_download, 
+        inputs=[url_input, download_video_checkbox], 
+        outputs=[output_file, output_audio]
     )
 
-    download_button.click(download_media, inputs=[url_input, download_video_checkbox], outputs=[dynamic_output, download_video_checkbox])
-
 demo.launch()