Update app.py

app.py

@@ -33,28 +33,33 @@ def download_media(url, download_video):
 
     return output_file
 
-def download_handler(url, download_video):
-    if url:
-        output_file = download_media(url, download_video)
-        return "Download completed!", output_file
+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 "Please enter a YouTube URL", None
+        return gr.update(visible=False), gr.update(value=output_file, visible=True)
 
-with gr.Blocks() as demo:
-    gr.Markdown("# Download From YOUTUBE")
-    gr.Markdown("## Download MP3/MP4 from YouTube URL")
-    
-    url = gr.Textbox(label="YouTube URL")
-    download_video = gr.Checkbox(label="Download Video", value=False)
-    download_btn = gr.Button("Download")
+with gr.Blocks(theme=gr.themes.Soft(primary_hue="orange", secondary_hue="orange")) as demo:
+    gr.Markdown(f"# <div style='text-align: center;'>YOUTUBE Downloader</div>")
+    gr.Markdown(f"## <div style='text-align: center;'>Download MP3/MP4 from YouTube URL</div>")
     
-    output_text = gr.Markdown()
-    output_file = gr.File()
-
-    download_btn.click(download_handler, inputs=[url, download_video], outputs=[output_text, output_file])
+    with gr.Row():
+        url_input = gr.Textbox(label="YouTube URL")
+    with gr.Row():
+        download_video_checkbox = gr.Checkbox(label="Download Video", value=False)
+    with gr.Row():    
+        download_button = gr.Button("Download")
+    with gr.Row():
+        output_audio = gr.Audio(label="Downloaded Media", visible=False)
+        output_file = gr.File(label="Downloaded Media", visible=False)
 
+    download_button.click(
+        handle_download, 
+        inputs=[url_input, download_video_checkbox], 
+        outputs=[output_file, output_audio]
+    )
 
-    gr.Markdown("Build AI with Haseeb Ahmed (AI/ML engineer)")
-    gr.Markdown("### Made with ❤ by [Haseeb Ahmed](https://www.linkedin.com/in/muhammad-haseeb-ahmed-1954b5230/)")
+    gr.Markdown(f"### <div style='text-align: center;'>Made with ❤ by <a href='https://huggingface.co/Gradio-Blocks'>Gradio-Blocks-Party-Member</a></div>")
 
 demo.launch()