Update app.py

app.py

@@ -349,6 +349,72 @@ def generate_quiz_from_history(
     return quiz_text
 
 
+# ---------- 概念总结（知识点摘要） ----------
+def summarize_conversation(
+    history: List[Tuple[str, str]],
+    course_outline: List[str],
+    weaknesses: List[str],
+    model_name: str,
+    language_preference: str,
+) -> str:
+    conversation_text = ""
+    for user, assistant in history[-10:]:  # 最近 10 轮足够
+        conversation_text += f"Student: {user}\nClare: {assistant}\n"
+
+    topics_text = "; ".join(course_outline[:8])
+    weakness_text = "; ".join(weaknesses[-5:]) if weaknesses else "N/A"
+
+    messages = [
+        {"role": "system", "content": CLARE_SYSTEM_PROMPT},
+        {
+            "role": "system",
+            "content": (
+                "Your task now is to produce a **concept-only summary** of this tutoring "
+                "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."
+            ),
+        },
+        {
+            "role": "system",
+            "content": f"Course topics context: {topics_text}",
+        },
+        {
+            "role": "system",
+            "content": f"Student known difficulties: {weakness_text}",
+        },
+        {
+            "role": "user",
+            "content": (
+                "Here is the recent conversation between you and the student:\n\n"
+                + conversation_text
+                + "\n\nPlease summarize only the concepts and key ideas learned."
+            ),
+        },
+    ]
+
+    if language_preference == "中文":
+        messages.append(
+            {
+                "role": "system",
+                "content": "请用中文给出要点总结，只保留知识点和结论，使用条目符号。"
+            }
+        )
+
+    try:
+        response = client.chat.completions.create(
+            model=model_name or DEFAULT_MODEL,
+            messages=messages,
+            temperature=0.4,
+        )
+        summary_text = response.choices[0].message.content
+    except Exception as e:
+        summary_text = f"⚠️ Error generating summary: {e}"
+
+    return summary_text
+
+
 # ---------- Gradio UI ----------
 with gr.Blocks(title="Clare – Hanbridge AI Teaching Assistant") as demo:
     gr.Markdown(
@@ -424,14 +490,19 @@ with gr.Blocks(title="Clare – Hanbridge AI Teaching Assistant") as demo:
         clear_btn = gr.Button("Reset conversation")
         export_btn = gr.Button("Export conversation")
         quiz_btn = gr.Button("Generate 3 quiz questions")
+        summary_btn = gr.Button("Summarize concepts")
 
     export_box = gr.Textbox(
         label="Conversation export (Markdown)",
-        lines=10,
+        lines=8,
     )
     quiz_box = gr.Textbox(
         label="Generated quiz (with answer key)",
-        lines=10,
+        lines=8,
+    )
+    summary_box = gr.Textbox(
+        label="Concept summary (for study notes)",
+        lines=8,
     )
 
     # 主对话逻辑：更新弱项 + 调用 Clare
@@ -474,14 +545,14 @@ with gr.Blocks(title="Clare – Hanbridge AI Teaching Assistant") as demo:
         [user_input, chatbot, weakness_state],
     )
 
-    # 清空对话 & 弱项 & 导出/quiz
+    # 清空对话 & 弱项 & 导出/quiz/summary
     def clear_all():
-        return [], [], "", ""
+        return [], [], "", "", ""
 
     clear_btn.click(
         clear_all,
         None,
-        [chatbot, weakness_state, export_box, quiz_box],
+        [chatbot, weakness_state, export_box, quiz_box, summary_box],
         queue=False,
     )
 
@@ -511,5 +582,21 @@ with gr.Blocks(title="Clare – Hanbridge AI Teaching Assistant") as demo:
         [quiz_box],
     )
 
+    # 概念总结按钮
+    def on_summary(chat_history, course_outline, weaknesses, model_name_val, language_pref_val):
+        return summarize_conversation(
+            chat_history,
+            course_outline,
+            weaknesses or [],
+            model_name_val,
+            language_pref_val,
+        )
+
+    summary_btn.click(
+        on_summary,
+        [chatbot, course_outline_state, weakness_state, model_name, language_preference],
+        [summary_box],
+    )
+
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()