Separation by link (#6)

- Separation by link (486ccbd939cc60ebc76dceb82481f8c917896947)


Co-authored-by: Eddycrack 864 <Eddycrack864@users.noreply.huggingface.co>

app.py

@@ -2,7 +2,10 @@ 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 spaces
 
 roformer_models = {
@@ -121,6 +124,26 @@ demucs_overlap_values = [
     '0.99',
 ]
 
+@spaces.GPU(duration=300)
+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
+
 @spaces.GPU(duration=300)
 def roformer_separator(roformer_audio, roformer_model, roformer_output_format, roformer_overlap, roformer_segment_size):
   files_list = []
@@ -286,6 +309,23 @@ with gr.Blocks(theme="NoCrypt/miku@1.2.2", title="🎵 UVR5 UI 🎵") as app:
                     type = "numpy",
                     interactive = True
                 )
+            with gr.Accordion("Separation by Link", open = False):
+                with gr.Row():
+                    roformer_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():
+                    roformer_download_button = gr.Button(
+                    "Download!",
+                    variant = "primary"
+                )
+
+            roformer_download_button.click(download_audio, [roformer_link], [roformer_audio])
+
             with gr.Row():
                 roformer_button = gr.Button("Separate!", variant = "primary")
             with gr.Row():
@@ -347,6 +387,23 @@ with gr.Blocks(theme="NoCrypt/miku@1.2.2", title="🎵 UVR5 UI 🎵") as app:
                     type = "numpy",
                     interactive = True
                 )
+            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"
+                )
+
+            mdx23c_download_button.click(download_audio, [mdx23c_link], [mdx23c_audio])
+
             with gr.Row():
                 mdx23c_button = gr.Button("Separate!", variant = "primary")
             with gr.Row():
@@ -405,6 +462,23 @@ with gr.Blocks(theme="NoCrypt/miku@1.2.2", title="🎵 UVR5 UI 🎵") as app:
                     type = "numpy",
                     interactive = True
                 )
+            with gr.Accordion("Separation by Link", open = False):
+                with gr.Row():
+                    mdxnet_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():
+                    mdxnet_download_button = gr.Button(
+                    "Download!",
+                    variant = "primary"
+                )
+
+            mdxnet_download_button.click(download_audio, [mdxnet_link], [mdxnet_audio])
+
             with gr.Row():
                 mdxnet_button = gr.Button("Separate!", variant = "primary")
             with gr.Row():
@@ -471,6 +545,23 @@ with gr.Blocks(theme="NoCrypt/miku@1.2.2", title="🎵 UVR5 UI 🎵") as app:
                     type = "numpy",
                     interactive = True
                 )
+            with gr.Accordion("Separation by Link", open = False):
+                with gr.Row():
+                   vrarch_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():
+                    vrarch_download_button = gr.Button(
+                    "Download!",
+                    variant = "primary"
+                )
+
+            vrarch_download_button.click(download_audio, [vrarch_link], [vrarch_audio])
+
             with gr.Row():
                 vrarch_button = gr.Button("Separate!", variant = "primary")
             with gr.Row():
@@ -523,6 +614,23 @@ with gr.Blocks(theme="NoCrypt/miku@1.2.2", title="🎵 UVR5 UI 🎵") as app:
                     type = "numpy",
                     interactive = True
                 )
+            with gr.Accordion("Separation by Link", open = False):
+                with gr.Row():
+                    demucs_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():
+                    demucs_download_button = gr.Button(
+                    "Download!",
+                    variant = "primary"
+                )
+
+            demucs_download_button.click(download_audio, [demucs_link], [demucs_audio])
+
             with gr.Row():
                 demucs_button = gr.Button("Separate!", variant = "primary")
             with gr.Row():
@@ -563,7 +671,8 @@ with gr.Blocks(theme="NoCrypt/miku@1.2.2", title="🎵 UVR5 UI 🎵") as app:
               * Special thanks to [Ilaria](https://github.com/TheStingerX) for hosting this space and help.
               * Thanks to [Mikus](https://github.com/cappuch) for the help with the code.
               * Thanks to [Nick088](https://huggingface.co/Nick088) for the help to fix roformers.
-              * Improvements by [Blane187](https://huggingface.co/Blane187).
+              * Thanks to [yt_dlp](https://github.com/yt-dlp/yt-dlp) devs.
+              * Separation by link source code and improvements by [Blane187](https://huggingface.co/Blane187).
 
               You can donate to the original UVR5 project here:
               [!["Buy Me A Coffee"](https://www.buymeacoffee.com/assets/img/custom_images/orange_img.png)](https://www.buymeacoffee.com/uvr5)

requirements.txt

@@ -1,3 +1,4 @@
 numpy==1.25.2
 audio-separator[gpu]==0.17.5
-scipy
+scipy
+yt_dlp