Update app.py

app.py

@@ -2,34 +2,43 @@ import gradio as gr
 import yt_dlp
 import os
 
+def download_media(url, output_format):
+    ydl_opts = {}
 
-os.system("python3 -m pip install --no-deps -U yt-dlp ")
-
+    if output_format == "Audio":
+        ydl_opts = {
+            'format': 'bestaudio/best',
+            'outtmpl': 'downloads/%(title)s.%(ext)s',
+            'postprocessors': [{
+                'key': 'FFmpegExtractAudio',
+                'preferredcodec': 'mp3',
+                'preferredquality': '192',
+            }],
+        }
+    elif output_format == "Video":
+        ydl_opts = {
+            'format': 'bestvideo+bestaudio/best',
+            'outtmpl': 'downloads/%(title)s.%(ext)s',
+        }
+    
+    try:
+        with yt_dlp.YoutubeDL(ydl_opts) as ydl:
+            ydl.download([url])
+            return "Download successful!"
+    except Exception as e:
+        return f"Error: {str(e)}"
 
-          
-def download_video(url):
-    ydl_opts = {
-        'format': 'best',
-        'outtmpl': 'downloads/%(title)s.%(ext)s',
-    }
+# Gradio Blocks UI
+with gr.Blocks() as demo:
+    gr.Markdown("# Video and Audio Downloader using yt-dlp")
     
-    with yt_dlp.YoutubeDL(ydl_opts) as ydl:
-        info_dict = ydl.extract_info(url, download=True)
-        video_title = info_dict.get('title', None)
-        file_extension = info_dict.get('ext', None)
-        file_name = f"{video_title}.wav"
-        file_path = f"downloads/{file_name}"
+    with gr.Row():
+        url_input = gr.Textbox(label="Video/Audio URL")
+        format_dropdown = gr.Dropdown(choices=["Audio", "Video"], label="Output Format")
+        
+    download_button = gr.Button("Download")
+    output_text = gr.Textbox(label="Output Message")
     
-    return file_path
-
-# Create a Gradio interface
-iface = gr.Interface(
-    fn=download_video, 
-    inputs="text", 
-    outputs="audio",
-    title="Video Downloader",
-    description="Enter the URL of the video you want to download."
-)
+    download_button.click(download_media, inputs=[url_input, format_dropdown], outputs=output_text)
 
-# Launch the interface
-iface.launch()
+demo.launch()