Update app.py

app.py

@@ -22,13 +22,25 @@ import string
 topic = st.sidebar.text_input("Enter the Wikipedia topic:", "Japanese History")
 #query = st.sidebar.text_input("Enter your first query:", "First query")
 model_name = 'mistralai/Mistral-7B-Instruct-v0.3'
-HF_TOKEN = st.sidebar.text_input("Enter your Hugging Face token:", "YOUR_HF_TOKEN", type="password")
-
-# Hugging Face login
-if HF_TOKEN and HF_TOKEN != "YOUR_HF_TOKEN":
-    login(token=HF_TOKEN)
-else:
-    st.sidebar.warning("Please enter a valid Hugging Face token to proceed.")
+HF_TOKEN = st.sidebar.text_input("Enter your Hugging Face token:", "", type="password")
+
+# Function to validate token
+def validate_token(token):
+    try:
+        # Attempt to log in with the provided token
+        login(token=token)
+        # Check if the token is valid by trying to access some data
+        HfApi().whoami()
+        return True
+    except Exception as e:
+        return False
+
+# Validate the token and display appropriate message
+if HF_TOKEN:
+    if validate_token(HF_TOKEN):
+        st.sidebar.success("Token is valid!")
+    else:
+        st.sidebar.error("Invalid token. Please try again.")
 
 # Memory for chat history
 if "history" not in st.session_state: