Update src/streamlit_app.py

src/streamlit_app.py

@@ -74,20 +74,16 @@ def generate_dynamic_suggestions_from_toc(toc, chunks, doc_name="Document"):
 # ==========================================================
 st.markdown("""
 <style>
-div.block-container {
-    padding-top: 1rem;
-    max-width: 1100px;
-}
+div.block-container { padding-top: 1rem; max-width: 1100px; }
 h2, h3, h4 { color: #f3f4f6; font-weight: 600; }
 h3 { font-size: 1.1rem; margin-bottom: 0.4rem; }
-h4 { font-size: 1rem; margin-bottom: 0.2rem; color: #9ca3af; }
 .status-line {
     background: #0f172a;
     border-left: 4px solid #10b981;
     border-radius: 6px;
     padding: 8px 14px;
     color: #d1fae5;
-    margin-bottom: 0.8rem;
+    margin-bottom: 0.6rem;
 }
 .suggest-chip {
     background-color: #111827;
@@ -115,17 +111,13 @@ h4 { font-size: 1rem; margin-bottom: 0.2rem; color: #9ca3af; }
     padding: 8px 12px;
     font-size: 14px;
 }
-.stTextInput > div > div > input:focus {
-    border-color: #3b82f6;
-    box-shadow: 0 0 6px rgba(59,130,246,0.3);
-}
 .answer-box {
     background: linear-gradient(135deg, #0f172a, #1e293b);
     border-left: 4px solid #3b82f6;
     border-radius: 8px;
     padding: 16px 18px;
     color: #f1f5f9;
-    margin-top: 0.6rem;
+    margin-top: 0.5rem;
 }
 section.ask-block { margin-top: 0.5rem; margin-bottom: 0.2rem; }
 section.answer-block { margin-top: 0.2rem; }
@@ -133,18 +125,19 @@ section.answer-block { margin-top: 0.2rem; }
 """, unsafe_allow_html=True)
 
 # ==========================================================
-# 🧭 Sidebar
+# 🧭 Sidebar (simplified)
 # ==========================================================
 with st.sidebar:
     st.markdown("### 🧠 Response Mode")
     mode = st.radio("", ["Strict (Document-only)", "Extended (Document + general)"], index=0)
     reasoning_mode = mode.startswith("Extended")
+    st.caption("Strict = answers only from the document. Extended = may include helpful general info.")
 
-    st.markdown("---")
-    with st.expander("⚙️ Advanced Settings", expanded=False):
-        chunk_size = st.slider("Chunk Size (characters)", 200, 1500, 1000, step=50)
+    with st.expander("⚙️ Advanced Settings (for developers)", expanded=False):
+        chunk_size = st.slider("Chunk Size", 200, 1500, 1000, step=50)
         overlap = st.slider("Chunk Overlap", 50, 200, 120, step=10)
         top_k = st.slider("Top K Results", 1, 10, 5)
+
     st.markdown("---")
     st.caption("✨ Built by Shubham Sharma")
 
@@ -179,14 +172,14 @@ if doc_choice == "-- Select --":
 else:
     if doc_choice == "Sample PDF":
         temp_path = os.path.join(os.path.dirname(__file__), "sample.pdf")
-        st.success("📘 Using built-in Sample PDF.")
+        st.markdown("<div class='status-line'>📘 Using built-in Sample PDF — ready to query below.</div>", unsafe_allow_html=True)
     else:
         uploaded_file = st.file_uploader("📂 Upload your PDF", type="pdf")
         if uploaded_file:
             temp_path = os.path.join("/tmp", uploaded_file.name)
             with open(temp_path, "wb") as f:
                 f.write(uploaded_file.getbuffer())
-            st.success(f"✅ '{uploaded_file.name}' uploaded successfully.")
+            st.markdown(f"<div class='status-line'>✅ '{uploaded_file.name}' uploaded successfully — ready to query below.</div>", unsafe_allow_html=True)
         else:
             temp_path = None
 
@@ -207,7 +200,6 @@ else:
         with st.spinner("⚙️ Building FAISS index..."):
             embeddings = cache_embeddings(os.path.basename(temp_path), chunks, embed_chunks)
             index = build_faiss_index(embeddings)
-            st.markdown("<div class='status-line'>✅ Document ready — you can now ask questions below.</div>", unsafe_allow_html=True)
 
         # ----------------------------------------------------------
         # 💬 Ask a Question
@@ -222,15 +214,11 @@ else:
                 if cols[i % 3].button(f"🔍 {q}", key=f"sugg_{i}", on_click=select_suggestion, args=(q,)):
                     pass
 
-            toggle_text = "Show less ▲" if st.session_state.show_more else "More suggestions ▼"
+            toggle_text = "Show all sample questions ▲" if st.session_state.show_more else "Show all sample questions ▼"
             if st.button(toggle_text, key="toggle_sugg"):
                 st.session_state.show_more = not st.session_state.show_more
                 st.experimental_rerun()
 
-        # 🧠 Clean version (no yellow warning)
-        if "user_query_input" not in st.session_state:
-            st.session_state.user_query_input = ""
-
         user_query = st.text_input("Type your question or click one above:", key="user_query_input")
         st.markdown("</section>", unsafe_allow_html=True)
 
@@ -238,24 +226,23 @@ else:
         # 🤖 Assistant
         # ----------------------------------------------------------
         if user_query.strip():
-            with st.spinner("💭 Analyzing document..."):
+            with st.spinner("💭 Generating response..."):
                 retrieved = retrieve_chunks(user_query, index, chunks, top_k=top_k, embeddings=embeddings)
                 answer = generate_answer(user_query, retrieved, reasoning_mode=reasoning_mode)
 
             st.markdown("<section class='answer-block'>", unsafe_allow_html=True)
             st.subheader("Assistant")
             st.markdown(f"<div class='answer-box'>💡 {answer}</div>", unsafe_allow_html=True)
-            st.caption("Strict mode = document-only; Extended = may include general context.")
             st.markdown("</section>", unsafe_allow_html=True)
 
-            with st.expander("📄 Supporting Context"):
+            with st.expander("🧩 See Source Passages"):
                 for i, r in enumerate(retrieved, start=1):
                     st.markdown(f"**Chunk {i}:** {r}")
 
         # ----------------------------------------------------------
         # 📚 Explore Document
         # ----------------------------------------------------------
-        with st.expander("📚 Explore Document"):
+        with st.expander("📖 View Original Document Content"):
             if toc:
                 st.markdown("**Table of Contents**")
                 toc_text = "\n".join([f"{sec}. {title}" for sec, title in toc])