Update app.py

app.py

@@ -1,22 +1,34 @@
-import webui
+import gradio as gr
 import yt_dlp
 
-class YoutubeDownloader(webui.Block):
-    def __init__(self):
-        super().__init__()
-        self.url_input = webui.Textbox(label="YouTube URL")
-        self.download_button = webui.Button(label="Download")
-        self.result_text = webui.Textarea(rows=10)
+def download_media(url, media_type):
+    ydl_opts = {
+        'format': 'bestaudio/best' if media_type == 'Audio' else 'bestvideo+bestaudio/best',
+        'outtmpl': '%(title)s.%(ext)s',
+    }
+    with yt_dlp.YoutubeDL(ydl_opts) as ydl:
+        result = ydl.extract_info(url, download=True)
+        filename = ydl.prepare_filename(result)
+    return filename
 
-    def on_button_click(self, event):
-        url = self.url_input.get()
-        ydl = yt_dlp.YoutubeDLP()
-        try:
-            results = ydl.extract_info(url, download=True)
-            self.result_text.set("Download information:\n" + str(results))
-        except Exception as e:
-            self.result_text.set("Error: " + str(e))
+def download_audio(url):
+    return download_media(url, 'Audio')
 
+def download_video(url):
+    return download_media(url, 'Video')
 
-if __name__ == "__main__":
-    webui.launch(YoutubeDownloader(), port=8000)
+with gr.Blocks() as demo:
+    gr.Markdown("# YouTube Audio/Video Downloader")
+
+    url_input = gr.Textbox(label="YouTube URL")
+    media_type = gr.Radio(label="Select Media Type", choices=["Audio", "Video"])
+    output = gr.Textbox(label="Downloaded File")
+
+    download_button = gr.Button("Download")
+    download_button.click(
+        fn=lambda url, media: download_audio(url) if media == 'Audio' else download_video(url),
+        inputs=[url_input, media_type],
+        outputs=output
+    )
+
+demo.launch()