Update api/server.py

api/server.py

@@ -102,6 +102,8 @@ def _get_session(user_id: str) -> Dict:
             # preload base reading
             "rag_chunks": list(MODULE10_CHUNKS_CACHE),
             "model_name": DEFAULT_MODEL,
+            # ✅ NEW: track last syllabus filename for refs fallback
+            "last_syllabus_file": None,
         }
     return SESSIONS[user_id]
 
@@ -202,6 +204,25 @@ def chat(req: ChatReq):
         for c in (rag_used_chunks or [])
     ]
 
+    # ✅ NEW: Syllabus chat should show syllabus in refs (fallback if retrieval didn't hit it)
+    doc_type_norm = (req.doc_type or "").strip().lower()
+    if doc_type_norm == "syllabus":
+        syllabus_name = sess.get("last_syllabus_file") or "uploaded_syllabus"
+        # We consider it a hit if any ref source_file contains the syllabus filename
+        def _has_syllabus_ref() -> bool:
+            for r in refs:
+                sf = (r.get("source_file") or "").strip()
+                if not sf:
+                    continue
+                # robust matching: exact basename containment OR endswith
+                if syllabus_name in sf or sf.endswith(syllabus_name):
+                    return True
+            return False
+
+        if not _has_syllabus_ref():
+            # put syllabus at the front, preserve existing refs after
+            refs = [{"source_file": syllabus_name, "section": "syllabus_outline"}] + refs
+
     return {
         "reply": answer,
         "session_status_md": render_session_status(
@@ -236,6 +257,10 @@ async def upload(
     with open(tmp_path, "wb") as f:
         f.write(content)
 
+    # ✅ NEW: remember the latest syllabus filename for refs fallback
+    if doc_type == "Syllabus":
+        sess["last_syllabus_file"] = os.path.basename(file.filename) or safe_name
+
     # Update topics only for syllabus
     if doc_type == "Syllabus":
         class _F: