Update app.py

app.py

@@ -8,6 +8,18 @@ st.set_page_config(page_title="Indian Spiritual RAG")
 from rag_engine import process_query, load_model
 from utils import setup_all_auth
 
+# Initialize session state variables (at the very top)
+if 'initialized' not in st.session_state:
+    st.session_state.initialized = False
+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 'show_success' not in st.session_state:
+    st.session_state.show_success = False
+if 'success_time' not in st.session_state:
+    st.session_state.success_time = None
+
 # Display title with custom styling
 st.markdown("""
 <style>
@@ -21,7 +33,17 @@ st.markdown("""
     border: 2px solid #FF5722 !important;
     border-radius: 8px !important;
 }
-/* Set default green border for all input fields */
+/* Remove any default styling for input fields */
+div[data-baseweb="input"] {
+    border: none !important;
+    box-shadow: none !important;
+}
+div[data-baseweb="input"] > div {
+    background-color: transparent !important;
+}
+div[data-baseweb="input"] > div > div {
+    box-shadow: none !important;
+}
 .stTextInput>div>div>input {
     border: 2px solid #4CAF50 !important;
     border-radius: 8px !important;
@@ -30,37 +52,44 @@ st.markdown("""
 </style>
 
 <script>
-// Run once DOM is fully loaded
 document.addEventListener('DOMContentLoaded', function() {
-    // Function to add events to input elements
-    function addInputEvents() {
-        // Find all input elements in text inputs
+    function applyInputStyles() {
         const inputs = document.querySelectorAll('.stTextInput input');
         
         inputs.forEach(function(input) {
-            // Focus event - red border
-            input.addEventListener('focus', function() {
-                this.style.border = '2px solid #FF5722 !important';
-                this.style.boxShadow = 'none !important';
+            // Clean up old event listeners if any
+            const newInput = input.cloneNode(true);
+            input.parentNode.replaceChild(newInput, input);
+            
+            // Set initial green border
+            newInput.style.border = '2px solid #4CAF50';
+            newInput.style.borderRadius = '8px';
+            newInput.style.boxShadow = 'none';
+            
+            // Focus - change to red
+            newInput.addEventListener('focus', function() {
+                this.style.border = '2px solid #FF5722';
+                this.style.borderRadius = '8px';
+                this.style.boxShadow = 'none';
             });
             
-            // Blur event - green border
-            input.addEventListener('blur', function() {
-                this.style.border = '2px solid #4CAF50 !important';
-                this.style.boxShadow = 'none !important';
+            // Blur - back to green
+            newInput.addEventListener('blur', function() {
+                this.style.border = '2px solid #4CAF50';
+                this.style.borderRadius = '8px';
+                this.style.boxShadow = 'none';
             });
         });
     }
     
-    // Call immediately
-    addInputEvents();
+    // Run initially
+    applyInputStyles();
     
-    // Watch for changes to the DOM (when Streamlit adds new elements)
+    // Watch for changes
     const observer = new MutationObserver(function() {
-        addInputEvents();
+        applyInputStyles();
     });
     
-    // Start observing
     observer.observe(document.body, {
         childList: true,
         subtree: true
@@ -71,47 +100,40 @@ document.addEventListener('DOMContentLoaded', function() {
 <div class="main-title">Indian Spiritual Texts Q&A</div>
 """, unsafe_allow_html=True)
 
-# Initialize session state
-if 'initialized' not in st.session_state:
-    st.session_state.initialized = False
-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 'init_start_time' not in st.session_state:
-    st.session_state.init_start_time = None
-
 # Create placeholder for initialization message
 init_message = st.empty()
 
-# Handle initialization
+# Handle initialization only once
 if not st.session_state.initialized:
+    # Show the loading message
+    init_message.info("Hang in there! We are setting the system up for you. 😊")
+    
     try:
-        # Show the loading message
-        init_message.info("Hang in there! We are setting the system up for you. 😊")
-        
-        # Record start time
-        st.session_state.init_start_time = time.time()
-        
         # Setup authentication
         setup_all_auth()
         
-        # Force model loading at startup to avoid session state issues
+        # Force model loading
         load_model()
         
         # Mark as initialized
         st.session_state.initialized = True
         
+        # Set flags to show success message temporarily
+        st.session_state.show_success = True
+        st.session_state.success_time = time.time()
+        
         # Show success message
         init_message.success("System initialized successfully!")
     except Exception as e:
-        st.error(f"Error initializing: {str(e)}")
-        st.session_state.init_start_time = None
-# Check if we need to clear the success message (if it's been more than 2 seconds since initialization)
-elif st.session_state.init_start_time is not None:
-    if time.time() - st.session_state.init_start_time > 2:
-        init_message.empty()
-        st.session_state.init_start_time = None
+        init_message.error(f"Error initializing: {str(e)}")
+        st.stop()  # Stop execution if initialization fails
+else:
+    # Check if we need to clear the success message
+    current_time = time.time()
+    if st.session_state.show_success:
+        if current_time - st.session_state.success_time > 2:
+            init_message.empty()
+            st.session_state.show_success = False
 
 # Function to process when enter is pressed or button is clicked
 def process_input():