Update app.py

app.py

@@ -1,6 +1,9 @@
 import os
 from scipy.io.wavfile import write
 import gradio as gr
+import yt_dlp
+
+os.makedirs("ytdl", exist_ok=True)
 
 def separator(audio):
     os.makedirs("output", exist_ok=True)
@@ -20,12 +23,50 @@ def separator(audio):
 
     return files
 
-with gr.Blocks() as demo:
-    audio_input = gr.Audio(type="numpy", label="Input")
+
+
+
+def download_audio(url):
+    ydl_opts = {
+        'format': 'bestaudio/best',
+        'outtmpl': 'ytdl/%(title)s.%(ext)s',
+        'postprocessors': [{
+            'key': 'FFmpegExtractAudio',
+            'preferredcodec': 'mp3',
+            'preferredquality': '192',
+        }],
+    }
+
+    with yt_dlp.YoutubeDL(ydl_opts) as ydl:
+        info_dict = ydl.extract_info(url, download=True)
+        file_path = ydl.prepare_filename(info_dict).rsplit('.', 1)[0] + '.mp3'
+        return file_path
+
+def download_and_show(url):
+    file_path = download_audio(url)
+    return file_path
     
-    output1 = gr.Audio(type="filepath", label="Stem 1", interactive=False)
-    output2 = gr.Audio(type="filepath", label="Stem 2", interactive=False)
-    submit_button = gr.Button("Separate")
-    submit_button.click(separator, inputs=audio_input, outputs=[output1, output2])
+
+with gr.Blocks() as demo:
+    with gr.Tab("Esparator"):
+        audio_input = gr.Audio(type="numpy", label="Input")
+        output1 = gr.Audio(type="filepath", label="Stem 1", interactive=False)
+        output2 = gr.Audio(type="filepath", label="Stem 2", interactive=False)
+        submit_button = gr.Button("Separate")
+        submit_button.click(separator, inputs=audio_input, outputs=[output1, output2])
+    with gr.Tab("Download youtube audio for esparation"):
+        url = gr.Textbox(label="Url To youtube")
+        download = gr.Button("download")
+        download.click
+        (  
+            download_and_show,
+            inputs=[
+                url,
+            ],
+        outputs=
+        [
+            audio_input,
+        ],
+        )
 
 demo.launch()