9

app.py

@@ -12,9 +12,9 @@ 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 = [(item["role"], item["content"]) for item in structured_dialogue]
+    chatbot_dialogue = structured_dialogue  # 使用结构化对话
     
-    audio_path = dialogue_to_audio(chatbot_dialogue)
+    audio_path = dialogue_to_audio([(item["role"], item["content"]) for item in structured_dialogue])
     json_output = json.dumps({"dialogue": structured_dialogue}, ensure_ascii=False, indent=4)
     
     # 儲存對話為 JSON 文件
@@ -25,8 +25,8 @@ def main_function(rounds: int, method: str, role1: str, role2: str):
     return chatbot_dialogue, audio_path, file_name
 
 def dialogue_to_audio(dialogue):
-    text = " ".join([item[1] for item in dialogue])
-    tts = gTTS(text=text, lang='en')
+    text = " ".join([f"{item['role']}: {item['content']}" for item in dialogue])
+    tts = gTTS(text=text, lang='zh-tw')
     file_path = "temp_audio.mp3"
     tts.save(file_path)
     return file_path
@@ -41,7 +41,7 @@ if __name__ == "__main__":
             gr.components.Textbox(label="角色 2 名稱", default="B"),
         ], 
         [
-            gr.components.Chatbot(label="生成的對話"),
+            gr.components.Chatbot(roles=["A", "B"], label="生成的對話"),  # 确保此处的角色与默认值相匹配
             gr.components.Audio(type="filepath", label="對話朗讀"),
             gr.components.File(label="下載對話 JSON 文件")
         ]