Update src/streamlit_app.py

src/streamlit_app.py

@@ -212,7 +212,7 @@ st.caption("Query SAP documentation and enterprise PDFs — powered by reasoning
 doc_choice = st.radio("Select a document:", ["-- Select --", "Sample PDF", "Upload Custom PDF"], index=0)
 
 # ==========================================================
-# 📂 DOCUMENT HANDLING — CLEAN + ACCURATE UI FLOW
+# 📂 DOCUMENT HANDLING — CLEAN + ACCURATE UI FLOW (final stable)
 # ==========================================================
 if doc_choice == "-- Select --":
     st.info("⬅️ Select or upload a document to begin.")
@@ -234,17 +234,14 @@ else:
             with open(temp_path, "wb") as f:
                 f.write(uploaded_file.getbuffer())
         else:
-            st.stop()  # wait until file uploaded before proceeding
+            st.stop()
 
     # --- Real processing begins here ---
     if temp_path:
         doc_name = os.path.basename(temp_path)
 
-        # ✅ Process only if new document or not processed before
-        if (
-            "doc_ready" not in st.session_state
-            or st.session_state.get("last_doc") != doc_name
-        ):
+        # Process only once per document
+        if "doc_ready" not in st.session_state or st.session_state.get("last_doc") != doc_name:
             status = st.empty()
             status.info("📤 Upload complete — reading document...")
 
@@ -253,12 +250,16 @@ else:
             chunks = chunk_text(text, chunk_size=chunk_size, overlap=overlap)
 
             status.info("🧠 Building embeddings and search index...")
-            embeddings = cache_embeddings(os.path.basename(temp_path), chunks, embed_chunks)
+            embeddings = cache_embeddings(doc_name, chunks, embed_chunks)
             index = build_faiss_index(embeddings)
 
-            status.success("✅ Document processed successfully — hang tight, your AI assistant is gearing up to help you out!")
+            final_message = (
+                "✅ Document processed successfully — hang tight, "
+                "your AI assistant is gearing up to help you out!"
+            )
+            status.success(final_message)
 
-            # Save all data in session state for reuse
+            # Store everything so UI stays consistent on rerun
             st.session_state.update({
                 "text": text,
                 "toc": toc,
@@ -267,30 +268,26 @@ else:
                 "index": index,
                 "doc_ready": True,
                 "last_doc": doc_name,
+                "status_text": final_message
             })
 
+            # Build suggestions once
+            query_suggestions = generate_dynamic_suggestions_from_toc(toc, chunks, doc_name)
+            st.session_state["query_suggestions_fixed"] = query_suggestions
+            st.session_state["user_query_input"] = ""
+            st.session_state["selected_suggestion"] = None
+            st.session_state["show_more"] = False
+            st.rerun()
+
         else:
-            # ✅ Reuse previous data instead of reprocessing
+            # Reuse cached content and status message
             text = st.session_state["text"]
             toc = st.session_state["toc"]
             chunks = st.session_state["chunks"]
             embeddings = st.session_state["embeddings"]
             index = st.session_state["index"]
-            st.info(f"📄 {st.session_state['last_doc']} is ready for queries.")
-
-        # --- Suggestion setup ---
-        if st.session_state.get("last_doc") != doc_name:
-            query_suggestions = generate_dynamic_suggestions_from_toc(toc, chunks, doc_name)
-            st.session_state.update({
-                "query_suggestions_fixed": query_suggestions,
-                "last_doc": doc_name,
-                "user_query_input": "",
-                "selected_suggestion": None,
-                "show_more": False
-            })
-            st.rerun()
-        else:
             query_suggestions = st.session_state.get("query_suggestions_fixed", [])
+            st.info(st.session_state.get("status_text", f"📄 {doc_name} is ready for queries."))
 
         # --- Ask section ---
         st.markdown("### 💬 Ask the Assistant")
@@ -322,6 +319,7 @@ else:
                 answer = re.sub(r"(^|\n)-\s*", r"\1<br>• ", answer)
             st.markdown(f"<div class='answer-box'>{answer}</div>", unsafe_allow_html=True)
 
+
 # ==========================================================
 # 🎨 Optional Sidebar Scroll Styling (keeps it clean)
 # ==========================================================