Update src/main.py

src/main.py

@@ -187,7 +187,7 @@ def preprocess_song(song_input, mdx_model_params, song_id, is_webui, input_type,
     #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
+    return orig_song_path, vocals_path, instrumentals_path, main_vocals_path, backup_vocals_path, main_vocals_dereverb_path
 
 
 def voice_change(voice_model, vocals_path, output_path, pitch_change, f0_method, index_rate, filter_radius, rms_mix_rate, protect, crepe_hop_length, is_webui):
@@ -278,9 +278,9 @@ def song_cover_pipeline(song_input, voice_model, pitch_change, keep_files,
 
             # if any of the audio files aren't available or keep intermediate files, rerun preprocess
             if any(path is None for path in paths) or keep_files:
-                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:
                 print("ENTRATO ELSE, ERRORE!")
                 orig_song_path, instrumentals_path, main_vocals_dereverb_path, backup_vocals_path = paths