Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,5 +1,5 @@
 # ==========================================================
-# streamlit_app.py — Restored Stable UI (Pre-rerun Fix)
+# streamlit_app.py — Restored Centered Layout + Green Status UI
 # ==========================================================
 import os
 import re
@@ -8,10 +8,11 @@ from ingestion import extract_text_from_pdf, chunk_text
 from vectorstore import build_faiss_index
 from qa import retrieve_chunks, generate_answer, cache_embeddings, embed_chunks, genai_generate
 
-st.set_page_config(page_title="Enterprise Knowledge Assistant", layout="wide")
+# ✅ Centered layout again (this caused the stretch difference)
+st.set_page_config(page_title="Enterprise Knowledge Assistant", layout="centered")
 
 # ==========================================================
-# 🧭 SIDEBAR — Original Clean Version
+# 🧭 SIDEBAR — As before
 # ==========================================================
 with st.sidebar:
     st.markdown("### 🧭 Response Mode")
@@ -36,7 +37,7 @@ with st.sidebar:
     st.caption("✨ Built by Shubham Sharma")
 
 # ==========================================================
-# 🧠 SESSION STATE — Clean and Minimal
+# 🧠 SESSION STATE
 # ==========================================================
 for key, val in {
     "user_query_input": "",
@@ -54,100 +55,127 @@ def set_user_query(q, idx):
     st.experimental_rerun()
 
 # ==========================================================
-# 📘 APP HEADER
+# 📘 PAGE HEADER (Centered)
 # ==========================================================
 st.markdown(
     """
-    <h1 style="text-align:center;">📄 Enterprise Knowledge Assistant</h1>
-    <p style="text-align:center;color:gray;">
-        Query SAP documentation and enterprise PDFs — powered by reasoning and retrieval.
-    </p>
+    <div style="text-align:center; margin-top:-10px;">
+        <h1>📄 Enterprise Knowledge Assistant</h1>
+        <p style="color:gray; font-size:15px;">
+            Query SAP documentation and enterprise PDFs — powered by reasoning and retrieval.
+        </p>
+    </div>
     """,
     unsafe_allow_html=True,
 )
 
 # ==========================================================
-# 📂 DOCUMENT UPLOAD / SELECTION
+# 📄 MAIN CONTAINER — Centered
 # ==========================================================
-st.markdown("### Select a document:")
-doc_choice = st.radio("", ("-- Select --", "Sample PDF", "Upload Custom PDF"))
-
-temp_path = None
-if doc_choice == "Sample PDF":
-    temp_path = os.path.join(os.path.dirname(__file__), "sample.pdf")
-    st.success("📘 Sample PDF loaded successfully. Ask questions below.")
-elif doc_choice == "Upload Custom PDF":
-    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 — ready to query below.")
-else:
-    st.info("⬅️ Please select or upload a document to begin.")
-
-# ==========================================================
-# 🧠 PROCESS DOCUMENT (only when selected)
-# ==========================================================
-if temp_path:
-    text, toc = extract_text_from_pdf(temp_path)
-    chunks = chunk_text(text, chunk_size=chunk_size, overlap=overlap)
-    st.success("✅ Document ready — you can now ask questions below.")
-
-    embeddings = cache_embeddings(os.path.basename(temp_path), chunks, embed_chunks)
-    index = build_faiss_index(embeddings)
-
-    # ==========================================================
-    # 💬 SUGGESTED QUESTIONS (Smart Defaults)
-    # ==========================================================
-    if not st.session_state["query_suggestions_fixed"]:
-        st.session_state["query_suggestions_fixed"] = [
-            "What is the purpose of this document?",
-            "How can integration be set up in SAP Cloud?",
-            "What are the prerequisites mentioned?",
-            "What steps are involved in configuration?",
-            "How to troubleshoot integration issues?",
-            "What is the key functionality covered?"
-        ]
-
-    st.markdown("### Ask the Assistant")
-
-    visible_qs = st.session_state["query_suggestions_fixed"][:3] if not st.session_state["show_more"] else st.session_state["query_suggestions_fixed"]
-    cols = st.columns(3)
-    for i, q in enumerate(visible_qs):
-        if cols[i % 3].button(f"🔹 {q}", key=f"suggest_{i}"):
-            set_user_query(q, i)
-
-    if st.button("Show more ▼" if not st.session_state["show_more"] else "Show less ▲"):
-        st.session_state["show_more"] = not st.session_state["show_more"]
-
-    # ==========================================================
-    # 🧩 QUERY INPUT
-    # ==========================================================
-    user_query = st.text_input("Type your question or click one above:", value=st.session_state["user_query_input"], key="user_query_input")
-
-    # ==========================================================
-    # 🤖 GENERATE ANSWER
-    # ==========================================================
-    if user_query:
-        reasoning_mode = mode == "Extended (Document + general)"
-        st.markdown(f"**Mode:** {'🧠 Extended' if reasoning_mode else '📄 Strict Document'}")
-        with st.spinner("Thinking..."):
-            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("### Assistant’s Answer")
-        st.markdown(f"<div style='background:#0f172a;border-left:4px solid #3b82f6;padding:12px;border-radius:8px;color:#f1f5f9'>{answer}</div>", unsafe_allow_html=True)
-
-        with st.expander("📘 Supporting Context", expanded=False):
-            for i, chunk in enumerate(retrieved, 1):
-                st.markdown(f"**Chunk {i}:** {chunk.strip()}")
-
-# ==========================================================
-# 🗂️ DOCUMENT PREVIEW
-# ==========================================================
-if temp_path:
-    st.markdown("---")
-    with st.expander("📄 Document Preview", expanded=False):
-        st.text_area("Extracted text (first 1000 chars):", text[:1000], height=180)
-        st.caption(f"📦 {len(chunks)} chunks processed.")
+col_spacer_left, col_main, col_spacer_right = st.columns([0.15, 0.7, 0.15])
+with col_main:
+
+    # ----------------------------------------------------------
+    # 📂 Document Upload / Selection
+    # ----------------------------------------------------------
+    st.markdown("### Select a document:")
+    doc_choice = st.radio("", ("-- Select --", "Sample PDF", "Upload Custom PDF"))
+
+    temp_path = None
+    if doc_choice == "Sample PDF":
+        temp_path = os.path.join(os.path.dirname(__file__), "sample.pdf")
+        st.success("📘 Sample PDF loaded successfully. Ask questions below.")
+        st.success("✅ Document ready — you can now ask questions below.")
+    elif doc_choice == "Upload Custom PDF":
+        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 — ready to query below.")
+            st.success("✅ Document ready — you can now ask questions below.")
+    else:
+        st.info("⬅️ Please select or upload a document to begin.")
+
+    # ----------------------------------------------------------
+    # 🧠 Process document when loaded
+    # ----------------------------------------------------------
+    if temp_path:
+        text, toc = extract_text_from_pdf(temp_path)
+        chunks = chunk_text(text, chunk_size=chunk_size, overlap=overlap)
+        embeddings = cache_embeddings(os.path.basename(temp_path), chunks, embed_chunks)
+        index = build_faiss_index(embeddings)
+
+        # ----------------------------------------------------------
+        # 💡 Suggested Questions
+        # ----------------------------------------------------------
+        if not st.session_state["query_suggestions_fixed"]:
+            st.session_state["query_suggestions_fixed"] = [
+                "What is the purpose of this document?",
+                "How can integration be set up in SAP Cloud?",
+                "What are the prerequisites mentioned?",
+                "What steps are involved in configuration?",
+                "How to troubleshoot integration issues?",
+                "What is the key functionality covered?"
+            ]
+
+        st.markdown("### Ask the Assistant")
+
+        visible_qs = (
+            st.session_state["query_suggestions_fixed"][:3]
+            if not st.session_state["show_more"]
+            else st.session_state["query_suggestions_fixed"]
+        )
+        cols = st.columns(3)
+        for i, q in enumerate(visible_qs):
+            if cols[i % 3].button(f"💬 {q}", key=f"suggest_{i}"):
+                set_user_query(q, i)
+
+        if st.button("Show more ▼" if not st.session_state["show_more"] else "Show less ▲"):
+            st.session_state["show_more"] = not st.session_state["show_more"]
+
+        # ----------------------------------------------------------
+        # 🧩 Query Input
+        # ----------------------------------------------------------
+        user_query = st.text_input(
+            "Type your question or click one above:",
+            value=st.session_state["user_query_input"],
+            key="user_query_input"
+        )
+
+        # ----------------------------------------------------------
+        # 🤖 Generate Answer
+        # ----------------------------------------------------------
+        if user_query:
+            reasoning_mode = mode == "Extended (Document + general)"
+            st.markdown(f"**Mode:** {'🧠 Extended' if reasoning_mode else '📄 Strict Document'}")
+
+            with st.spinner("💭 Generating answer..."):
+                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("### Assistant’s Answer")
+            st.markdown(
+                f"""
+                <div style='background:#0f172a;border-left:4px solid #22c55e;
+                padding:12px;border-radius:8px;color:#f1f5f9'>{answer}</div>
+                """,
+                unsafe_allow_html=True,
+            )
+
+            with st.expander("📘 Supporting Context", expanded=False):
+                for i, chunk in enumerate(retrieved, 1):
+                    st.markdown(f"**Chunk {i}:** {chunk.strip()}")
+
+    # ----------------------------------------------------------
+    # 📑 Document Preview
+    # ----------------------------------------------------------
+    if temp_path:
+        st.markdown("---")
+        with st.expander("📄 Document Preview", expanded=False):
+            st.text_area("Extracted text (first 1000 chars):", text[:1000], height=180)
+            st.caption(f"📦 {len(chunks)} chunks processed.")