Update src/streamlit_app.py

src/streamlit_app.py

@@ -7,20 +7,6 @@ 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
@@ -100,44 +86,6 @@ 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):
-    """Load an existing registered document into the active context."""
-    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 all available preprocessed data
-    st.session_state.active_doc = doc
-    st.session_state.chunks = doc.get("chunks", [])
-    st.session_state.embeddings = doc.get("embeddings")
-    st.session_state.index = doc.get("index")
-
-    # --- 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.")
-
-    # --- 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
@@ -189,6 +137,7 @@ h1, h2, h3 {color: #f3f4f6; font-weight: 600;}
 # 🧭 SIDEBAR
 # ==========================================================
 with st.sidebar:
+    # --- Response Style ---
     st.markdown("### 🧭 Response Style")
     mode = st.radio(
         "",
@@ -199,50 +148,23 @@ with st.sidebar:
 
     st.markdown("---")
 
-    # 🧩 Document Registry Viewer
-    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:
-            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 isinstance(st.session_state.active_doc, dict)
-                        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.")
+    # --- Registered Documents (simple, user-friendly) ---
+    registry = st.session_state.get("registry", None)
+    registered_docs = registry.list_docs() if registry and hasattr(registry, "list_docs") else []
+
+    if registered_docs:
+        st.markdown("### 📚 Uploaded 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", "—")
+            st.markdown(f"**{i}. {doc_name}** — {chunks} chunks *(TOC: {toc_source})*")
     else:
-        st.caption("📭 No registry initialized yet.")
+        st.caption("📭 No documents uploaded yet.")
 
     st.markdown("---")
 
-    # 🧠 Developer Options Section
+    # --- Developer Options ---
     show_dev = st.checkbox("Show advanced settings (for developers)", value=False)
     if show_dev:
         st.markdown("### ⚙️ Developer Options")
@@ -255,7 +177,7 @@ with st.sidebar:
     st.markdown("---")
     st.caption("✨ Built by Shubham Sharma")
 
-    # 🧩 Developer Insights
+    # --- Developer Insights (optional, hidden by default) ---
     if show_dev:
         st.markdown("---")
         with st.expander("🧩 Developer Insights", expanded=False):
@@ -278,6 +200,7 @@ with st.sidebar:
                 st.caption(f"{len(st.session_state.get('chunks', []))} chunks processed.")
 
 
+
 # ==========================================================
 # 🧠 SESSION STATE SAFETY INITIALIZATION
 # ==========================================================
@@ -351,12 +274,13 @@ else:
             embeddings = cache_embeddings(doc_name, chunks, embed_chunks)
             index = build_faiss_index(embeddings)
 
-            if "registry" not in st.session_state:
-                st.session_state["registry"] = DocumentRegistry()
+            registry = st.session_state.get("registry")
+            if not registry:
+                registry = DocumentRegistry()
+                st.session_state["registry"] = registry
 
-            registry = st.session_state["registry"]
-            doc_id = registry.register(temp_path, chunks, embeddings, index)
-            st.session_state["active_doc"] = doc_id
+            registry.register(temp_path, chunks, embeddings, index)
+            
 
             status.success("✅ Document processed successfully — all set to query your assistant!")