Update src/streamlit_app.py

src/streamlit_app.py

@@ -63,10 +63,10 @@ from vectorstore import build_faiss_index
 from qa import retrieve_chunks, generate_answer, cache_embeddings, embed_chunks, genai_generate
 
 # ==========================================================
-# 🧠 Smart Suggestion Generator (Strong Version)
+# 🧠 Smart Suggestion Generator
 # ==========================================================
 def generate_dynamic_suggestions_from_toc(toc, chunks, doc_name="Document"):
-    """Generate 5–7 short, smart, natural suggestions based on document context."""
+    """Generate natural-sounding, concise professional questions from TOC + context."""
     if not toc or not chunks:
         return []
 
@@ -79,16 +79,15 @@ def generate_dynamic_suggestions_from_toc(toc, chunks, doc_name="Document"):
 
     context_sample = " ".join(chunks[:3])[:4000]
     prompt = f"""
-    You are generating intelligent, concise questions for an enterprise assistant.
-    Document name: "{doc_name}"
+    You are an enterprise documentation assistant.
+    Generate 5–7 professional, relevant questions a user might ask about this document: "{doc_name}".
+    Use the table of contents and content sample below.
+
     TABLE OF CONTENTS:
     {chr(10).join(['- ' + t for t in titles[:10]])}
 
     CONTENT SAMPLE:
     {context_sample}
-
-    Generate 5–7 smart, helpful, and professional questions that a user might ask about this document.
-    Keep them short (<20 words), clear, and well-formed.
     """
 
     try:
@@ -105,7 +104,7 @@ def generate_dynamic_suggestions_from_toc(toc, chunks, doc_name="Document"):
         return [
             "What is this document about?",
             "How do I start using this process?",
-            "What key configurations are included?",
+            "What configurations are needed?",
         ]
 
 # ==========================================================
@@ -149,20 +148,23 @@ with st.sidebar:
     st.caption("✨ Built by Shubham Sharma")
 
 # ==========================================================
-# 🧾 Document Handling + Question UI (Perfect Sync)
+# 🧾 Document Handling
 # ==========================================================
 text, chunks, index, embeddings, toc = None, None, None, None, None
 
-# --- Initialize State ---
+# --- Initialize session state ---
 for key, default in {
     "show_more": False,
     "user_query_input": "",
     "selected_suggestion": None,
+    "pending_query": None,
 }.items():
     if key not in st.session_state:
         st.session_state[key] = default
 
-# --- Document Handling ---
+# ----------------------------------------------------------
+# 📄 Document Handling + Ask a Question
+# ----------------------------------------------------------
 if doc_choice == "-- Select --":
     st.info("⬅️ Please choose a document from the sidebar to begin.")
 else:
@@ -184,7 +186,17 @@ else:
             text, toc = extract_text_from_pdf(temp_path)
             chunks = chunk_text(text, chunk_size=chunk_size)
             st.markdown("✅ Document loaded successfully.")
-            query_suggestions = generate_dynamic_suggestions_from_toc(toc, chunks, os.path.basename(temp_path))
+
+            if toc:
+                st.markdown("### 🧭 Table of Contents Detected")
+                toc_text = "\n".join([f"{sec}. {title}" for sec, title in toc])
+                st.text_area("TOC Preview", toc_text, height=180)
+                query_suggestions = generate_dynamic_suggestions_from_toc(toc, chunks, os.path.basename(temp_path))
+            else:
+                query_suggestions = [
+                    "What is this document about?",
+                    "How do I start using this process?",
+                ]
 
         with st.spinner("⚙️ Preparing search index..."):
             embeddings = cache_embeddings(os.path.basename(temp_path), chunks, embed_chunks)
@@ -192,7 +204,7 @@ else:
             st.markdown("🚀 Document ready — you can now ask questions below.")
 
         # ----------------------------------------------------------
-        # 💬 Ask a Question
+        # 💬 Ask a Question (Perfect Suggestion Sync)
         # ----------------------------------------------------------
         st.markdown("## 🤖 Ask a Question")
 
@@ -204,25 +216,29 @@ else:
 
             for i, q in enumerate(visible):
                 col = cols[i % 3]
-                if col.button(f"🔍 {q}", key=f"q_{i}"):
-                    st.session_state.selected_suggestion = i
-                    st.session_state.user_query_input = q
+                if col.button(f"🔍 {q}", key=f"suggest_{i}"):
+                    st.session_state.pending_query = q
+                    st.experimental_rerun()
 
             toggle_text = "Show less ▲" if st.session_state.show_more else "Show more ▼"
             if st.button(toggle_text, key="toggle_show_more"):
                 st.session_state.show_more = not st.session_state.show_more
                 st.experimental_rerun()
 
-        # --- Text Input synced with suggestion ---
+        # --- Handle query sync ---
+        if st.session_state.pending_query:
+            st.session_state.user_query_input = st.session_state.pending_query
+            st.session_state.pending_query = None
+
         user_query = st.text_input(
             "Type your question or pick one above:",
             value=st.session_state.user_query_input,
             key="user_query_input",
         )
 
-        # --- Answer Generation ---
         if user_query.strip():
             st.caption("Mode: 🧠 Reasoning" if st.session_state.reasoning_mode else "Mode: 📄 Strict Document")
+
             with st.spinner("💭 Analyzing your document..."):
                 retrieved = retrieve_chunks(user_query, index, chunks, top_k=top_k, embeddings=embeddings)
                 answer = generate_answer(user_query, retrieved, reasoning_mode=st.session_state.reasoning_mode)
@@ -234,7 +250,9 @@ else:
                 for i, r in enumerate(retrieved, start=1):
                     st.markdown(f"**Chunk {i}:** {r}")
 
-# --- Document Preview ---
+# ----------------------------------------------------------
+# 📑 Document Preview
+# ----------------------------------------------------------
 if chunks:
     st.markdown("---")
     st.subheader("📑 Document Preview")