Update app.py

app.py

@@ -8,14 +8,6 @@ st.set_page_config(page_title="Indian Spiritual RAG")
 from rag_engine import process_query, load_model
 from utils import setup_all_auth
 
-# Define state for input box style
-if 'input_focused' not in st.session_state:
-    st.session_state.input_focused = False
-
-# Function to update input focus state
-def set_input_focus(focused):
-    st.session_state.input_focused = focused
-
 # Display title with custom styling
 st.markdown("""
 <style>
@@ -25,55 +17,70 @@ st.markdown("""
     text-align: center;
     margin-bottom: 1rem;
 }
-.button-style {
-    border: 2px solid #FF5722;
-    border-radius: 8px;
-}
 .stButton>button {
     border: 2px solid #FF5722 !important;
     border-radius: 8px !important;
 }
-/* Default style for input (green) */
-.stTextInput>div>div>input {
-    border: 2px solid #4CAF50 !important;
-    border-radius: 8px !important;
+/* Override Streamlit's default input styling */
+div[data-baseweb="input"] {
+    border-radius: 8px;
+}
+div[data-baseweb="input"] > div {
+    background-color: transparent;
 }
 </style>
 
 <script>
 document.addEventListener('DOMContentLoaded', function() {
-    // Function to apply focus styles
-    function applyFocusStyles() {
-        const inputElements = document.querySelectorAll('.stTextInput input');
+    function setupInputStyles() {
+        const inputs = document.querySelectorAll('.stTextInput input');
         
-        inputElements.forEach(function(input) {
-            // Add focus event
-            input.addEventListener('focus', function() {
-                this.style.border = '2px solid #FF5722 !important';
-                this.style.borderWidth = '2px !important';
-                this.style.borderStyle = 'solid !important';
-                this.style.borderColor = '#FF5722 !important';
-            });
+        inputs.forEach(function(input) {
+            // Remove any existing listeners to prevent duplicates
+            input.removeEventListener('focus', onFocus);
+            input.removeEventListener('blur', onBlur);
+            
+            // Set initial style (green)
+            const parent = input.closest('div[data-baseweb="input"]');
+            if (parent) {
+                parent.style.border = '2px solid #4CAF50';
+                parent.style.boxShadow = 'none';
+            }
             
-            // Add blur event
-            input.addEventListener('blur', function() {
-                this.style.border = '2px solid #4CAF50 !important';
-                this.style.borderWidth = '2px !important';
-                this.style.borderStyle = 'solid !important';
-                this.style.borderColor = '#4CAF50 !important';
-            });
+            // Add new listeners
+            input.addEventListener('focus', onFocus);
+            input.addEventListener('blur', onBlur);
         });
     }
     
-    // Initial call
-    applyFocusStyles();
+    function onFocus(event) {
+        const parent = event.target.closest('div[data-baseweb="input"]');
+        if (parent) {
+            parent.style.border = '2px solid #FF5722';
+            parent.style.boxShadow = 'none';
+        }
+    }
+    
+    function onBlur(event) {
+        const parent = event.target.closest('div[data-baseweb="input"]');
+        if (parent) {
+            parent.style.border = '2px solid #4CAF50';
+            parent.style.boxShadow = 'none';
+        }
+    }
+    
+    // Initial setup
+    setupInputStyles();
     
-    // MutationObserver to handle dynamically added elements
+    // Setup observer to handle dynamically added elements
     const observer = new MutationObserver(function(mutations) {
-        applyFocusStyles();
+        setupInputStyles();
     });
     
-    observer.observe(document.body, { childList: true, subtree: true });
+    observer.observe(document.body, {
+        childList: true,
+        subtree: true
+    });
 });
 </script>
 
@@ -104,10 +111,12 @@ if not st.session_state.initialized:
         # Force model loading at startup to avoid session state issues
         load_model()
         
-        # Keep the message for 2 seconds before replacing it
+        # First set initialized to True
+        st.session_state.initialized = True
+        
+        # Then keep the message for 2 seconds before replacing it
         time.sleep(2)
         init_message.success("System initialized successfully!")
-        st.session_state.initialized = True
     except Exception as e:
         st.error(f"Error initializing: {str(e)}")