Update app.py

app.py

@@ -72,11 +72,13 @@ if not st.session_state.initialized:
     except Exception as e:
         init_message.error(f"Error initializing: {str(e)}")
 
-elif st.session_state.init_time is not None:
-    elapsed_time = time.time() - st.session_state.init_time
-    if elapsed_time >= 2.0:
+# 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:
         init_message.empty()
-        st.session_state.init_time = None
+        st.session_state.init_time = None  # Reset the time after clearing
 
 # Function to handle form submission
 def handle_form_submit():