Update app.py

app.py

@@ -27,7 +27,10 @@ def save_session(username, password):
     st.session_state.username = username
     st.session_state.scratch_pad = st.session_state.scratch_pad
     encrypted_password = fernet.encrypt(password.encode()).decode()
-    user_history[username] = encrypted_password
+    user_history[username] = {
+        'password': encrypted_password,
+        'scratch_pad': st.session_state.scratch_pad
+    }
     with open(history_file, 'w') as file:
         json.dump(user_history, file)
 
@@ -56,8 +59,7 @@ def login_user(username, password):
     else:
         st.warning("Username not found. Please register first.")
     return False
-
-
+    
 def reset_password(username):
     if username in user_history:
         new_password = Fernet.generate_password(8).decode()