Update lib/tts_infer.py

lib/tts_infer.py

@@ -4,6 +4,7 @@ import gc
 import torch
 from multiprocessing import cpu_count
 from lib.modules import VC
+from lib.language_tts import language_dict
 from lib.split_audio import split_silence_nonsilent, adjust_audio_lengths, combine_silence_nonsilent
 import edge_tts
 import tempfile
@@ -109,7 +110,8 @@ def get_model(voice_model):
 
 def infer_audio(
     model_name,
-    audio_path,
+    text,
+    language_code,
     f0_change=0,
     f0_method="rmvpe",
     min_pitch="50",
@@ -140,6 +142,17 @@ def infer_audio(
     vc = VC(configs)
     pth_path, index_path = get_model(model_name)
     vc_data = vc.get_vc(pth_path, protect, 0.5)
+
+
+    voice = language_dict.get(language_code, "default_voice")
+    communicate = edge_tts.Communicate(text, voice)
+    with tempfile.NamedTemporaryFile(delete=False, suffix=".wav") as tmp_file:
+        tmp_path = tmp_file.name
+        await communicate.save(tmp_path)
+
+
+
+
     
     if split_infer:
         inferred_files = []
@@ -195,7 +208,7 @@ def infer_audio(
     else:
         inference_info, audio_data, output_path = vc.vc_single(
             0,
-            audio_path,
+            audio_path=tmp_path,
             f0_change,
             f0_method,
             index_path,