chatbot_dialogue

app.py

@@ -11,8 +11,16 @@ 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]
-    print(chatbot_dialogue)
+    
+    # Convert structured dialogue for Chatbot component to show "role1: content1" and "role2: content2" side by side
+    chatbot_dialogue = []
+    for i in range(0, len(structured_dialogue), 2):  # We iterate with a step of 2 to take pairs
+        # Get the content for the two roles in the pair
+        role1_content = f"{structured_dialogue[i]['role']}: {structured_dialogue[i]['content']}"
+        role2_content = f"{structured_dialogue[i+1]['role']}: {structured_dialogue[i+1]['content']}" if i+1 < len(structured_dialogue) else ""
+        
+        chatbot_dialogue.append((role1_content, role2_content))
+
     audio_path = dialogue_to_audio(structured_dialogue)
     json_output = json.dumps({"dialogue": structured_dialogue}, ensure_ascii=False, indent=4)
 
@@ -23,6 +31,7 @@ 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([f"{item['role']}: {item['content']}" for item in dialogue])
     tts = gTTS(text=text, lang='zh-tw')