Update app.py

app.py

@@ -42,20 +42,24 @@ LEARNING_MODES = [
 LEARNING_MODE_INSTRUCTIONS = {
     "Concept Explainer": (
         "Explain concepts step by step. Use clear definitions, key formulas or structures, "
-        "and one or two simple examples. Focus on clarity over depth."
+        "and one or two simple examples. Focus on clarity over depth. Regularly check if "
+        "the student is following."
     ),
     "Socratic Tutor": (
-        "Use a Socratic style. Ask the student short questions first, guide them to reason "
-        "step by step, and only give full explanations after they try."
+        "Use a Socratic style. Ask the student ONE short question at a time, guide them to "
+        "reason step by step, and only give full explanations after they try. Prioritize "
+        "questions and hints over long lectures."
     ),
     "Exam Prep / Quiz": (
         "Behave like an exam prep coach. Often propose short quiz-style questions "
-        "(multiple choice or short answer), then explain the solutions."
+        "(multiple choice or short answer), then explain the solutions clearly. Emphasize "
+        "common traps and how to avoid them."
     ),
     "Assignment Helper": (
-        "Help with assignments without giving full final solutions. Clarify requirements, "
-        "break tasks into smaller steps, and provide hints or partial examples instead of "
-        "complete code or final answers."
+        "Help with assignments WITHOUT giving full final solutions. Clarify requirements, "
+        "break tasks into smaller steps, and provide hints, partial examples, or pseudo-code "
+        "instead of complete code or final answers. Encourage the student to attempt each "
+        "step before revealing more."
     ),
     "Quick Summary": (
         "Provide concise, bullet-point style summaries and cheat-sheet style notes. "
@@ -131,6 +135,22 @@ WEAKNESS_KEYWORDS = [
     "很难",
 ]
 
+# ---------- 简单“掌握”检测 ----------
+MASTERY_KEYWORDS = [
+    "got it",
+    "makes sense",
+    "now i see",
+    "i see",
+    "understand now",
+    "clear now",
+    "easy",
+    "no problem",
+    "没问题",
+    "懂了",
+    "明白了",
+    "清楚了",
+]
+
 def update_weaknesses_from_message(message: str, weaknesses: List[str]) -> List[str]:
     lower_msg = message.lower()
     if any(k in lower_msg for k in WEAKNESS_KEYWORDS):
@@ -139,6 +159,39 @@ def update_weaknesses_from_message(message: str, weaknesses: List[str]) -> List[
     return weaknesses
 
 
+def update_cognitive_state_from_message(
+    message: str,
+    state: Optional[Dict[str, int]],
+) -> Dict[str, int]:
+    """
+    简单认知状态统计：
+    - 遇到困惑类关键词 → confusion +1
+    - 遇到掌握类关键词 → mastery +1
+    """
+    if state is None:
+        state = {"confusion": 0, "mastery": 0}
+
+    lower_msg = message.lower()
+    if any(k in lower_msg for k in WEAKNESS_KEYWORDS):
+        state["confusion"] = state.get("confusion", 0) + 1
+    if any(k in lower_msg for k in MASTERY_KEYWORDS):
+        state["mastery"] = state.get("mastery", 0) + 1
+    return state
+
+
+def describe_cognitive_state(state: Optional[Dict[str, int]]) -> str:
+    if not state:
+        return "unknown"
+    confusion = state.get("confusion", 0)
+    mastery = state.get("mastery", 0)
+    if confusion >= 2 and confusion >= mastery + 1:
+        return "student shows signs of HIGH cognitive load (often confused)."
+    elif mastery >= 2 and mastery >= confusion + 1:
+        return "student seems COMFORTABLE; material may be slightly easy."
+    else:
+        return "mixed or uncertain cognitive state."
+
+
 # ---------- 构建 messages ----------
 def build_messages(
     user_message: str,
@@ -148,6 +201,7 @@ def build_messages(
     doc_type: str,
     course_outline: Optional[List[str]],
     weaknesses: Optional[List[str]],
+    cognitive_state: Optional[Dict[str, int]],
 ) -> List[Dict[str, str]]:
     messages: List[Dict[str, str]] = [
         {"role": "system", "content": CLARE_SYSTEM_PROMPT}
@@ -203,6 +257,45 @@ def build_messages(
             }
         )
 
+    # 认知状态提示（动态复杂度调整）
+    if cognitive_state:
+        confusion = cognitive_state.get("confusion", 0)
+        mastery = cognitive_state.get("mastery", 0)
+        if confusion >= 2 and confusion >= mastery + 1:
+            messages.append(
+                {
+                    "role": "system",
+                    "content": (
+                        "The student is currently under HIGH cognitive load. "
+                        "Use simpler language, shorter steps, and more concrete examples. "
+                        "Avoid long derivations in a single answer, and check understanding "
+                        "frequently."
+                    ),
+                }
+            )
+        elif mastery >= 2 and mastery >= confusion + 1:
+            messages.append(
+                {
+                    "role": "system",
+                    "content": (
+                        "The student seems comfortable with the material. "
+                        "You may increase difficulty slightly, introduce deeper follow-up "
+                        "questions, and connect concepts across topics."
+                    ),
+                }
+            )
+        else:
+            messages.append(
+                {
+                    "role": "system",
+                    "content": (
+                        "The student's cognitive state is mixed or uncertain. "
+                        "Keep explanations clear and moderately paced, and probe for "
+                        "understanding with short questions."
+                    ),
+                }
+            )
+
     # 语言偏好控制
     if language_preference == "English":
         messages.append(
@@ -233,6 +326,7 @@ def chat_with_clare(
     doc_type: str,
     course_outline: Optional[List[str]],
     weaknesses: Optional[List[str]],
+    cognitive_state: Optional[Dict[str, int]],
 ):
     try:
         messages = build_messages(
@@ -243,6 +337,7 @@ def chat_with_clare(
             doc_type=doc_type,
             course_outline=course_outline,
             weaknesses=weaknesses,
+            cognitive_state=cognitive_state,
         )
         response = client.chat.completions.create(
             model=model_name or DEFAULT_MODEL,
@@ -263,11 +358,13 @@ def export_conversation(
     course_outline: List[str],
     learning_mode_val: str,
     weaknesses: List[str],
+    cognitive_state: Optional[Dict[str, int]],
 ) -> str:
     lines: List[str] = []
     lines.append("# Clare – Conversation Export\n")
     lines.append(f"- Learning mode: **{learning_mode_val}**\n")
     lines.append("- Course topics (short): " + "; ".join(course_outline[:5]) + "\n")
+    lines.append(f"- Cognitive state snapshot: {describe_cognitive_state(cognitive_state)}\n")
 
     if weaknesses:
         lines.append("- Observed student difficulties:\n")
@@ -288,6 +385,7 @@ def generate_quiz_from_history(
     history: List[Tuple[str, str]],
     course_outline: List[str],
     weaknesses: List[str],
+    cognitive_state: Optional[Dict[str, int]],
     model_name: str,
     language_preference: str,
 ) -> str:
@@ -297,6 +395,7 @@ def generate_quiz_from_history(
 
     topics_text = "; ".join(course_outline[:8])
     weakness_text = "; ".join(weaknesses[-5:]) if weaknesses else "N/A"
+    cog_text = describe_cognitive_state(cognitive_state)
 
     messages = [
         {"role": "system", "content": CLARE_SYSTEM_PROMPT},
@@ -307,7 +406,8 @@ def generate_quiz_from_history(
                 "Based on the conversation and course topics, generate **3 questions** "
                 "(a mix of multiple-choice and short-answer is fine). After listing the "
                 "questions, provide an answer key at the end under a heading 'Answer Key'. "
-                "Number the questions Q1, Q2, Q3."
+                "Number the questions Q1, Q2, Q3. Adjust the difficulty according to the "
+                "student's cognitive state."
             ),
         },
         {
@@ -318,6 +418,10 @@ def generate_quiz_from_history(
             "role": "system",
             "content": f"Student known difficulties: {weakness_text}",
         },
+        {
+            "role": "system",
+            "content": f"Student cognitive state: {cog_text}",
+        },
         {
             "role": "user",
             "content": (
@@ -354,6 +458,7 @@ def summarize_conversation(
     history: List[Tuple[str, str]],
     course_outline: List[str],
     weaknesses: List[str],
+    cognitive_state: Optional[Dict[str, int]],
     model_name: str,
     language_preference: str,
 ) -> str:
@@ -363,6 +468,7 @@ def summarize_conversation(
 
     topics_text = "; ".join(course_outline[:8])
     weakness_text = "; ".join(weaknesses[-5:]) if weaknesses else "N/A"
+    cog_text = describe_cognitive_state(cognitive_state)
 
     messages = [
         {"role": "system", "content": CLARE_SYSTEM_PROMPT},
@@ -373,7 +479,8 @@ def summarize_conversation(
                 "session. Only include knowledge points, definitions, key formulas, "
                 "examples, and main takeaways. Do **not** include any personal remarks, "
                 "jokes, or off-topic chat. Write in clear bullet points. This summary "
-                "should be suitable for the student to paste into their study notes."
+                "should be suitable for the student to paste into their study notes. "
+                "Take into account what the student struggled with and their cognitive state."
             ),
         },
         {
@@ -384,6 +491,10 @@ def summarize_conversation(
             "role": "system",
             "content": f"Student known difficulties: {weakness_text}",
         },
+        {
+            "role": "system",
+            "content": f"Student cognitive state: {cog_text}",
+        },
         {
             "role": "user",
             "content": (
@@ -457,9 +568,10 @@ with gr.Blocks(title="Clare – Hanbridge AI Teaching Assistant") as demo:
             label="File type",
         )
 
-    # 状态：课程大纲 + 学生弱项
+    # 状态：课程大纲 + 学生弱项 + 认知状态
     course_outline_state = gr.State(DEFAULT_COURSE_TOPICS)
     weakness_state = gr.State([])
+    cognitive_state_state = gr.State({"confusion": 0, "mastery": 0})
 
     # syllabus 上传后更新课程大纲（仅当类型是 Syllabus 时解析）
     def update_outline(file, doc_type_val):
@@ -505,19 +617,21 @@ with gr.Blocks(title="Clare – Hanbridge AI Teaching Assistant") as demo:
         lines=8,
     )
 
-    # 主对话逻辑：更新弱项 + 调用 Clare
+    # 主对话逻辑：更新弱项 + 认知状态 + 调用 Clare
     def respond(
         message,
         chat_history,
         course_outline,
         weaknesses,
+        cognitive_state,
         model_name_val,
         language_pref_val,
         learning_mode_val,
         doc_type_val,
     ):
-        # 更新弱项记忆
         weaknesses = update_weaknesses_from_message(message, weaknesses or [])
+        cognitive_state = update_cognitive_state_from_message(message, cognitive_state)
+
         answer, new_history = chat_with_clare(
             message=message,
             history=chat_history,
@@ -527,8 +641,9 @@ with gr.Blocks(title="Clare – Hanbridge AI Teaching Assistant") as demo:
             doc_type=doc_type_val,
             course_outline=course_outline,
             weaknesses=weaknesses,
+            cognitive_state=cognitive_state,
         )
-        return "", new_history, weaknesses
+        return "", new_history, weaknesses, cognitive_state
 
     user_input.submit(
         respond,
@@ -537,64 +652,101 @@ with gr.Blocks(title="Clare – Hanbridge AI Teaching Assistant") as demo:
             chatbot,
             course_outline_state,
             weakness_state,
+            cognitive_state_state,
             model_name,
             language_preference,
             learning_mode,
             doc_type,
         ],
-        [user_input, chatbot, weakness_state],
+        [user_input, chatbot, weakness_state, cognitive_state_state],
     )
 
-    # 清空对话 & 弱项 & 导出/quiz/summary
+    # 清空对话 & 状态 & 导出/quiz/summary
     def clear_all():
-        return [], [], "", "", ""
+        return [], [], {"confusion": 0, "mastery": 0}, "", "", ""
 
     clear_btn.click(
         clear_all,
         None,
-        [chatbot, weakness_state, export_box, quiz_box, summary_box],
+        [chatbot, weakness_state, cognitive_state_state, export_box, quiz_box, summary_box],
         queue=False,
     )
 
     # 导出对话按钮
-    def on_export(chat_history, course_outline, learning_mode_val, weaknesses):
-        return export_conversation(chat_history, course_outline, learning_mode_val, weaknesses or [])
+    def on_export(chat_history, course_outline, learning_mode_val, weaknesses, cognitive_state):
+        return export_conversation(
+            chat_history,
+            course_outline,
+            learning_mode_val,
+            weaknesses or [],
+            cognitive_state,
+        )
 
     export_btn.click(
         on_export,
-        [chatbot, course_outline_state, learning_mode, weakness_state],
+        [chatbot, course_outline_state, learning_mode, weakness_state, cognitive_state_state],
         [export_box],
     )
 
     # 生成 quiz 按钮
-    def on_quiz(chat_history, course_outline, weaknesses, model_name_val, language_pref_val):
+    def on_quiz(
+        chat_history,
+        course_outline,
+        weaknesses,
+        cognitive_state,
+        model_name_val,
+        language_pref_val,
+    ):
         return generate_quiz_from_history(
             chat_history,
             course_outline,
             weaknesses or [],
+            cognitive_state,
             model_name_val,
             language_pref_val,
         )
 
     quiz_btn.click(
         on_quiz,
-        [chatbot, course_outline_state, weakness_state, model_name, language_preference],
+        [
+            chatbot,
+            course_outline_state,
+            weakness_state,
+            cognitive_state_state,
+            model_name,
+            language_preference,
+        ],
         [quiz_box],
     )
 
     # 概念总结按钮
-    def on_summary(chat_history, course_outline, weaknesses, model_name_val, language_pref_val):
+    def on_summary(
+        chat_history,
+        course_outline,
+        weaknesses,
+        cognitive_state,
+        model_name_val,
+        language_pref_val,
+    ):
         return summarize_conversation(
             chat_history,
             course_outline,
             weaknesses or [],
+            cognitive_state,
             model_name_val,
             language_pref_val,
         )
 
     summary_btn.click(
         on_summary,
-        [chatbot, course_outline_state, weakness_state, model_name, language_preference],
+        [
+            chatbot,
+            course_outline_state,
+            weakness_state,
+            cognitive_state_state,
+            model_name,
+            language_preference,
+        ],
         [summary_box],
     )