7

app.py

@@ -3,29 +3,26 @@ from gtts import gTTS
 import json
 
 def generate_dialogue(rounds, method, role1, role2):
-    # 這裡只是一個示例，你可能需要更高級的生成方式
     if method == "auto":
-        dialogue = []
-        for i in range(rounds):
-            dialogue.append((role1 if i % 2 == 0 else role2, f"自動文本 {i+1}"))
+        dialogue = [{"role": role1 if i % 2 == 0 else role2, "content": f"自動文本 {i+1}"} for i in range(rounds)]
     else:
-        dialogue = [(role1, "手動輸入文本 1"), (role2, "手動輸入文本 2")]
+        dialogue = [{"role": role1, "content": "手動輸入文本 1"}, {"role": role2, "content": "手動輸入文本 2"}]
     return dialogue
 
-
 def main_function(rounds: int, method: str, role1: str, role2: str):
-    dialogue = generate_dialogue(rounds, method, role1, role2)
+    structured_dialogue = generate_dialogue(rounds, method, role1, role2)
+    
+    chatbot_dialogue = [(item["role"], item["content"]) for item in structured_dialogue]
     
-    audio_path = dialogue_to_audio(dialogue)
-    json_output = json.dumps({"dialogue": dialogue}, ensure_ascii=False, indent=4)
+    audio_path = dialogue_to_audio(chatbot_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:
         f.write(json_output)
 
-    return dialogue, audio_path, file_name
-
+    return chatbot_dialogue, audio_path, file_name
 
 def dialogue_to_audio(dialogue):
     text = " ".join([item[1] for item in dialogue])