cefr_level

app.py

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