Update app.py

app.py

@@ -38,14 +38,18 @@ def register_user(username, password):
     else:
         st.warning("Username already exists. Please choose a different username.")
 
+
 def login_user(username, password):
     if username in user_history:
         encrypted_password = user_history[username]
-        decrypted_password = fernet.decrypt(encrypted_password.encode()).decode()
-        if password == decrypted_password:
-            save_session(username, password)
-            st.success("Logged in successfully!")
-        else:
+        try:
+            decrypted_password = fernet.decrypt(encrypted_password.encode()).decode()
+            if password == decrypted_password:
+                save_session(username, password)
+                st.success("Logged in successfully!")
+            else:
+                st.error("Incorrect password. Please try again.")
+        except InvalidToken:
             st.error("Invalid password. Please try again.")
     else:
         st.warning("Username not found. Please register first.")