Update app.py

app.py

@@ -63,6 +63,14 @@ LEARNING_MODE_INSTRUCTIONS = {
     ),
 }
 
+# ---------- 上传文件类型 ----------
+DOC_TYPES = [
+    "Syllabus",
+    "Lecture Slides / PPT",
+    "Literature Review / Paper",
+    "Other Course Document",
+]
+
 # ---------- Clare 的基础 System Prompt ----------
 CLARE_SYSTEM_PROMPT = """
 You are Clare, an AI teaching assistant for Hanbridge University.
@@ -106,13 +114,40 @@ def parse_syllabus_docx(file_path: str, max_lines: int = 15) -> List[str]:
     return topics
 
 
+# ---------- 简单“弱项”检测 ----------
+WEAKNESS_KEYWORDS = [
+    "don't understand",
+    "do not understand",
+    "not understand",
+    "not sure",
+    "confused",
+    "hard to",
+    "difficult",
+    "struggle",
+    "不会",
+    "不懂",
+    "看不懂",
+    "搞不清",
+    "很难",
+]
+
+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):
+        weaknesses = weaknesses or []
+        weaknesses.append(message)
+    return weaknesses
+
+
 # ---------- 构建 messages ----------
 def build_messages(
     user_message: str,
     history: List[Tuple[str, str]],
     language_preference: str,
     learning_mode: str,
+    doc_type: str,
     course_outline: Optional[List[str]],
+    weaknesses: Optional[List[str]],
 ) -> List[Dict[str, str]]:
     messages: List[Dict[str, str]] = [
         {"role": "system", "content": CLARE_SYSTEM_PROMPT}
@@ -142,6 +177,32 @@ def build_messages(
         }
     )
 
+    # 上传文件类型提示（仅作语境）
+    if doc_type and doc_type != "Syllabus":
+        messages.append(
+            {
+                "role": "system",
+                "content": (
+                    f"The student also uploaded a {doc_type} document as supporting material. "
+                    "You do not see the full content directly, but you may assume it is relevant "
+                    "to the same course and topics."
+                ),
+            }
+        )
+
+    # 学生弱项提示（会话内记忆）
+    if weaknesses:
+        weak_text = " | ".join(weaknesses[-5:])  # 最近几条即可
+        messages.append(
+            {
+                "role": "system",
+                "content": (
+                    "The student seems to struggle with the following questions or topics. "
+                    "Be extra gentle and clear when these appear: " + weak_text
+                ),
+            }
+        )
+
     # 语言偏好控制
     if language_preference == "English":
         messages.append(
@@ -169,7 +230,9 @@ def chat_with_clare(
     model_name: str,
     language_preference: str,
     learning_mode: str,
+    doc_type: str,
     course_outline: Optional[List[str]],
+    weaknesses: Optional[List[str]],
 ):
     try:
         messages = build_messages(
@@ -177,7 +240,9 @@ def chat_with_clare(
             history=history,
             language_preference=language_preference,
             learning_mode=learning_mode,
+            doc_type=doc_type,
             course_outline=course_outline,
+            weaknesses=weaknesses,
         )
         response = client.chat.completions.create(
             model=model_name or DEFAULT_MODEL,
@@ -192,6 +257,98 @@ def chat_with_clare(
     return answer, history
 
 
+# ---------- 导出对话为 Markdown ----------
+def export_conversation(
+    history: List[Tuple[str, str]],
+    course_outline: List[str],
+    learning_mode_val: str,
+    weaknesses: List[str],
+) -> 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")
+
+    if weaknesses:
+        lines.append("- Observed student difficulties:\n")
+        for w in weaknesses[-5:]:
+            lines.append(f"  - {w}\n")
+    lines.append("\n---\n\n")
+
+    for user, assistant in history:
+        lines.append(f"**Student:** {user}\n\n")
+        lines.append(f"**Clare:** {assistant}\n\n")
+        lines.append("---\n\n")
+
+    return "".join(lines)
+
+
+# ---------- 生成 3 个 quiz 题目 ----------
+def generate_quiz_from_history(
+    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[-8:]:  # 用最近几轮
+        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": (
+                "Now your task is to create a **short concept quiz** for the student. "
+                "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."
+            ),
+        },
+        {
+            "role": "system",
+            "content": f"Course topics: {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 create the quiz now."
+            ),
+        },
+    ]
+
+    if language_preference == "中文":
+        messages.append(
+            {
+                "role": "system",
+                "content": "请用中文给出问题和答案。",
+            }
+        )
+
+    try:
+        response = client.chat.completions.create(
+            model=model_name or DEFAULT_MODEL,
+            messages=messages,
+            temperature=0.5,
+        )
+        quiz_text = response.choices[0].message.content
+    except Exception as e:
+        quiz_text = f"⚠️ Error generating quiz: {e}"
+
+    return quiz_text
+
+
 # ---------- Gradio UI ----------
 with gr.Blocks(title="Clare – Hanbridge AI Teaching Assistant") as demo:
     gr.Markdown(
@@ -202,7 +359,7 @@ with gr.Blocks(title="Clare – Hanbridge AI Teaching Assistant") as demo:
         - Ask in English → Clare answers in English.  
         - Ask in Chinese → Clare can answer in Chinese.  
         - Use different **learning modes** to change Clare's teaching style.  
-        - Optionally upload your **course syllabus (.docx)** so Clare stays aligned with your course.
+        - Optionally upload your **course syllabus / slides / literature (.docx)** so Clare stays aligned with your course.
         """
     )
 
