gr.components.Dropdown(choices=["中文", "英文"], label="語言"),

app.py

@@ -50,12 +50,12 @@ def extract_json_from_response(response_text):
         raise ValueError("JSON dialogue not found in the response.")
 
 
-def create_chat_dialogue(rounds, role1, role2, theme="購物"):
+def create_chat_dialogue(rounds, role1, role2, theme="購物", language):
     openai.api_key = os.environ["OPEN_AI_KEY"]
 
     # 初始化對話
     sentenses_count = int(rounds) * 2
-    sys_content = f"你是一個英文家教，請用英文生成對話"
+    sys_content = f"你是一個{language}家教，請用{language}生成對話"
     prompt = f"您將進行一場以{theme}為主題的對話。{role1}和{role2}將是參與者。請依次交談{rounds}輪。(1輪對話的定義是 {role1} 和 {role2} 各說一句話，總共 {sentenses_count} 句話。)以json格式儲存對話。並回傳對話JSON文件。格式為：[{{role:\"{role1}\", content: \".....\"}}, {{role:\"{role2}\", content: \".....\"}}]"
     messages = [
         {"role": "system", "content": sys_content},
@@ -84,18 +84,18 @@ def create_chat_dialogue(rounds, role1, role2, theme="購物"):
     return dialogue
 
 
-def generate_dialogue(rounds, method, role1, role2, theme):
+def generate_dialogue(rounds, method, role1, role2, theme, language):
     if method == "auto":
         dialogue = create_chat_dialogue(rounds, role1, role2, theme)
     else:
         dialogue = [{"role": role1, "content": "手動輸入文本 1"}, {"role": role2, "content": "手動輸入文本 2"}]
     return dialogue
 
-def main_function(password: str, theme: str, method: str, rounds: int, role1: str, role2: str):
+def main_function(password: str, theme: str, language: str, method: str, rounds: int, role1: str, role2: str):
     if password != os.environ.get("PASSWORD", ""):
         return "错误的密码，请重新输入。", "", ""
 
-    structured_dialogue = generate_dialogue(rounds, method, role1, role2, theme)
+    structured_dialogue = generate_dialogue(rounds, method, role1, role2, theme, language)
 
     # Convert structured dialogue for Chatbot component to show "role1: content1" and "role2: content2" side by side
     chatbot_dialogue = []
@@ -130,6 +130,7 @@ if __name__ == "__main__":
         [
             gr.components.Textbox(label="输入密码", type="password"),
             gr.components.Textbox(label="對話主題"),  # 加入 theme 的輸入框，設定預設值為 '購物'
+            gr.components.Dropdown(choices=["中文", "英文"], label="語言"),
             gr.components.Dropdown(choices=["auto", "manual"], label="生成方式"),
             gr.components.Slider(minimum=2, maximum=6, step=2, label="對話輪數"),
             gr.components.Textbox(label="角色 1 名稱"),