Update rvc.py

rvc.py

@@ -31,6 +31,34 @@ model_library = CachedModels()
 
 
 
+
+
+
+
+def roformer_separator(roformer_audio, roformer_output_format="wav", roformer_overlap="4", roformer_segment_size="256"):
+  files_list = []
+  files_list.clear()
+  directory = "./audios"
+  random_id = str(random.randint(10000, 99999))
+  pattern = f"{random_id}"
+  os.makedirs("outputs", exist_ok=True)
+  write(f'{random_id}.wav', roformer_audio[0], roformer_audio[1])
+  full_roformer_model = "model_bs_roformer_ep_317_sdr_12.9755.ckpt"
+  prompt = f"audio-separator {random_id}.wav --model_filename {full_roformer_model} --output_dir=./outputs --output_format={roformer_output_format} --normalization=0.9 --mdxc_overlap={roformer_overlap} --mdxc_segment_size={roformer_segment_size}"
+  os.system(prompt)
+
+  for file in os.listdir(directory):
+    if re.search(pattern, file):
+      files_list.append(os.path.join(directory, file))
+
+  stem1_file = files_list[0]
+  stem2_file = files_list[1]
+
+  return stem1_file, stem2_file
+
+
+
+
 with gr.Blocks(title="🔊 Nex RVC Mobile",theme=gr.themes.Base()) as app:
 
     gr.Markdown("# Nex RVC MOBILE GUI")
@@ -91,16 +119,66 @@ with gr.Blocks(title="🔊 Nex RVC Mobile",theme=gr.themes.Base()) as app:
 
                             with gr.Row():
 
-                                tts_text = gr.Textbox(label="Text to Speech", placeholder="Enter text to convert to speech")
-
-                            with gr.Row():
-
-                                tts_lang = gr.Radio(choices=["en","es","it","pt"],label="",value="en")
+                                roformer_audio = gr.Audio(
+                                    
+                                    label = "Input Audio",
+
+                                    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():
 
-                                tts_button = gr.Button(value="Speak", variant="primary")
-
+                                roformer_button = gr.Button("Separate!", variant = "primary")
+                            with gr.Column():
+                                roformer_stem1 = gr.Audio(
+                                    
+                                    show_download_button = True,
+                    
+                                    interactive = False,
+                                    
+                                    label = "Stem 1",
+                    
+                                    type = "filepath"
+                
+                                )
+                
+                                roformer_stem2 = gr.Audio(
+                    
+                                    show_download_button = True,
+                    
+                                    interactive = False,
+                    
+                                    label = "Stem 2",
+                    
+                                    type = "filepath"
+                
+                                )
+                                
                     with gr.Row():
 
                         paths_for_files = lambda path:[os.path.abspath(os.path.join(path, f)) for f in os.listdir(path) if os.path.splitext(f)[1].lower() in ('.mp3', '.wav', '.flac', '.ogg')]
@@ -147,23 +225,8 @@ with gr.Blocks(title="🔊 Nex RVC Mobile",theme=gr.themes.Base()) as app:
 
                             outputs=[input_audio0])
 
-                        tts_button.click(
-
-                            fn=whisperspeak,
-
-                            inputs=[tts_text,tts_lang],
-
-                            outputs=[input_audio0],
-
-                            show_progress=True)
-
-                        tts_button.click(
-
-                            fn=lambda: {"choices":paths_for_files('audios'),"__type__":"update"},
-
-                            inputs=[],
-
-                            outputs=[input_audio0])
+                        roformer_button.click(roformer_separator, [roformer_audio], [roformer_stem1, roformer_stem2])
+        
 
                 with gr.Column():