Update app.py

app.py

@@ -69,16 +69,18 @@ if not st.session_state.initialized:
         st.session_state.initialized = True
         st.session_state.init_time = time.time()
         init_message.success("System initialized successfully!")
+        time.sleep(2)
+        init_message.empty()
     except Exception as e:
         init_message.error(f"Error initializing: {str(e)}")
 
 # Handle timing of success message disappearance
 # This will run on each rerender after initialization
-if st.session_state.initialized and st.session_state.init_time:
-    current_time = time.time()
-    if (current_time - st.session_state.init_time) >= 2.0:
+elif st.session_state.init_time is not None:
+    elapsed_time = time.time() - st.session_state.init_time
+    if elapsed_time >= 2.0:
         init_message.empty()
-        st.session_state.init_time = None  # Reset the time after clearing
+        st.session_state.init_time = None
 
 # Function to handle form submission
 def handle_form_submit():