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 (final polished)
+# 📂 DOCUMENT HANDLING — POLISHED UI FLOW
 # ==========================================================
 if doc_choice == "-- Select --":
     st.info("⬅️ Select or upload a document to begin.")
@@ -234,7 +234,7 @@ else:
             with open(temp_path, "wb") as f:
                 f.write(uploaded_file.getbuffer())
         else:
-            st.stop()
+            st.stop()  # Wait until file is uploaded
 
     # --- Real processing begins here ---
     if temp_path:
@@ -243,23 +243,22 @@ else:
         # 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...")
+            status.info("📤 Upload complete — hang tight while we process your document...")
 
+            # Step 1: Extract text
             text, toc, toc_source = extract_text_from_pdf(temp_path)
             status.info("📑 Parsing and chunking document...")
             chunks = chunk_text(text, chunk_size=chunk_size, overlap=overlap)
 
+            # Step 2: Build embeddings + FAISS index
             status.info("🧠 Building embeddings and search index...")
             embeddings = cache_embeddings(doc_name, chunks, embed_chunks)
             index = build_faiss_index(embeddings)
 
-            # ✅ Smooth transition for ready state
-            import time
-            status.success("✅ Document processed successfully.")
-            time.sleep(1.2)
-            status.empty()
+            # Step 3: Final ready state
+            status.success("✅ All set — your AI assistant is ready to help.")
 
-            # ✅ Simplified final ready message for reruns
+            # Persist session data for reruns
             st.session_state.update({
                 "text": text,
                 "toc": toc,
@@ -268,19 +267,21 @@ else:
                 "index": index,
                 "doc_ready": True,
                 "last_doc": doc_name,
-                "status_text": f"📄 {doc_name} ready — your AI assistant is standing by!"
+                "status_text": "📄 Document is ready for queries."
             })
 
-            # Build suggestions once
+            # Build question suggestions (once per doc)
             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
+
+            # Refresh to cleanly show "ready" state
             st.rerun()
 
         else:
-            # Reuse cached content and show final ready message with fade-in
+            # --- Reuse existing state (rerun-safe) ---
             text = st.session_state["text"]
             toc = st.session_state["toc"]
             chunks = st.session_state["chunks"]
@@ -288,28 +289,9 @@ else:
             index = st.session_state["index"]
             query_suggestions = st.session_state.get("query_suggestions_fixed", [])
 
-            # ✨ Fading "ready" message for better UX
-            st.markdown(f"""
-            <div style="
-                animation: fadeIn 1.2s ease-in-out;
-                background: #072541;
-                border-left: 4px solid #10b981;
-                padding: 10px 14px;
-                border-radius: 6px;
-                color: #d1fae5;
-                margin-top: 8px;
-                font-size: 15px;">
-                ✅ {st.session_state.get('status_text', f"{doc_name} ready for queries.")}
-            </div>
-            <style>
-            @keyframes fadeIn {{
-                from {{opacity: 0; transform: translateY(-4px);}}
-                to {{opacity: 1; transform: translateY(0);}}
-            }}
-            </style>
-            """, unsafe_allow_html=True)
-
-        # --- Ask section ---
+            st.info(st.session_state.get("status_text", f"📄 {doc_name} is ready for queries."))
+
+        # --- Ask the Assistant section ---
         st.markdown("### 💬 Ask the Assistant")
         if query_suggestions:
             visible = query_suggestions if st.session_state["show_more"] else query_suggestions[:3]
@@ -323,6 +305,7 @@ else:
                 st.session_state["show_more"] = not st.session_state["show_more"]
                 st.rerun()
 
+        # --- Query input box ---
         user_query = st.text_input("Type your question or click one above:", key="user_query_input")
 
         if user_query.strip():
@@ -334,13 +317,13 @@ else:
 
             st.markdown("### 🤖 Assistant’s Answer")
 
+            # Clean up answer format (bullets, bold)
             if not reasoning_mode and not answer.startswith("⚠️"):
                 answer = re.sub(r"\*\*(.*?)\*\*", r"\1", answer)
                 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)
 # ==========================================================