import streamlit as st
import os
import json

def save_user_data(email, data):
    """Save user data to a file named after the user's email."""
    with open(f"{email}.txt", "w") as file:
        file.write(json.dumps(data))

def load_user_data(email):
    """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 containing '@' in filename."""
    return [f[:-4] for f in os.listdir() if f.endswith('.txt') and '@' in f]

def main():
    st.title('User Data Management')

    # 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=['New User'] + saved_users)

    # 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': ''}

    # 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'])
    new_password = st.text_input("🔑 Password", value=cached_data['password'], type='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.sidebar.success("Data updated and saved!")

    # Show email link with GET parameter
    if new_email:
        st.write(f"Link for this user: https://huggingface.co/spaces/awacke1/Streamlit-Memory-Cache-Resource-and-Data/?email={new_email}`")
        st.session_state['last_email'] = new_email

    # Display current data without password
    display_data = cached_data.copy()
    display_data['password'] = '*****'  # Hide the password
    st.write("Current Data:")
    st.json(display_data)

    # 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}")

# Run the app
if __name__ == "__main__":
    main()