Update app.py

app.py

@@ -22,58 +22,45 @@ st.markdown("""
     border-radius: 8px !important;
 }
 /* Set default green border for all input fields */
-div[data-baseweb="input"] {
+.stTextInput>div>div>input {
     border: 2px solid #4CAF50 !important;
     border-radius: 8px !important;
+    box-shadow: none !important;
 }
 </style>
 
 <script>
+// Run once DOM is fully loaded
 document.addEventListener('DOMContentLoaded', function() {
-    function setupInputStyles() {
+    // Function to add events to input elements
+    function addInputEvents() {
+        // Find all input elements in text inputs
         const inputs = document.querySelectorAll('.stTextInput input');
         
         inputs.forEach(function(input) {
-            // Remove any existing listeners to prevent duplicates
-            input.removeEventListener('focus', onFocus);
-            input.removeEventListener('blur', onBlur);
+            // Focus event - red border
+            input.addEventListener('focus', function() {
+                this.style.border = '2px solid #FF5722 !important';
+                this.style.boxShadow = 'none !important';
+            });
             
-            // Get the parent which holds the visible border
-            const parent = input.closest('div[data-baseweb="input"]');
-            if (parent) {
-                // Force the initial green style
-                parent.style.border = '2px solid #4CAF50';
-                parent.style.borderRadius = '8px';
-                
-                // Add new listeners
-                input.addEventListener('focus', onFocus);
-                input.addEventListener('blur', onBlur);
-            }
+            // Blur event - green border
+            input.addEventListener('blur', function() {
+                this.style.border = '2px solid #4CAF50 !important';
+                this.style.boxShadow = 'none !important';
+            });
         });
     }
     
-    function onFocus(event) {
-        const parent = event.target.closest('div[data-baseweb="input"]');
-        if (parent) {
-            parent.style.border = '2px solid #FF5722';
-        }
-    }
-    
-    function onBlur(event) {
-        const parent = event.target.closest('div[data-baseweb="input"]');
-        if (parent) {
-            parent.style.border = '2px solid #4CAF50';
-        }
-    }
+    // Call immediately
+    addInputEvents();
     
-    // Initial setup
-    setupInputStyles();
-    
-    // Setup observer to handle dynamically added elements
-    const observer = new MutationObserver(function(mutations) {
-        setupInputStyles();
+    // Watch for changes to the DOM (when Streamlit adds new elements)
+    const observer = new MutationObserver(function() {
+        addInputEvents();
     });
     
+    // Start observing
     observer.observe(document.body, {
         childList: true,
         subtree: true
@@ -91,19 +78,10 @@ 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_time' not in st.session_state:
-    st.session_state.init_time = None
-if 'success_shown' not in st.session_state:
-    st.session_state.success_shown = False
-
-# Setup all authentication
-if not st.session_state.initialized:
-    try:
-        setup_all_auth()
-    except Exception as e:
-        st.error(f"Authentication error: {str(e)}")
+if 'init_start_time' not in st.session_state:
+    st.session_state.init_start_time = None
 
-# Create a placeholder for initialization messages
+# Create placeholder for initialization message
 init_message = st.empty()
 
 # Handle initialization
@@ -112,24 +90,28 @@ if not st.session_state.initialized:
         # Show the loading message
         init_message.info("Hang in there! We are setting the system up for you. 😊")
         
-        # Load the model
+        # 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
         load_model()
         
-        # Mark as initialized and record the time
+        # Mark as initialized
         st.session_state.initialized = True
-        st.session_state.init_time = time.time()
         
-        # Show success message (will be removed later)
+        # Show success message
         init_message.success("System initialized successfully!")
-        st.session_state.success_shown = True
     except Exception as e:
         st.error(f"Error initializing: {str(e)}")
-else:
-    # Check if we need to clear the success message
-    if st.session_state.success_shown and st.session_state.init_time is not None:
-        if time.time() - st.session_state.init_time > 2:
-            init_message.empty()
-            st.session_state.success_shown = False
+        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
 
 # Function to process when enter is pressed or button is clicked
 def process_input():