Update app.py

app.py

@@ -1,7 +1,7 @@
 import streamlit as st
 import json
 import os
-from cryptography.fernet import Fernet
+from cryptography.fernet import Fernet, InvalidToken
 
 # Initialize session state
 if 'username' not in st.session_state:
@@ -41,18 +41,22 @@ def register_user(username, password):
 
 def login_user(username, password):
     if username in user_history:
-        encrypted_password = user_history[username]
+        encrypted_password = user_history[username]['password']
         try:
             decrypted_password = fernet.decrypt(encrypted_password.encode()).decode()
             if password == decrypted_password:
-                save_session(username, password)
+                st.session_state.username = username
+                st.session_state.scratch_pad = user_history[username]['scratch_pad']
                 st.success("Logged in successfully!")
+                return True
             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.")
+    return False
+
 
 def reset_password(username):
     if username in user_history:
@@ -62,7 +66,7 @@ def reset_password(username):
     else:
         st.warning("Username not found.")
 
-st.title("Welcome to My App")
+st.title("Memory of Me - State Aware Authentication and User Data Memory")
 
 if st.session_state.username:
     st.subheader(f"Welcome back, {st.session_state.username}!")
@@ -77,7 +81,8 @@ else:
     col1, col2 = st.columns(2)
     with col1:
         if st.button("🔑 Login"):
-            login_user(username, password)
+            if login_user(username, password):
+                st.experimental_rerun()
     with col2:
         if st.button("📝 Register"):
             register_user(username, password)