Update app.py

app.py

@@ -1,17 +1,12 @@
 import os
 
 def list_files_tree(directory, indent=""):
-    # 获取当前目录下的所有文件和文件夹
     items = os.listdir(directory)
     for i, item in enumerate(items):
-        # 定义前缀，最后一个文件或文件夹使用不同的前缀
         prefix = "└── " if i == len(items) - 1 else "├── "
-        # 打印文件或文件夹
         print(indent + prefix + item)
-        # 如果是文件夹，则递归调用
         item_path = os.path.join(directory, item)
         if os.path.isdir(item_path):
-            # 如果是最后一个子文件夹，使用缩进
             next_indent = indent + ("    " if i == len(items) - 1 else "│   ")
             list_files_tree(item_path, next_indent)
 
@@ -19,7 +14,8 @@ from huggingface_hub import snapshot_download
 print("Models...")
 models_id = """None1145/GPT-SoVITS-Lappland-the-Decadenza
 None1145/GPT-SoVITS-Theresa
-None1145/GPT-SoVITS-Vulpisfoglia"""
+None1145/GPT-SoVITS-Vulpisfoglia
+None1145/GPT-SoVITS-Theresa-Recording"""
 for model_id in models_id.split("\n"):
     if model_id in ["", " "]:
         break
@@ -93,7 +89,6 @@ def load_model(sovits_path, gpt_path):
         def __init__(self, input_dict):
             for key, value in input_dict.items():
                 if isinstance(value, dict):
-                    # 如果值是字典，递归调用构造函数
                     setattr(self, key, DictToAttrRecursive(value))
                 else:
                     setattr(self, key, value)
@@ -312,11 +307,17 @@ with gr.Blocks() as app:
                         '</div>')
                 with gr.Row():
                     with gr.Column():
-                        prompt_text = gr.Dropdown(
-                            label="Transcript of the Reference Audio",
-                            value=example_reference,
-                            choices=list(transcript.keys())
-                        )
+                        with gr.Column():
+                            prompt_text = gr.Dropdown(
+                                label="Transcript of the Reference Audio",
+                                value=example_reference,
+                                choices=list(transcript.keys())
+                            )
+                            prompt_language = gr.Dropdown(
+                                label="Language of the Reference Audio",
+                                value="zh",
+                                choices=["zh", "ja", "en"]
+                            )
                         inp_ref_audio = gr.Audio(
                             label="Reference Audio",
                             type="filepath",
@@ -329,9 +330,17 @@ with gr.Blocks() as app:
                             inputs=[prompt_text, transcripts_state],
                             outputs=[inp_ref_audio]
                         )
-                        prompt_language = gr.State(value="zh")
+                        # prompt_language = gr.State(value="zh")
                     with gr.Column():
-                        text = gr.Textbox(label="Input Text", value="你好")
+                        if prompt_language.value == "zh":
+                            initial_text = "你好"
+                        elif prompt_language.value == "ja":
+                            initial_text = "こんにちは"
+                        elif prompt_language.value == "en":
+                            initial_text = "Hello"
+                        else:
+                            initial_text = "你好"
+                        text = gr.Textbox(label="Input Text", value=initial_text)
                         text_language = gr.Dropdown(
                             label="Language",
                             choices=["zh", "en", "ja"],