Update app.py

app.py

@@ -21,12 +21,10 @@ st.markdown("""
     border: 2px solid #FF5722 !important;
     border-radius: 8px !important;
 }
-/* Override Streamlit's default input styling */
+/* Set default green border for all input fields */
 div[data-baseweb="input"] {
-    border-radius: 8px;
-}
-div[data-baseweb="input"] > div {
-    background-color: transparent;
+    border: 2px solid #4CAF50 !important;
+    border-radius: 8px !important;
 }
 </style>
 
@@ -40,16 +38,17 @@ document.addEventListener('DOMContentLoaded', function() {
             input.removeEventListener('focus', onFocus);
             input.removeEventListener('blur', onBlur);
             
-            // Set initial style (green)
+            // 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.boxShadow = 'none';
+                parent.style.borderRadius = '8px';
+                
+                // Add new listeners
+                input.addEventListener('focus', onFocus);
+                input.addEventListener('blur', onBlur);
             }
-            
-            // Add new listeners
-            input.addEventListener('focus', onFocus);
-            input.addEventListener('blur', onBlur);
         });
     }
     
@@ -57,7 +56,6 @@ document.addEventListener('DOMContentLoaded', function() {
         const parent = event.target.closest('div[data-baseweb="input"]');
         if (parent) {
             parent.style.border = '2px solid #FF5722';
-            parent.style.boxShadow = 'none';
         }
     }
     
@@ -65,7 +63,6 @@ document.addEventListener('DOMContentLoaded', function() {
         const parent = event.target.closest('div[data-baseweb="input"]');
         if (parent) {
             parent.style.border = '2px solid #4CAF50';
-            parent.style.boxShadow = 'none';
         }
     }
     
@@ -94,6 +91,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:
@@ -102,23 +103,33 @@ if not st.session_state.initialized:
     except Exception as e:
         st.error(f"Authentication error: {str(e)}")
 
-# Preload the model to avoid session state issues
+# Create a placeholder for initialization messages
+init_message = st.empty()
+
+# Handle initialization
 if not st.session_state.initialized:
     try:
-        init_message = st.empty()
+        # Show the loading message
         init_message.info("Hang in there! We are setting the system up for you. 😊")
         
-        # Force model loading at startup to avoid session state issues
+        # Load the model
         load_model()
         
-        # First set initialized to True
+        # Mark as initialized and record the time
         st.session_state.initialized = True
+        st.session_state.init_time = time.time()
         
-        # Then keep the message for 2 seconds before replacing it
-        time.sleep(2)
+        # Show success message (will be removed later)
         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
 
 # Function to process when enter is pressed or button is clicked
 def process_input():