Upload infer_new.py

infer_new.py

@@ -105,16 +105,26 @@ def clean():
         gr.Slider.update(visible=False)
     )
 
+def get_file_base_name(file_path):
+    # Extract the base name (including extension)
+    base_name = os.path.basename(file_path)
+    # Split the base name into the name and extension, and return just the name
+    file_name_without_extension, _ = os.path.splitext(base_name)
+    return file_name_without_extension
 
 
 def api_convert_voice(spk_id,voice_transform,input_audio_path):
     
     #split audio
+    base_name = get_file_base_name(input_audio_path)
     cut_vocal_and_inst(input_audio_path,spk_id)
     print("audio splitting performed")
-    vocal_path = f"output/{split_model}/{spk_id}_input_audio/vocals.wav"
-    inst = f"output/{split_model}/{spk_id}_input_audio/no_vocals.wav"
+    #vocal_path = f"output/{split_model}/{spk_id}_input_audio/vocals.wav"
+    #inst = f"output/{split_model}/{spk_id}_input_audio/no_vocals.wav"
     
+    vocal_path = f"output/{base_name}/vocals.wav"
+    inst = f"output/{base_name}/{spk_id}_input_audio/no_vocals.wav"
+
     output_path = convert_voice(spk_id, vocal_path, voice_transform)
     output_path1= combine_vocal_and_inst(output_path,inst)
     print(output_path1)
@@ -325,7 +335,7 @@ def get_vc(sid, to_return_protect0):
 
 def cut_vocal_and_inst(audio_path,spk_id):
     
-    vocal_path = "output/result/audio.wav"
+    #vocal_path = "output/result/audio.wav"
     os.makedirs("output/result", exist_ok=True)
     #wavfile.write(vocal_path, audio_data[0], audio_data[1])
     #logs.append("Starting the audio splitting process...")