Update src/main.py

src/main.py

@@ -181,11 +181,11 @@ def preprocess_song(song_input, mdx_model_params, song_id, is_webui, input_type,
     display_progress('[~] Separating Vocals from Instrumental...', 0.1, is_webui, progress)
     vocals_path, instrumentals_path = run_mdx(mdx_model_params, song_output_dir, os.path.join(mdxnet_models_dir, 'UVR-MDX-NET-Voc_FT.onnx'), orig_song_path, denoise=True, keep_orig=keep_orig, _stemname1 = "_origvocals", _stemname2="_originstr")
 
-    #display_progress('[~] Separating Main Vocals from Backup Vocals...', 0.2, is_webui, progress)
-    #backup_vocals_path, main_vocals_path = run_mdx(mdx_model_params, song_output_dir, os.path.join(mdxnet_models_dir, 'UVR_MDXNET_KARA_2.onnx'), vocals_path, suffix='Backup', invert_suffix='Main', denoise=True)
+    display_progress('[~] Separating Main Vocals from Backup Vocals...', 0.2, is_webui, progress)
+    backup_vocals_path, main_vocals_path = run_mdx(mdx_model_params, song_output_dir, os.path.join(mdxnet_models_dir, 'UVR_MDXNET_KARA_2.onnx'), vocals_path, suffix='Backup', invert_suffix='Main', denoise=True)
 
-    #display_progress('[~] Applying DeReverb to Vocals...', 0.3, is_webui, progress)
-    #_, main_vocals_dereverb_path = run_mdx(mdx_model_params, song_output_dir, os.path.join(mdxnet_models_dir, 'Reverb_HQ_By_FoxJoy.onnx'), main_vocals_path, invert_suffix='DeReverb', exclude_main=True, denoise=True)
+    display_progress('[~] Applying DeReverb to Vocals...', 0.3, is_webui, progress)
+    _, main_vocals_dereverb_path = run_mdx(mdx_model_params, song_output_dir, os.path.join(mdxnet_models_dir, 'Reverb_HQ_By_FoxJoy.onnx'), main_vocals_path, invert_suffix='DeReverb', exclude_main=True, denoise=True)
 
     return orig_song_path, vocals_path, instrumentals_path, main_vocals_path, backup_vocals_path, main_vocals_dereverb_path
 
@@ -268,9 +268,9 @@ def song_cover_pipeline(song_input, voice_model, pitch_change, keep_files,
 
         if not os.path.exists(song_dir):
             os.makedirs(song_dir)
-            orig_song_path, vocals_path, instrumentals_path= preprocess_song(song_input, mdx_model_params, song_id, is_webui, input_type, progress)
-            print(vocals_path, instrumentals_path)
-            return vocals_path, instrumentals_path
+            orig_song_path, vocals_path, instrumentals_path, backup_vocals_path, main_vocals_dereverb_path= preprocess_song(song_input, mdx_model_params, song_id, is_webui, input_type, progress)
+            print(vocals_path, instrumentals_path, backup_vocals_path, main_vocals_dereverb_path)
+            return vocals_path, instrumentals_path, backup_vocals_path, main_vocals_dereverb_path
 
         else:
             vocals_path, main_vocals_path = None, None