1

app.py

@@ -11,12 +11,10 @@ def generate_dialogue(rounds, method, role1, role2):
 
 def main_function(rounds: int, method: str, role1: str, role2: str):
     structured_dialogue = generate_dialogue(rounds, method, role1, role2)
-    
-    chatbot_dialogue = structured_dialogue  # 使用结构化对话
-    
-    audio_path = dialogue_to_audio([(item["role"], item["content"]) for item in structured_dialogue])
+    chatbot_dialogue = structured_dialogue
+    audio_path = dialogue_to_audio(structured_dialogue)
     json_output = json.dumps({"dialogue": structured_dialogue}, ensure_ascii=False, indent=4)
-    
+
     # 儲存對話為 JSON 文件
     file_name = "dialogue_output.txt"
     with open(file_name, "w", encoding="utf-8") as f:
@@ -36,12 +34,12 @@ if __name__ == "__main__":
         main_function, 
         [
             gr.components.Slider(minimum=2, maximum=6, step=2, label="對話輪數"),
-            gr.components.Dropdown(choices=["auto", "manual"], default="auto", label="生成方式"),
-            gr.components.Textbox(label="角色 1 名稱", default="A"),
-            gr.components.Textbox(label="角色 2 名稱", default="B"),
+            gr.components.Dropdown(choices=["auto", "manual"], label="生成方式"),
+            gr.components.Textbox(label="角色 1 名稱"),
+            gr.components.Textbox(label="角色 2 名稱"),
         ], 
         [
-            gr.components.Chatbot(roles=["A", "B"], label="生成的對話"),  # 确保此处的角色与默认值相匹配
+            gr.components.Chatbot(label="生成的對話"),
             gr.components.Audio(type="filepath", label="對話朗讀"),
             gr.components.File(label="下載對話 JSON 文件")
         ]