Update src/main.py

src/main.py

@@ -277,15 +277,15 @@ def vocal_only_pipeline(song_input, voice_model, pitch_change, is_webui=0,index_
                 orig_song_path, vocals_path = preprocess_vocals_only(song_input, mdx_model_params, song_id, is_webui, input_type, progress)
         else:
             vocals_path = None
-            for file in os.listdir(song_dir):
-                vocals_path = os.path.join(song_dir, file)
-            return vocals_path
+            #for file in os.listdir(song_dir):
+            #    vocals_path = os.path.join(song_dir, file)
+            #return vocals_path
          
-            # if any of the audio files aren't available or keep intermediate files, rerun preprocess
-            #if any(path is None for path in paths):
-            #    orig_song_path, vocals_path= preprocess_vocals_only(song_input, mdx_model_params, song_id, is_webui, input_type, progress)
-            #else:
-            #    orig_song_path = paths
+            #if any of the audio files aren't available or keep intermediate files, rerun preprocess
+            if any(path is None for path in paths):
+                orig_song_path, vocals_path= preprocess_vocals_only(song_input, mdx_model_params, song_id, is_webui, input_type, progress)
+            else:
+                orig_song_path = paths
         #orig_song_path, vocals_path = preprocess_vocals_only(song_input, mdx_model_params, song_id, is_webui, input_type, progress)
         
         pitch_change = pitch_change * 12 + pitch_change_all
@@ -295,10 +295,10 @@ def vocal_only_pipeline(song_input, voice_model, pitch_change, is_webui=0,index_
         
         display_progress('[~] Converting voice using RVC...', 0.5, is_webui, progress)
         voice_change(voice_model, vocals_path, ai_vocals_path, pitch_change, f0_method, index_rate, filter_radius, rms_mix_rate, protect, crepe_hop_length, is_webui)
-        display_progress('[~] Applying audio effects to Vocals...', 0.8, is_webui, progress)
-        ai_vocals_mixed_path = add_audio_effects(ai_vocals_path, reverb_rm_size, reverb_wet, reverb_dry, reverb_damping)
+        #display_progress('[~] Applying audio effects to Vocals...', 0.8, is_webui, progress)
+        #ai_vocals_mixed_path = add_audio_effects(ai_vocals_path, reverb_rm_size, reverb_wet, reverb_dry, reverb_damping)
 
-        return ai_vocals_mixed_path
+        return ai_vocals_path
     except Exception as e:
         raise_exception(str(e), is_webui)