Update src/streamlit_app.py

src/streamlit_app.py

@@ -102,6 +102,7 @@ Output: Write each question on a new line. Do not invent facts — base question
 
 # --- Function to activate a document from the registry ---
 def activate_document(doc_name):
+    """Load an existing registered document into the active context."""
     registry = st.session_state.registry
     doc = registry.get_doc(doc_name)
 
@@ -109,21 +110,35 @@ def activate_document(doc_name):
         st.warning(f"Document {doc_name} not found in registry.")
         return
 
-    # Load this document's preprocessed data into session
+    # Load all available preprocessed data
     st.session_state.active_doc = doc
-    st.session_state.chunks = doc["chunks"]
-    st.session_state.embeddings = doc["embeddings"]
-    st.session_state.index = doc["index"]
+    st.session_state.chunks = doc.get("chunks", [])
+    st.session_state.embeddings = doc.get("embeddings")
+    st.session_state.index = doc.get("index")
 
-    # Regenerate suggestions for this document
-    st.session_state.query_suggestions_fixed = generate_dynamic_suggestions_from_toc(
-        doc["toc"], doc["chunks"], doc["name"]
-    )
+    # --- Handle TOC safely ---
+    # Some docs may not have toc saved in registry; fallback to empty list
+    toc_data = doc.get("toc", [])
+    if not toc_data:
+        st.info(f"⚠️ No TOC found for {doc_name}. Using empty TOC for suggestions.")
 
-    # Trigger a clean rerun so UI updates with new context
+    # --- Regenerate suggestions safely ---
+    try:
+        st.session_state.query_suggestions_fixed = generate_dynamic_suggestions_from_toc(
+            toc_data,
+            st.session_state.chunks,
+            doc.get("name", "Unknown Document")
+        )
+    except Exception as e:
+        # Defensive guard: even if suggestion generator fails, app won't break
+        st.warning(f"Suggestion generation failed: {e}")
+        st.session_state.query_suggestions_fixed = []
+
+    # --- Trigger a clean UI rerun so the switch shows immediately ---
     st.rerun()
 
 
+
 # ==========================================================
 # 🎨 STYLING — REVERT TO ORIGINAL
 # ==========================================================