Update app.py

app.py

@@ -210,6 +210,20 @@ with gr.Blocks(title="Clare – Hanbridge AI Teaching Assistant") as demo:
             rag_context=rag_context,
         )
 
+        # RAG 标记
+        if rag_context:
+            # 根据语言给不同提示
+            if resolved_lang == "中文":
+                prefix = "（本次回答参考了你上传的课程文档内容。）\n\n"
+            else:
+                prefix = "(This answer used excerpts from your uploaded course document.)\n\n"
+    
+            tagged_answer = prefix + answer
+            # 更新最后一条记录里的 assistant 内容
+            if new_history:
+                new_history[-1] = (message, tagged_answer)
+            answer = tagged_answer
+    
         status_text = render_session_status(learning_mode_val, weaknesses, cognitive_state)
         return "", new_history, weaknesses, cognitive_state, status_text