Update app.py

app.py

@@ -1,12 +1,10 @@
 import streamlit as st
 import time
-from rag_engine import process_query, load_model
-from utils import setup_all_auth
 
-# FIRST: Set page config
+# FIRST: Set page config before ANY other Streamlit command
 st.set_page_config(page_title="Spirituality Q&A")
 
-# Initialize session state variables
+# Initialize ALL session state variables right at the beginning
 if 'initialized' not in st.session_state:
     st.session_state.initialized = False
 if 'model' not in st.session_state:
@@ -19,10 +17,12 @@ if 'submit_clicked' not in st.session_state:
     st.session_state.submit_clicked = False
 if 'init_time' not in st.session_state:
     st.session_state.init_time = None
-if 'border_color' not in st.session_state:
-    st.session_state.border_color = "#4CAF50"  # Green by default
 
-# CSS Styling
+# THEN: Import your modules
+from rag_engine import process_query, load_model
+from utils import setup_all_auth
+
+# Custom styling (pure CSS)
 st.markdown("""
 <style>
 .main-title {
@@ -34,18 +34,19 @@ st.markdown("""
 .stButton>button {
     background-color: #4CAF50 !important;
     color: white !important;
-    border: 2px solid #388E3C !important;
+    border: none !important;
     border-radius: 8px !important;
-    box-shadow: 0px 3px 8px rgba(0, 0, 0, 0.2);
+    padding: 8px 16px !important;
 }
 .stButton>button:hover {
-    background-color: #66BB6A !important;
-    border-color: #388E3C !important;
+    background-color: #45A049 !important;
 }
+/* Default input box - Green */
 div[data-baseweb="input"] {
-    border: 2px solid """ + st.session_state.border_color + """ !important;
+    border: 2px solid #4CAF50 !important;
     border-radius: 8px !important;
 }
+/* While editing - Red */
 div[data-baseweb="input"]:focus-within {
     border: 2px solid #FF5722 !important;
     border-radius: 8px !important;
@@ -54,52 +55,57 @@ div[data-baseweb="input"]:focus-within {
 <div class="main-title">Spirituality Q&A</div>
 """, unsafe_allow_html=True)
 
-# Initialization logic
+# Handle initialization and success message timing
+init_message = st.empty()
+
 if not st.session_state.initialized:
-    init_message = st.info("Hang in there! Setting up the system for you. 😊")
+    init_message.info("Hang in there! We are setting the system up for you. 😊")
+
     try:
         setup_all_auth()
         load_model()
         st.session_state.initialized = True
         st.session_state.init_time = time.time()
         init_message.success("System initialized successfully!")
-        time.sleep(0.1)  # Small delay
-        st.rerun()
+        time.sleep(0.1)
+        st.experimental_rerun()
     except Exception as e:
         init_message.error(f"Error initializing: {str(e)}")
 
-# Handle form submission
-if 'form_submitted' not in st.session_state:
-    st.session_state.form_submitted = False
+elif st.session_state.init_time is not None:
+    elapsed_time = time.time() - st.session_state.init_time
+    if elapsed_time >= 2.0:
+        init_message.empty()
+        st.session_state.init_time = None
 
+# Handle form submission
 def handle_form_submit():
     if st.session_state.query_input:
         st.session_state.last_query = st.session_state.query_input
         st.session_state.submit_clicked = True
-        st.session_state.form_submitted = True
-        st.session_state.border_color = "#4CAF50"  # Reset to green
-        st.rerun()
+        st.session_state.query_input = ""
 
-# Create form for user input
+# Create a form for handling the user input
 with st.form(key="query_form"):
-    query = st.text_input("", key="query_input", placeholder="Press enter to submit question")
+    query = st.text_input("Ask your question:", key="query_input", placeholder="Press enter to submit question")
     submit_button = st.form_submit_button("Get Answer", on_click=handle_form_submit)
 
-# Display last question
+# Display the current question if there is one
 if st.session_state.last_query:
     st.markdown("### Current Question:")
     st.info(st.session_state.last_query)
 
-# Customization sliders
+# Sliders for customization
 col1, col2 = st.columns(2)
 with col1:
     top_k = st.slider("Number of sources:", 3, 10, 5)
 with col2:
     word_limit = st.slider("Word limit:", 50, 500, 200)
 
-# Process query
+# Only process the query if explicitly submitted
 if st.session_state.submit_clicked and st.session_state.last_query:
     st.session_state.submit_clicked = False
+
     with st.spinner("Processing your question..."):
         try:
             result = process_query(st.session_state.last_query, top_k=top_k, word_limit=word_limit)
@@ -111,7 +117,7 @@ if st.session_state.submit_clicked and st.session_state.last_query:
         except Exception as e:
             st.error(f"Error processing query: {str(e)}")
 
-# About section
+# Add helpful information
 st.markdown("---")
 st.markdown("""
 ### About this app