Update app.py

app.py

@@ -1,80 +1,112 @@
 import streamlit as st
 import time
 
-st.set_page_config(page_title="Indian Spiritual RAG")
+# FIRST: Set page config before ANY other Streamlit command
+st.set_page_config(page_title="Indian Spirituality Q&A")
 
-# Initialize session state variables
+# Initialize ALL session state variables at the start
 if 'initialized' not in st.session_state:
     st.session_state.initialized = False
+if 'model' not in st.session_state:
+    st.session_state.model = None
+if 'tokenizer' not in st.session_state:
+    st.session_state.tokenizer = None
 if 'last_query' not in st.session_state:
     st.session_state.last_query = ""
 if 'submit_clicked' not in st.session_state:
     st.session_state.submit_clicked = False
-if 'query_input' not in st.session_state:
-    st.session_state.query_input = ""
+if 'init_time' not in st.session_state:
+    st.session_state.init_time = None
 
-# Import other modules
+# THEN: Import your modules
 from rag_engine import process_query, load_model
 from utils import setup_all_auth
 
-# Custom CSS for styling
+# Create a placeholder for our initialization message
+init_message = st.empty()
+
+# Custom styling (pure CSS)
 st.markdown("""
 <style>
+/* Main title */
 .main-title {
     font-size: 2.5rem;
     color: #FF5722;
     text-align: center;
     margin-bottom: 1rem;
 }
+
+/* Input field: Green by default, Red when editing */
+div[data-baseweb="input"] {
+    border: 2px solid #81C784 !important;
+    border-radius: 8px !important;
+}
+div[data-baseweb="input"]:focus-within {
+    border: 2px solid #FF5722 !important;
+    border-radius: 8px !important;
+}
+
+/* Custom Button: Green with Hover */
 .stButton>button {
-    background-color: #4CAF50 !important;
+    background-color: #81C784 !important;
     color: white !important;
     border-radius: 8px !important;
-    font-weight: bold !important;
+    border: none !important;
+    padding: 0.6rem 1.2rem !important;
+    font-size: 1rem !important;
+    transition: 0.3s ease-in-out;
 }
-div[data-baseweb="input"] {
-    border: 3px solid #4CAF50 !important;
-    border-radius: 8px !important;
-    transition: border-color 0.3s ease-in-out;
+.stButton>button:hover {
+    background-color: #66BB6A !important;
 }
-div[data-baseweb="input"]:focus-within {
-    border: 3px solid #FF5722 !important;
+
+/* Remove orange focus outline */
+.stButton>button:focus {
+    outline: none !important;
+    box-shadow: 0px 0px 8px rgba(129, 199, 132, 0.8) !important;
 }
+
 </style>
-<div class="main-title">Indian Spiritual Texts Q&A</div>
+<div class="main-title">Indian Spirituality Q&A</div>
 """, unsafe_allow_html=True)
 
-# Initialization logic
+# Handle initialization and success message timing
 if not st.session_state.initialized:
-    init_message = st.empty()
     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
-        time.sleep(0.5)
+        st.session_state.init_time = time.time()
+
         init_message.success("System initialized successfully!")
-        time.sleep(1)
-        init_message.empty()
+        time.sleep(0.1)  # Ensure message appears before rerun
+        st.rerun()
     except Exception as e:
         init_message.error(f"Error initializing: {str(e)}")
 
+# Remove success message after 2 seconds
+elif st.session_state.init_time is not None:
+    if time.time() - st.session_state.init_time >= 2.0:
+        init_message.empty()
+        st.session_state.init_time = None
+
 # Function to handle form submission
 def handle_form_submit():
-    st.session_state.last_query = st.session_state.query_input
-    st.session_state.submit_clicked = True
-    st.session_state.query_input = ""  # ✅ Clears input field on submission
+    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.query_input = ""  # Clear input field
+        st.rerun()
 
 # Form for user input
 with st.form(key="query_form"):
-    query = st.text_input(
-        "Ask your question:", 
-        key="query_input", 
-        value=st.session_state.query_input
-    )
+    query = st.text_input("Ask your question:", key="query_input", value="", placeholder="Press enter to submit question")
     submit_button = st.form_submit_button("Get Answer", on_click=handle_form_submit)
 
-# Display last question
+# Display last submitted question
 if st.session_state.last_query:
     st.markdown("### Current Question:")
     st.info(st.session_state.last_query)
@@ -86,9 +118,10 @@ with col1:
 with col2:
     word_limit = st.slider("Word limit:", 50, 500, 200)
 
-# Process query when submitted
+# Process query on submission
 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)
@@ -100,7 +133,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)}")
 
-# Footer
+# About Section
 st.markdown("---")
 st.markdown("""
 ### About this app