@@ -225,21 +382,32 @@ with gr.Blocks(title="Clare – Hanbridge AI Teaching Assistant") as demo:
 
     with gr.Row():
         syllabus_file = gr.File(
-            label="Upload course syllabus (.docx)",
+            label="Upload course file (.docx)",
             file_types=[".docx"],
         )
-        course_outline_state = gr.State(DEFAULT_COURSE_TOPICS)
+        doc_type = gr.Dropdown(
+            choices=DOC_TYPES,
+            value="Syllabus",
+            label="File type",
+        )
+
+    # 状态：课程大纲 + 学生弱项
+    course_outline_state = gr.State(DEFAULT_COURSE_TOPICS)
+    weakness_state = gr.State([])
 
-    # syllabus 上传后更新课程大纲
-    def update_outline(file):
+    # syllabus 上传后更新课程大纲（仅当类型是 Syllabus 时解析）
+    def update_outline(file, doc_type_val):
         if file is None:
             return DEFAULT_COURSE_TOPICS
-        topics = parse_syllabus_docx(file.name)
-        return topics
+        if doc_type_val == "Syllabus" and file.name.lower().endswith(".docx"):
+            topics = parse_syllabus_docx(file.name)
+            return topics
+        # 其他类型文件目前不解析，只保留默认大纲
+        return DEFAULT_COURSE_TOPICS
 
     syllabus_file.change(
         fn=update_outline,
-        inputs=[syllabus_file],
+        inputs=[syllabus_file, doc_type],
         outputs=[course_outline_state],
     )
 
@@ -251,37 +419,97 @@ with gr.Blocks(title="Clare – Hanbridge AI Teaching Assistant") as demo:
         label="Your question",
         placeholder="Ask Clare anything about your course, assignment, or study plan...",
     )
-    clear_btn = gr.Button("Reset conversation")
 
-    # 注意这里显式把组件的当前值传进来，而不是 .value
+    with gr.Row():
+        clear_btn = gr.Button("Reset conversation")
+        export_btn = gr.Button("Export conversation")
+        quiz_btn = gr.Button("Generate 3 quiz questions")
+
+    export_box = gr.Textbox(
+        label="Conversation export (Markdown)",
+        lines=10,
+    )
+    quiz_box = gr.Textbox(
+        label="Generated quiz (with answer key)",
+        lines=10,
+    )
+
+    # 主对话逻辑：更新弱项 + 调用 Clare
     def respond(
         message,
         chat_history,
         course_outline,
+        weaknesses,
         model_name_val,
         language_pref_val,
         learning_mode_val,
+        doc_type_val,
     ):
+        # 更新弱项记忆
+        weaknesses = update_weaknesses_from_message(message, weaknesses or [])
         answer, new_history = chat_with_clare(
             message=message,
             history=chat_history,
             model_name=model_name_val,
             language_preference=language_pref_val,
             learning_mode=learning_mode_val,
+            doc_type=doc_type_val,
             course_outline=course_outline,
+            weaknesses=weaknesses,
         )
-        return "", new_history
+        return "", new_history, weaknesses
 
     user_input.submit(
         respond,
-        [user_input, chatbot, course_outline_state, model_name, language_preference, learning_mode],
-        [user_input, chatbot],
+        [
+            user_input,
+            chatbot,
+            course_outline_state,
+            weakness_state,
+            model_name,
+            language_preference,
+            learning_mode,
+            doc_type,
+        ],
+        [user_input, chatbot, weakness_state],
     )
 
-    def clear_chat():
-        return []
+    # 清空对话 & 弱项 & 导出/quiz
+    def clear_all():
+        return [], [], "", ""
+
+    clear_btn.click(
+        clear_all,
+        None,
+        [chatbot, weakness_state, export_box, quiz_box],
+        queue=False,
+    )
 
-    clear_btn.click(clear_chat, None, chatbot, 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 [])
+
+    export_btn.click(
+        on_export,
+        [chatbot, course_outline_state, learning_mode, weakness_state],
+        [export_box],
+    )
+
+    # 生成 quiz 按钮
+    def on_quiz(chat_history, course_outline, weaknesses, model_name_val, language_pref_val):
+        return generate_quiz_from_history(
+            chat_history,
+            course_outline,
+            weaknesses or [],
+            model_name_val,
+            language_pref_val,
+        )
+
+    quiz_btn.click(
+        on_quiz,
+        [chatbot, course_outline_state, weakness_state, model_name, language_preference],
+        [quiz_box],
+    )
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()