Update src/streamlit_app.py

src/streamlit_app.py

@@ -179,20 +179,34 @@ if chunks:
     avg_len = int(sum(len(c) for c in chunks) / len(chunks))
     st.caption(f"📦 {len(chunks)} chunks created | Avg chunk length: {avg_len} chars")
 
-# ==========================================================
-# 💬 Query Section
-# ==========================================================
+# ---------------------------
+# Query Section
+# ---------------------------
 if index and chunks:
     st.markdown("---")
     st.subheader("🤖 Ask a Question")
 
     user_query = st.text_input("🔍 Your question about the document:")
+
     if user_query:
+        # Show which mode is active
+        mode_label = (
+            "🧠 Reasoning Mode (expanded thinking)"
+            if st.session_state.reasoning_mode
+            else "📄 Strict Document Mode (factual only)"
+        )
+        st.caption(f"Mode: {mode_label}")
+
+        # Generate the answer
         with st.spinner("🧠 Thinking... retrieving context and generating answer..."):
             retrieved = retrieve_chunks(user_query, index, chunks, top_k=top_k)
-            answer = generate_answer(user_query, retrieved, reasoning_mode=reasoning_mode)
+            answer = generate_answer(
+                user_query,
+                retrieved,
+                reasoning_mode=st.session_state.reasoning_mode  # ✅ FIXED
+            )
 
-        # ✅ Answer Display
+        # ✅ Display Answer
         st.markdown("### ✅ Assistant’s Answer")
         st.markdown(
             f"<div style='background-color:#0E1117;padding:12px;border-radius:10px;color:white;'>{answer}</div>",
@@ -210,5 +224,7 @@ if index and chunks:
                     """,
                     unsafe_allow_html=True,
                 )
+
 else:
     st.info("📥 Upload or select a document to start exploring.")
+