Update app.py

app.py

@@ -1,10 +1,12 @@
 import streamlit as st
 import time
+from rag_engine import process_query, load_model
+from utils import setup_all_auth
 
-# FIRST: Set page config before ANY other Streamlit command
-st.set_page_config(page_title="Indian Spirituality Q&A")
+# FIRST: Set page config
+st.set_page_config(page_title="Spirituality Q&A")
 
-# Initialize ALL session state variables at the start
+# Initialize session state variables
 if 'initialized' not in st.session_state:
     st.session_state.initialized = False
 if 'model' not in st.session_state:
@@ -17,111 +19,87 @@ 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
 
-# THEN: Import your modules
-from rag_engine import process_query, load_model
-from utils import setup_all_auth
-
-# Create a placeholder for our initialization message
-init_message = st.empty()
-
-# Custom styling (pure CSS)
+# CSS Styling
 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: #81C784 !important;
+    background-color: #4CAF50 !important;
     color: white !important;
+    border: 2px solid #388E3C !important;
     border-radius: 8px !important;
-    border: none !important;
-    padding: 0.6rem 1.2rem !important;
-    font-size: 1rem !important;
-    transition: 0.3s ease-in-out;
+    box-shadow: 0px 3px 8px rgba(0, 0, 0, 0.2);
 }
 .stButton>button:hover {
     background-color: #66BB6A !important;
+    border-color: #388E3C !important;
 }
-
-/* Remove orange focus outline */
-.stButton>button:focus {
-    outline: none !important;
-    box-shadow: 0px 0px 8px rgba(129, 199, 132, 0.8) !important;
+div[data-baseweb="input"] {
+    border: 2px solid """ + st.session_state.border_color + """ !important;
+    border-radius: 8px !important;
+}
+div[data-baseweb="input"]:focus-within {
+    border: 2px solid #FF5722 !important;
+    border-radius: 8px !important;
 }
-
 </style>
-<div class="main-title">Indian Spirituality Q&A</div>
+<div class="main-title">Spirituality Q&A</div>
 """, unsafe_allow_html=True)
 
-# Handle initialization and success message timing
+# Initialization logic
 if not st.session_state.initialized:
-    init_message.info("Hang in there! We are setting the system up for you. 😊")
-
+    init_message = st.info("Hang in there! Setting up the system 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)  # Ensure message appears before rerun
+        time.sleep(0.1)  # Small delay
         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
+# Handle form submission
+if 'form_submitted' not in st.session_state:
+    st.session_state.form_submitted = False
 
-# Function to 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.query_input = ""  # Clear input field
+        st.session_state.form_submitted = True
+        st.session_state.border_color = "#4CAF50"  # Reset to green
         st.rerun()
 
-# Form for user input
+# Create form for user input
 with st.form(key="query_form"):
-    query = st.text_input("Ask your question:", key="query_input", value="", placeholder="Press enter to submit question")
+    query = st.text_input("", key="query_input", placeholder="Press enter to submit question")
     submit_button = st.form_submit_button("Get Answer", on_click=handle_form_submit)
 
-# Display last submitted question
+# Display last question
 if st.session_state.last_query:
     st.markdown("### Current Question:")
     st.info(st.session_state.last_query)
 
-# Sliders for customization
+# Customization sliders
 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 on submission
+# Process query
 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)
@@ -133,7 +111,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
+# About section
 st.markdown("---")
 st.markdown("""
 ### About this app