Update src/streamlit_app.py

src/streamlit_app.py

@@ -7,6 +7,21 @@ import streamlit as st
 import torch
 from document_registry import DocumentRegistry
 
+# --- Initialize session keys to avoid rerun issues ---
+if "registry" not in st.session_state:
+    st.session_state.registry = DocumentRegistry()
+if "active_doc" not in st.session_state:
+    st.session_state.active_doc = None
+if "chunks" not in st.session_state:
+    st.session_state.chunks = []
+if "embeddings" not in st.session_state:
+    st.session_state.embeddings = None
+if "index" not in st.session_state:
+    st.session_state.index = None
+if "query_suggestions_fixed" not in st.session_state:
+    st.session_state.query_suggestions_fixed = []
+
+
 # ==========================================================
 # ✅ PAGE CONFIGS
 # ==========================================================
@@ -85,6 +100,30 @@ Output: Write each question on a new line. Do not invent facts — base question
     except Exception:
         return ["How do I start using this guide?", "What does this document cover?"]
 
+# --- Function to activate a document from the registry ---
+def activate_document(doc_name):
+    registry = st.session_state.registry
+    doc = registry.get_doc(doc_name)
+
+    if not doc:
+        st.warning(f"Document {doc_name} not found in registry.")
+        return
+
+    # Load this document's preprocessed data into session
+    st.session_state.active_doc = doc
+    st.session_state.chunks = doc["chunks"]
+    st.session_state.embeddings = doc["embeddings"]
+    st.session_state.index = doc["index"]
+
+    # Regenerate suggestions for this document
+    st.session_state.query_suggestions_fixed = generate_dynamic_suggestions_from_toc(
+        doc["toc"], doc["chunks"], doc["name"]
+    )
+
+    # Trigger a clean rerun so UI updates with new context
+    st.rerun()
+
+
 # ==========================================================
 # 🎨 STYLING — REVERT TO ORIGINAL
 # ==========================================================
@@ -149,13 +188,36 @@ with st.sidebar:
     if "registry" in st.session_state:
         registry = st.session_state["registry"]
         registered_docs = registry.list_docs() if hasattr(registry, "list_docs") else []
+
         if registered_docs:
-            with st.expander("📚 Registered Documents", expanded=False):
-                for i, doc in enumerate(registered_docs, start=1):
-                    doc_name = doc.get("name", "Unknown")
-                    chunks = doc.get("num_chunks", "?")
-                    toc_source = doc.get("toc_source", "—")
-                    st.markdown(f"**{i}. {doc_name}** — {chunks} chunks *(TOC: {toc_source})*")
+            st.markdown("### 📚 Registered Documents")
+
+            # Loop through registered documents
+            for i, doc in enumerate(registered_docs, start=1):
+                doc_name = doc.get("name", f"Document {i}")
+                chunks = doc.get("num_chunks", "?")
+                toc_source = doc.get("toc_source", "—")
+
+                # Use an expander for each document
+                with st.expander(f"{i}. {doc_name}", expanded=False):
+                    st.markdown(f"- 🧩 **Chunks:** {chunks}")
+                    st.markdown(f"- 🗂️ **TOC Source:** {toc_source}")
+
+                    # If this document is currently active, show status
+                    if (
+                        "active_doc" in st.session_state
+                        and st.session_state.active_doc
+                        and st.session_state.active_doc.get("name") == doc_name
+                    ):
+                        st.success("✅ Active Document")
+                    else:
+                        # Button to activate this document
+                        st.button(
+                            "Activate",
+                            key=f"activate_{doc_name}",
+                            on_click=activate_document,
+                            args=(doc_name,),
+                        )
         else:
             st.caption("📭 No documents registered yet.")
     else:
@@ -163,6 +225,7 @@ with st.sidebar:
 
     st.markdown("---")
 
+    # 🧠 Developer Options Section
     show_dev = st.checkbox("Show advanced settings (for developers)", value=False)
     if show_dev:
         st.markdown("### ⚙️ Developer Options")
@@ -175,6 +238,7 @@ with st.sidebar:
     st.markdown("---")
     st.caption("✨ Built by Shubham Sharma")
 
+    # 🧩 Developer Insights
     if show_dev:
         st.markdown("---")
         with st.expander("🧩 Developer Insights", expanded=False):
@@ -196,6 +260,7 @@ with st.sidebar:
                 st.text_area("", doc_text[:1000], height=120)
                 st.caption(f"{len(st.session_state.get('chunks', []))} chunks processed.")
 
+
 # ==========================================================
 # 🧠 SESSION STATE SAFETY INITIALIZATION
 # ==========================================================