6

app.py

@@ -2,16 +2,17 @@ import gradio as gr
 from gtts import gTTS
 import json
 
-def generate_dialogue(rounds: int, method: str, role1: str, role2: str) -> list:
-    roles = [role1, role2]
-    dialogue = []
-    for i in range(rounds):
-        if method == "auto":
-            dialogue.append((roles[i % 2], f"自動文本 {i + 1}"))
-        else:
-            dialogue.append((roles[i % 2], ""))
+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}"))
+    else:
+        dialogue = [(role1, "手動輸入文本 1"), (role2, "手動輸入文本 2")]
     return dialogue
 
+
 def main_function(rounds: int, method: str, role1: str, role2: str):
     dialogue = generate_dialogue(rounds, method, role1, role2)
     
@@ -36,10 +37,10 @@ if __name__ == "__main__":
     gr.Interface(
         main_function, 
         [
-            gr.components.Slider(minimum=2, maximum=6, step=2, default=2, label="對話輪數"),
-            gr.components.Dropdown(choices=["auto", "manual"], label="生成方式"),
-            gr.components.Textbox(default="A", label="角色 1 名稱"),
-            gr.components.Textbox(default="B", label="角色 2 名稱"),
+            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.Chatbot(label="生成的對話"),