Upload infertest.py

infertest.py

@@ -121,10 +121,10 @@ def api_convert_voice():
     file.save(input_audio_path)
     
     #split audio
-    cut_vocal_and_inst(input_audio_path)
+    cut_vocal_and_inst(input_audio_path,spk_id)
     print("audio splitting performed")
-    vocal_path = f"output/{split_model}/audio/vocals.wav"
-    inst = f"output/{split_model}/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"
     
     output_path = convert_voice(spk_id, vocal_path, voice_transform)
     output_path= combine_vocal_and_inst(output_path,inst)
@@ -436,7 +436,7 @@ def download_audio(url, audio_provider):
         logs.append("Download Complete.")
         yield audio_path, "\n".join(logs)
 
-def cut_vocal_and_inst_yt(split_model):
+def cut_vocal_and_inst_yt(split_model,spk_id):
     logs = []
     logs.append("Starting the audio splitting process...")
     yield "\n".join(logs), None, None, None
@@ -446,12 +446,12 @@ def cut_vocal_and_inst_yt(split_model):
         logs.append(line)
         yield "\n".join(logs), None, None, None
     print(result.stdout)
-    vocal = f"output/{split_model}/audio/vocals.wav"
-    inst = f"output/{split_model}/audio/no_vocals.wav"
+    vocal = f"output/{split_model}/{spk_id}_input_audio/vocals.wav"
+    inst = f"output/{split_model}/{spk_id}_input_audio/no_vocals.wav"
     logs.append("Audio splitting complete.")
     yield "\n".join(logs), vocal, inst, vocal
 
-def cut_vocal_and_inst(audio_path):
+def cut_vocal_and_inst(audio_path,spk_id):
     
     vocal_path = "output/result/audio.wav"
     os.makedirs("output/result", exist_ok=True)
@@ -466,8 +466,8 @@ def cut_vocal_and_inst(audio_path):
     #    logs.append(line)
     #    yield "\n".join(logs), None, None
     print(result.stdout)
-    vocal = f"output/{split_model}/audio/vocals.wav"
-    inst = f"output/{split_model}/audio/no_vocals.wav"
+    vocal = f"output/{split_model}/{spk_id}_input_audio/vocals.wav"
+    inst = f"output/{split_model}/{spk_id}_input_audio/no_vocals.wav"
     #logs.append("Audio splitting complete.")