Update conversation_logic.py

conversation_logic.py

@@ -526,6 +526,34 @@ def _is_bad_generated_reply(text: str, user_text: str = "") -> bool:
 
     return False
 
+def _pick_teaching_line(
+    chunks: List[RetrievedChunk],
+    current_reply: str,
+) -> Optional[str]:
+    if not chunks:
+        return None
+
+    reply_keywords = _extract_keywords(current_reply)
+    best_line = None
+    best_score = -1
+
+    for chunk in chunks:
+        text = (chunk.text or "").strip().replace("\n", " ")
+        if not text:
+            continue
+        if len(text) > 180:
+            text = text[:177].rstrip() + "..."
+
+        chunk_keywords = _extract_keywords(text)
+        novelty = len(chunk_keywords - reply_keywords)
+        overlap = len(chunk_keywords & reply_keywords)
+
+        score = novelty - 0.3 * overlap
+        if score > best_score:
+            best_score = score
+            best_line = text
+
+    return best_line
 
 class ConversationEngine:
     def __init__(
@@ -648,8 +676,10 @@ class ConversationEngine:
                 result.used_retrieval = True
                 result.teaching_chunks = filtered
 
-        if selected_chunks and resolved_help_mode != "answer":
-            reply = f"{reply}\n\nRelevant study notes:\n" + "\n".join(_teaching_lines(selected_chunks))
+        if selected_chunks and resolved_help_mode in {"walkthrough", "step_by_step", "method", "explain", "concept"}:
+            teaching_line = _pick_teaching_line(selected_chunks, reply)
+            if teaching_line:
+                reply = f"{reply}\n\nKey idea: {teaching_line}"
 
         should_try_generator = (
             self.generator is not None