Update app.py

app.py

@@ -1,21 +1,26 @@
 import gradio as gr
 import os
 
-def fetch(url, name, ext):
-  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}"
-  os.system(f"yt-dlp {opts} {url} -o {filename}")
-  return filename
-  
-gr.Interface(
-  fetch, 
-  [gr.Textbox(label="Media link", placeholder="URL goes here..."),
-   gr.Textbox(label="File name", placeholder="best-vid"),
-   gr.Dropdown(["mp3", "mp4"], value="mp4", label="File type")],
-  gr.outputs.File(label="Download!"),
-  description="Download web media! Works on most sites like YouTube, Reddit, Twitter, Instagram, etc.",
-  enable_queue=True
-).launch()
+def fetch(url, name, audio_only):
+    filename = f"{name}.{'mp3' if audio_only else 'mp4'}"
+    
+    if audio_only:
+        opts = f'-x --audio-format mp3 -o "{filename}"'
+    else:
+        opts = f'-o "{filename}"'
+
+    os.system(f'yt-dlp {opts} "{url}"')
+    return filename if os.path.exists(filename) else "Download failed."
+
+with gr.Blocks() as demo:
+    gr.Markdown("Downloads video/audio from url as mp4 (or mp3). Simply a convenience wrapper around [yt-dlp](https://github.com/yt-dlp/yt-dlp) :)")
+
+    url = gr.Textbox(label="media URL")
+    name = gr.Textbox(label="output name", value="output")
+    audio_only = gr.Checkbox(label="audio only (mp3)", value=False)
+    button = gr.Button("download")
+    output = gr.File()
+
+    button.click(fetch, inputs=[url, name, audio_only], outputs=output)
+
+demo.launch()