import streamlit as st
import os
import json
import base64

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': '', 'social': {}}
    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 get_file_download_link(filename):
    """Generate a file download link."""
    with open(filename, "rb") as file:
        base64_file = base64.b64encode(file.read()).decode()
    href = f'<a href="data:file/txt;base64,{base64_file}" download="{filename}">Download {filename}</a>'
    return href

def display_social_links(social_data):
    """Display social media links as Markdown."""
    st.markdown("## Social Media Links")
    for platform, url in social_data.items():
        if url:
            emoji = {"instagram": "📷", "twitter": "🐦", "facebook": "📘", "huggingface": "🤗", "github": "💻", "linkedin": "🔗"}.get(platform, "")
            st.markdown(f"{emoji} [{platform.capitalize()}]({url})")

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': '', 'social': {}}

    # 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')

    # Social media fields
    st.markdown("### Social Media Profiles")
    social_fields = ["instagram", "twitter", "facebook", "huggingface", "github", "linkedin"]
    for field in social_fields:
        emoji = {"instagram": "📷", "twitter": "🐦", "facebook": "📘", "huggingface": "🤗", "github": "💻", "linkedin": "🔗"}.get(field, "")
        cached_data['social'][field] = st.text_input(f"{emoji} {field.capitalize()}", value=cached_data['social'].get(field, ''))

    # Save data when changes are made
    if st.button("Save Data"):
        save_user_data(new_email, {'email': new_email, 'phone': new_phone, 'password': new_password, 'social': cached_data['social']})
        st.sidebar.success("Data updated and saved!")

    # Download link for user data file
    if selected_user and selected_user != 'New User':
        st.sidebar.markdown(get_file_download_link(f"{selected_user}.txt"), unsafe_allow_html=True)

    # File uploader
    uploaded_file = st.sidebar.file_uploader("📤 Upload User File", type='txt')
    if uploaded_file is not None:
        uploaded_data = json.loads(uploaded_file.getvalue().decode())
        save_user_data(uploaded_data['email'], uploaded_data)
        st.sidebar.success("Uploaded and saved!")

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

    # Display social media links
    display_social_links(cached_data['social'])

    # 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()