Update app.py

app.py

@@ -1,28 +1,45 @@
 import yt_dlp
 import gradio as gr
 
-def download_youtube_wav(url, custom_name):
+import os
+import yt_dlp
+import gradio as gr
+
+def downloader(video_url, save_folder, audio_format, audio_name):
+    save_path = os.path.join(save_folder, audio_name)
     ydl_opts = {
         'format': 'bestaudio/best',
         'postprocessors': [{
             'key': 'FFmpegExtractAudio',
-            'preferredcodec': 'wav',
-            'preferredquality': '192',
+            'preferredcodec': audio_format,
         }],
-        'outtmpl': f'{custom_name}.%(ext)s',
+        'outtmpl': save_path,
     }
 
     with yt_dlp.YoutubeDL(ydl_opts) as ydl:
-        ydl.download([url])
-    
-    return f'Download complete! File saved as: {custom_name}.wav'
+        ydl.download([video_url])
+    return f"Audio downloaded successfully to {save_path}"
+
+inputs = [
+    gr.inputs.Textbox(label="YouTube video link", default="https://youtu.be/L3P0LgwuPPo?si=3dLDS6WIGM2Lk9gf"),
+    gr.inputs.Textbox(label="Audio name", default="Harenchi"),
+    gr.inputs.Textbox(label="Output path for downloaded audio files", default="/content/drive/MyDrive/Separar"),
+    gr.inputs.Radio(["wav", "flac", "mp3"], label="Select the output format", default="wav")
+]
+
+output = gr.outputs.Textbox(label="Output")
+
+title = "YouTube Audio Downloader"
+description = "Download audio from YouTube videos"
+
+gr.Interface(
+    fn=downloader,
+    inputs=inputs,
+    outputs=output,
+    title=title,
+    description=description,
+    theme="light"
+).launch()
 
 
-with gr.Blocks() as demo:
-    url = gr.Textbox(label="url video")
-    custom_name = gr.Textbox(label="custom_name")
-    output = gr.Audio(label="output")
-    download_youtube_wav_btn = gr.Button("Greet")
-    download_youtube_wav_btn.click(fn=download_youtube_wav, inputs=url, custom_name, outputs=output)
 
-demo.launch()