Update app.py

app.py

@@ -9,6 +9,8 @@ os.environ['weight_root']="assets/weights"
 from infer.modules.vc.modules import VC
 from configs.config import Config
 import torch
+import yt_dlp
+import ffmpeg
 os.makedirs(os.path.join(".", "audios"), exist_ok=True)
 config = Config()
 vc = VC(config)
@@ -17,6 +19,21 @@ def warn(text):
     try: gr.Warning(text)
     except: pass
 
+
+def download_audio(url, audio_name):
+    ydl_opts = {
+        'format': 'bestaudio/best',
+        'postprocessors': [{
+            'key': 'FFmpegExtractAudio',
+            'preferredcodec': 'wav',
+            'preferredquality': '192',
+        }],
+        'outtmpl': 'audios/{audio_name}.%(ext)s',
+    }
+    with yt_dlp.YoutubeDL(ydl_opts) as ydl:
+        ydl.download([url])
+    return "audio downloaded as '{audio_name}.wav'"
+    
 def load_model(model_picker,index_picker):
     logs = show_available("logs")
     if model_picker.replace(".pth","") in logs:
@@ -252,6 +269,13 @@ with gr.Blocks() as app:
                     model_picker = gr.Dropdown(label="Model: ",choices=show_available('assets/weights','.pth'),value=show_available('assets/weights','.pth')[0],interactive=True,allow_custom_value=True)
                     index_picker = gr.Dropdown(label="Index:",interactive=True,choices=show_available('logs'),value=show_available('logs')[0],allow_custom_value=True)
                     model_picker.change(fn=load_model,inputs=[model_picker,index_picker],outputs=[index_picker])
+                    with gr.Tabs():
+                with gr.TabItem("2.download youtube wav:"):
+                    url = gr.Textbox(label="url youtube")
+                    audio_name = gr.Textbox(label="name audio")
+                    output_audio = gr.Audio(label="output audio")
+                    dwnl_button = gr.Button("Download")
+                    dwnl_button.click(fn=download_audio,inputs=[url,audio_name],outputs=[output_audio])
                 with gr.TabItem("(Or download a model here)"):
                     with gr.Row():
                         url = gr.Textbox(label="Paste the URL here:",value="",placeholder="(i.e. https://huggingface.co/repo/model/resolve/main/model.zip)")
@@ -268,7 +292,7 @@ with gr.Blocks() as app:
                 with gr.TabItem("Advanced"):
                     index_rate = gr.Slider(label='Index Rate: ',minimum=0,maximum=1,value=0.66,step=0.01)
                     pitch = gr.Slider(label='Pitch (-12 lowers it an octave, 0 keeps the original pitch, 12 lifts it an octave): ',minimum =-12, maximum=12, step=1, value=0, interactive=True)
-                    method = gr.Dropdown(label="Method:",choices=["rmvpe","pm"],value="rmvpe")
+                    method = gr.Dropdown(label="Method:",choices=["rmvpe","crepe", "pm"],value="rmvpe")
         
     with gr.Row():
         with gr.Tabs():