Update app.py

app.py

@@ -342,6 +342,34 @@ def get_bert_final(phones, word2ph, norm_text,language,device):
         bert = torch.zeros((1024, len(phones))).to(device)
     return bert
 
+def get_ref_path_decor(func):
+    # 为了hg部署添加的装饰函数，将参考文本的内容改为路径
+    def inner(*args):
+        ref_wav_path = text_to_audio_mappings.get(args[0], "")
+        if not ref_wav_path:
+            logger.warn("Audio file not found for the selected text.")
+            return
+        func(ref_wav_path, *args[1,])
+
+def load_audio_text_mappings(folder_path, list_file_name):
+    """ 用于hg部署，扫描指定文件夹获取音频文件列表 """
+    text_to_audio_mappings = {}
+    audio_to_text_mappings = {}
+    with open(os.path.join(folder_path, list_file_name), 'r', encoding='utf-8') as file:
+        for line in file:
+            parts = line.strip().split('|')
+            if len(parts) >= 4:
+                audio_file_name = parts[0]
+                text = parts[3]
+                audio_file_path = os.path.join(folder_path, audio_file_name)
+                text_to_audio_mappings[text] = audio_file_path
+                audio_to_text_mappings[audio_file_path] = text
+    return text_to_audio_mappings, audio_to_text_mappings
+
+audio_folder_path = 'audio'
+text_to_audio_mappings, audio_to_text_mappings = load_audio_text_mappings(audio_folder_path, 'slicer_opt.list')
+
+@get_ref_path_decor
 def get_tts_wav(ref_wav_path, prompt_text, prompt_language, text, text_language, how_to_cut=i18n("不切")):
     t0 = ttime()
     prompt_text = prompt_text.strip("\n")
@@ -566,23 +594,7 @@ def get_weights_names():
 SoVITS_names, GPT_names = get_weights_names()
 
 
-def load_audio_text_mappings(folder_path, list_file_name):
-    """ 用于hg部署，扫描指定文件夹获取音频文件列表 """
-    text_to_audio_mappings = {}
-    audio_to_text_mappings = {}
-    with open(os.path.join(folder_path, list_file_name), 'r', encoding='utf-8') as file:
-        for line in file:
-            parts = line.strip().split('|')
-            if len(parts) >= 4:
-                audio_file_name = parts[0]
-                text = parts[3]
-                audio_file_path = os.path.join(folder_path, audio_file_name)
-                text_to_audio_mappings[text] = audio_file_path
-                audio_to_text_mappings[audio_file_path] = text
-    return text_to_audio_mappings, audio_to_text_mappings
 
-audio_folder_path = 'audio'
-text_to_audio_mappings, audio_to_text_mappings = load_audio_text_mappings(audio_folder_path, 'slicer_opt.list')
 
 with gr.Blocks(title=f"GPT-SoVITS WebUI") as app:
     gr.Markdown(