import os
import streamlit as st

# Hardcoded credentials
USERNAME = os.environ["USERNAME"]
PASSWORD = os.environ["PASSWORD"]
BASE_CONTENT_CODE_ASSIST_T2_MICRO = os.environ["BASE_CONTENT_CODE_ASSIST_T2_MICRO"]

# Initialize session state
if 'logged_in' not in st.session_state:
    st.session_state.logged_in = False

# Sidebar for login/logout with emojis
st.sidebar.title("🔒 AIXNet")

if st.session_state.logged_in:
    st.sidebar.success("🎉 You are logged in!")
    if st.sidebar.button("🔓 Log out"):
        st.session_state.logged_in = False
        st.sidebar.info("You have logged out.")
else:
    with st.sidebar.form(key='login_form'):
        username = st.text_input("👤 Username")
        password = st.text_input("🔑 Password", type="password")
        login_button = st.form_submit_button(label="🔓 Log in")

    if login_button:
        if username == USERNAME and password == PASSWORD:
            st.session_state.logged_in = True
            st.sidebar.success("🎉 Login successful!")
        else:
            st.sidebar.error("❌ Invalid username or password. Please try again.")

# Display table only if logged in
if st.session_state.logged_in:
    st.title("📋 AIXNet Tasks")
    data = {
        "📝 Task": ["💻 Code assist"],
        "🖥️ Instance Type": ["t2.micro"],
        "💰 Price": ["$0.05 / hour"],
        "🌐 IPv4": [BASE_CONTENT_CODE_ASSIST_T2_MICRO]
    }

    st.table(data)
else:
    st.info("👉 Please log in to view the tasks.")