Update app.py

app.py

@@ -8,12 +8,11 @@ def save_user_data(email, data):
         file.write(json.dumps(data))
 
 def load_user_data(email):
-    """Load user data from a file named after the user's email."""
-    try:
-        with open(f"{email}.txt", "r") as file:
-            return json.loads(file.read())
-    except FileNotFoundError:
+    """Load user data from a file named after the user's email, or create a new file if it doesn't exist."""
+    if not os.path.isfile(f"{email}.txt"):
         return {'email': '', 'phone': '', 'password': ''}
+    with open(f"{email}.txt", "r") as file:
+        return json.loads(file.read())
 
 def list_saved_users():
     """List all files (users) with saved states."""
@@ -25,10 +24,10 @@ def main():
     # Sidebar for selecting a user file
     st.sidebar.title("Saved Users")
     saved_users = list_saved_users()
-    selected_user = st.sidebar.selectbox("Select a user", options=saved_users)
+    selected_user = st.sidebar.selectbox("Select a user", options=['New User'] + saved_users)
 
-    # Load data for the selected user
-    if selected_user:
+    # Load or initialize data
+    if selected_user and selected_user != 'New User':
         cached_data = load_user_data(selected_user)
     else:
         cached_data = {'email': '', 'phone': '', 'password': ''}
@@ -36,31 +35,30 @@ def main():
     # Input fields with emojis
     new_email = st.text_input("📧 Email Address", value=cached_data['email'])
     new_phone = st.text_input("📱 Mobile Phone", value=cached_data['phone'])
-
-    # Password field with an option to view contents
     show_password = st.checkbox("Show password")
     if show_password:
         new_password = st.text_input("🔑 Password", value=cached_data['password'])
     else:
         new_password = st.text_input("🔑 Password", value=cached_data['password'], type='password')
 
-    # Update and save data if changes are made
-    if new_email != cached_data['email'] or new_phone != cached_data['phone'] or new_password != cached_data['password']:
+    # Save data when changes are made
+    if new_email and (new_email != cached_data['email'] or new_phone != cached_data['phone'] or new_password != cached_data['password']):
         cached_data = {'email': new_email, 'phone': new_phone, 'password': new_password}
         save_user_data(new_email, cached_data)
-        st.success("Data updated and saved!")
+        st.sidebar.success("Data updated and saved!")
 
     st.write("Current Data:")
     st.json(cached_data)
 
-# Password Reset Simulation
-if st.sidebar.button("Reset Password"):
-    reset_email = st.sidebar.text_input("Enter email for password reset")
-    if reset_email in saved_users:
-        reset_data = load_user_data(reset_email)
-        reset_data['password'] = 'new_password'  # Reset password
-        save_user_data(reset_email, reset_data)
-        st.sidebar.success(f"Password reset for {reset_email}")
+    # Password Reset Simulation
+    if st.sidebar.button("Reset Password"):
+        reset_email = st.sidebar.text_input("Enter email for password reset")
+        if reset_email and reset_email in saved_users:
+            reset_data = load_user_data(reset_email)
+            reset_data['password'] = 'new_password'  # Reset password
+            save_user_data(reset_email, reset_data)
+            st.sidebar.success(f"Password reset for {reset_email}")
 
-if __name__ == "main":
-    main()
+# Run the app
+if __name__ == "__main__":
+    main()