Update app.py

app.py

@@ -1,32 +1,39 @@
 import gradio as gr
 import os
 
-def download_url(url, name, ext):
+def download_url(url, name):
+    opts = {
+        "mp3": "-f \"ba\" -x --audio-format mp3",
+    }
+    filename = f"{name}.mp3"
+    os.system(f"yt-dlp {opts} {url} -o {filename}")
+    return filename
+
+
+
+def download_url_vid(url, name):
     opts = {
-        "mp3": '-f "ba" -x --audio-format mp3',
         "mp4": '-f "bestvideo[ext=mp4]+bestaudio[ext=m4a]/best[ext=mp4]/best"',
-    }[ext]
-    filename = f"{name}.{ext}"
+    }[
+    filename = f"{name}.mp4"
     os.system(f"yt-dlp {opts} {url} -o {filename}")
     return filename
 
 with gr.Blocks(theme="Nex432/green") as demo:
-    with gr.Tab("main settings"):
+    gr.Markdown(f"# MEDIA downloader | Easy `DOWNLOAD media LINK` YT, IG ETC")
+    with gr.Tab("audio downloader"):
         url = gr.Textbox(label="Media URL")
         name_file = gr.Textbox(label="Media Name")
-        format = gr.Dropdown(label="Format File", choices=["mp3", "mp4"])
         download = gr.Button("Download")
-        audio_output = gr.Audio(visible=False)
-        video_output = gr.Video(visible=False)
-
-        def download_media(url, name_file, format):
-            filename = download_url(url, name_file, format)
-            if format == "mp3":
-                return filename, None  # Return filename for audio, None for video
-            else:
-                return None, filename  # Return None for audio, filename for video
+        audio_output = gr.Audio(label="Audo Output")    
+        download.click(download_media, inputs=[url, name_file], outputs=[audio_output])
 
-        download.click(download_media, inputs=[url, name_file, format], outputs=[audio_output, video_output])
+    with gr.Tab("video downloader"):
+        url = gr.Textbox(label="Media URL")
+        name_file = gr.Textbox(label="Media Name")
+        download = gr.Button("Download")
+        video_output = gr.Video(label="Video Output")
+        download.click(download_url_vid, inputs=[url, name_file,], outputs=[video_output])
 
     with gr.Tab("credits"):
         gr.Markdown(