Update app.py

app.py

@@ -24,6 +24,10 @@ topic = st.sidebar.text_input("Enter the Wikipedia topic:", "Japanese History")
 model_name = 'mistralai/Mistral-7B-Instruct-v0.3'
 HF_TOKEN = st.sidebar.text_input("Enter your Hugging Face token:", "", type="password")
 
+# Initialize session state for error message
+if 'error_message' not in st.session_state:
+    st.session_state.error_message = ""
+
 # Function to validate token
 def validate_token(token):
     try:
@@ -38,9 +42,13 @@ def validate_token(token):
 # Validate the token and display appropriate message
 if HF_TOKEN:
     if validate_token(HF_TOKEN):
+        st.session_state.error_message = ""  # Clear error message if the token is valid
         st.sidebar.success("Token is valid!")
     else:
-        st.sidebar.error("Invalid token. Please try again.")
+        st.session_state.error_message = "Invalid token. Please try again."
+        st.sidebar.error(st.session_state.error_message)
+elif st.session_state.error_message:
+    st.sidebar.error(st.session_state.error_message)
 
 # Memory for chat history
 if "history" not in st.session_state: