Clear warning messages after leaving login dialog

src/streamlit_app.py

@@ -11,6 +11,12 @@ chat_parameters = {
 }
 
 
+def switch_page():
+    if "message" in st.session_state:
+        st.session_state.message.empty()
+    st.rerun()
+
+
 def get_chat_parameters():
     with st.sidebar:
         chat_parameters["system_prompt"] = st.text_area(
@@ -59,16 +65,17 @@ def login():
     env_token = os.getenv("HF_TOKEN")
     if env_token:
         st.session_state.hf_token = env_token
-        st.rerun()
+        switch_page()
     else:
         st.title("🔐 Please input Hugging Face Token")
         user_token = st.text_input("HF_TOKEN", type="password")
         if st.button("✅ Login"):
             if user_token:
                 st.session_state.hf_token = user_token
-                st.rerun()
+                switch_page()
             else:
-                st.warning("Please enter a valid token.")
+                st.session_state.message = st.warning(
+                    "Please enter a valid token.")
 
 
 def chatbot():