Update src/covergen.py

src/covergen.py

@@ -6,7 +6,15 @@ from audio_effects import add_audio_effects
 from modules.model_management import ignore_files, update_models_list, download_from_url, upload_zip_model, upload_separate_files
 from modules.ui_updates import show_hop_slider, update_f0_method, update_button_text_voc, update_button_text_inst, swap_visibility, swap_buttons
 from modules.file_processing import process_file_upload
-
+import os
+import re
+import random
+from scipy.io.wavfile import write
+from scipy.io.wavfile import read
+import numpy as np
+import gradio as gr
+import yt_dlp
+import subprocess
 
 separator = Separator()
 
@@ -32,6 +40,30 @@ UVR_5_MODELS = [
     {"model_name": "UVR-DeEcho-DeReverb by FoxJoy", "checkpoint": "UVR-DeEcho-DeReverb.pth"},
 ]
 
+
+
+def download_audio(url):
+    ydl_opts = {
+        'format': 'bestaudio/best',
+        'outtmpl': 'ytdl/%(title)s.%(ext)s',
+        'postprocessors': [{
+            'key': 'FFmpegExtractAudio',
+            'preferredcodec': 'wav',
+            '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] + '.wav'
+        sample_rate, audio_data = read(file_path)
+        audio_array = np.asarray(audio_data, dtype=np.int16)
+
+        return sample_rate, audio_array
+
+
+
+
 def inf_handler(audio, model_name): 
     model_found = False
     for model_info in UVR_5_MODELS:
@@ -107,6 +139,19 @@ with gr.Blocks(title="🎤 RVC Inference", css="footer{display:none !important}"
 
     with gr.Tab(" 🔧 Vocal Separator (UVR)"):
         gr.Markdown("⭕ Separate vocals and instruments from an audio file using UVR models.")
+
+
+        with gr.Accordion("🔗 Separation by Link", open=False):
+            with gr.Row():
+                mdx23c_link = gr.Textbox(label="🔗 Link",placeholder="📋 Paste the link here",interactive=True)
+            with gr.Row():
+                gr.Markdown("💡 You can paste the link to the video/audio from many sites, check the complete list [here](https://github.com/yt-dlp/yt-dlp/blob/master/supportedsites.md)")
+            with gr.Row():
+                mdx23c_download_button = gr.Button("⬇️ Download!",variant="primary")
+
+
+
+        
         uvr5_audio_file = gr.Audio(label=" 📲 Audio File",type="filepath")
 
         with gr.Row():
@@ -116,6 +161,11 @@ with gr.Blocks(title="🎤 RVC Inference", css="footer{display:none !important}"
         uvr5_output_voc = gr.Audio(type="filepath", label="Output 1",)
         uvr5_output_inst = gr.Audio(type="filepath", label="Output 2",)
 
+
+
+
+        mdx23c_download_button.click(download_audio, [mdx23c_link], [uvr5_audio_file])
+
         uvr5_button.click(inference, [uvr5_audio_file, uvr5_model], [uvr5_output_voc, uvr5_output_inst])