Update src/main.py

src/main.py

@@ -301,7 +301,65 @@ def vocal_only_pipeline(song_input, voice_model, pitch_change, is_webui=0,index_
     except Exception as e:
         raise_exception(str(e), is_webui)
 
+# Function defining the pipeline to create the AI cover of an audio containing only voice information.
+# Returns the path of the cover vocal.
+def vocal_only_pipeline_no_reverb(song_input, voice_model, pitch_change, is_webui=0,index_rate=0.5, filter_radius=3,
+                        rms_mix_rate=0.25, f0_method='rmvpe', crepe_hop_length=128, protect=0.33, pitch_change_all=0,
+                        reverb_rm_size=0.15, reverb_wet=0.2, reverb_dry=0.8, reverb_damping=0.7, output_format='mp3',progress=gr.Progress()):
+    try:
+        #Load mdx parameters
+        with open(os.path.join(mdxnet_models_dir, 'model_data.json')) as infile:
+            mdx_model_params = json.load(infile)
+
+        print("HERE pre song_id")
+        # Get path of loaded vocal file
+        input_type = 'local'
 
+        # Extract the file name without extension
+        name, extension = os.path.splitext(song_input)
+        
+        # Append '_noreverb' to the file name
+        song_input = f"{name}_noreverb{extension}"
+        song_input = song_input.strip('\"')
+        if os.path.exists(song_input):
+            song_id = get_hash(song_input)
+        else:
+            error_msg = f'{song_input} does not exist.'
+            song_id = None
+            raise_exception(error_msg, is_webui)
+        print("HERE pre song_dir")
+        song_dir = os.path.join(output_dir, song_id, voice_model)
+
+        print("HERE pre IF")
+        if not os.path.exists(song_dir):
+                os.makedirs(song_dir)
+                #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
+         
+            #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
+        ai_vocals_path = os.path.join(song_dir, f'_{voice_model}_p{pitch_change}_i{index_rate}_fr{filter_radius}_rms{rms_mix_rate}_pro{protect}_{f0_method}{"" if f0_method != "mangio-crepe" else f"_{crepe_hop_length}"}_covervocals.wav')
+        display_progress(f'[~] Skipped Reverb {ai_vocals_path}', 0.5, is_webui, progress)
+        time.sleep(10)
+        
+        display_progress('[~] Converting voice using RVC...', 0.5, is_webui, progress)
+        voice_change(voice_model, song_input, 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)
+
+        return ai_vocals_path
+    except Exception as e:
+        raise_exception(str(e), is_webui)
 
 def song_cover_pipeline(song_input, voice_model, pitch_change, keep_files,
                         is_webui=0, main_gain=0, backup_gain=0, inst_gain=0, index_rate=0.5, filter_radius=3,