Update src/streamlit_app.py

src/streamlit_app.py

@@ -108,7 +108,22 @@ h1, h2, h3 {color: #f3f4f6; font-weight: 600;}
 """, unsafe_allow_html=True)
 
 # ==========================================================
-# 🧭 SIDEBAR — STABLE, NO RERUN
+# 🧠 SESSION STATE (initialize before widgets)
+# ==========================================================
+for key, val in {
+    "user_query_input": "",
+    "show_more": False,
+    "selected_suggestion": None,
+    "query_suggestions_fixed": None,
+    "last_doc": None,
+    "show_advanced": False,  # keep persistent checkbox state
+}.items():
+    if key not in st.session_state:
+        st.session_state[key] = val
+
+
+# ==========================================================
+# 🧭 SIDEBAR — STABLE VERSION (no rerun reloads)
 # ==========================================================
 with st.sidebar:
     st.markdown("### 🧭 Response Mode")
@@ -117,53 +132,48 @@ with st.sidebar:
         ("Strict (Document-only)", "Extended (Document + general)"),
         index=0,
         help="Strict = answers only from the document. Extended = may include general context.",
+        key="response_mode"
     )
 
     st.markdown("---")
 
-    # 🧠 Advanced settings — persistent, no rerun
-    if "show_advanced" not in st.session_state:
-        st.session_state.show_advanced = False
-
-    show_advanced_now = st.checkbox(
+    # ✅ Checkbox linked to session state (no rerun logic, Streamlit handles it automatically)
+    st.checkbox(
         "Show advanced settings (for developers)",
-        value=st.session_state.show_advanced,
-        key="show_adv_checkbox"
+        key="show_advanced",
+        help="Toggle developer settings without reloading document or clearing outputs."
     )
 
-    # Detect toggle change without triggering full rerun
-    if show_advanced_now != st.session_state.show_advanced:
-        st.session_state.show_advanced = show_advanced_now
-        st.stop()  # ✅ Prevents full rerun — keeps doc & state intact
-
+    # ✅ Sliders appear conditionally, state persists
     if st.session_state.show_advanced:
         st.markdown("### Developer Settings")
-        chunk_size = st.slider("Chunk Size (characters)", 200, 1500, 1000, step=50)
-        overlap = st.slider("Chunk Overlap (characters)", 50, 200, 120, step=10)
-        top_k = st.slider("Top K Results", 1, 10, 5)
+        chunk_size = st.slider("Chunk Size (characters)", 200, 1500, 1000, step=50, key="chunk_slider")
+        overlap = st.slider("Chunk Overlap (characters)", 50, 200, 120, step=10, key="overlap_slider")
+        top_k = st.slider("Top K Results", 1, 10, 5, key="topk_slider")
     else:
         chunk_size, overlap, top_k = 1000, 120, 5
 
     st.markdown("---")
     st.caption("✨ Built by Shubham Sharma")
 
+
 # ==========================================================
-# 🧠 SESSION STATE INITIALIZATION
+# 🧩 QUERY HANDLER
 # ==========================================================
-for key, val in {
-    "user_query_input": "",
-    "show_more": False,
-    "selected_suggestion": None,
-    "query_suggestions_fixed": None,
-    "last_doc": None,
-}.items():
-    if key not in st.session_state:
-        st.session_state[key] = val
-
 def set_user_query(q, idx):
     st.session_state["user_query_input"] = q
     st.session_state["selected_suggestion"] = idx
-    st.experimental_rerun()
+    # ⚠️ No explicit rerun — Streamlit does this automatically
+
+
+# ==========================================================
+# 🧠 USER INPUT
+# ==========================================================
+user_query = st.text_input(
+    "Type your question or click one above:",
+    key="user_query_input"
+)
+
 
 # ==========================================================
 # 📄 MAIN LAYOUT