Update app.py

app.py

@@ -27,19 +27,34 @@ def download_media(url, download_video):
         else:
             output_file = file_title.rsplit('.', 1)[0] + '.mp3'
 
-    return output_file
+    return output_file, download_video
 
-# brr 
+def get_output_component(download_video):
+    if download_video:
+        return gr.File(label="Downloaded Media")
+    else:
+        return gr.Audio(label="Downloaded Media")
+
+# Create the Gradio interface
 with gr.Blocks() as demo:
     gr.Markdown("# YouTube Media Downloader")
+    
     with gr.Row():
         url_input = gr.Textbox(label="YouTube URL")
-    with gr.Row():
-        download_video_checkbox = gr.Checkbox(label="Or Download Video", value=False)
+        download_video_checkbox = gr.Checkbox(label="Download Video", value=False)
     with gr.Row():
         download_button = gr.Button("Download")
+
     with gr.Row():
-        output = gr.File(label="Downloaded Media")
-        download_button.click(download_media, inputs=[url_input, download_video_checkbox], outputs=output)
+    dynamic_output = gr.State()
+
+    def update_output_component(download_video):
+        return get_output_component(download_video)
+    
+    dynamic_output = download_video_checkbox.change(
+        update_output_component, inputs=download_video_checkbox, outputs=dynamic_output
+    )
+
+    download_button.click(download_media, inputs=[url_input, download_video_checkbox], outputs=[dynamic_output, download_video_checkbox])
 
 demo.launch()