import streamlit as st

# Tes infos utilisateur
user_infos = {
    "username": "user123",
    "mdps": "pass123"
}
    
def login_page(): 

    col1, col2 , col3 = st.columns([5,2,2])
    with col1 : 
        st.image("logo Dataltist.png" , width=150) 
    
    with col2 : 
        st.write("")
    
    with col3 : 
        st.link_button("🌐 Notre Site","https://dataltist.fr/")

    

    st.title("Connexion au Chatbot")
    username = st.text_input("Nom d'utilisateur")
    password = st.text_input("Mot de passe", type="password")

    if st.button("Se Connecter"): 
        if username == user_infos["username"] and password == user_infos["mdps"]:
            st.session_state["is_logged_in"] = True  # On utilise la même clé
            st.success("Connexion réussie !")
            st.rerun()
        else : 
            st.error("Identifiants incorrects")

# --- LOGIQUE DE VÉRIFICATION ---

# 1. Initialisation si la clé n'existe pas encore
if "is_logged_in" not in st.session_state: 
    st.session_state["is_logged_in"] = False 

# 2. Affichage conditionnel
if not st.session_state["is_logged_in"]: 
    login_page()
else: 
    # --- ICI VOTRE CODE CHATBOT ---
    st.title("🤖 Mon Super Chatbot")
    
    if st.button("Se déconnecter"):
        st.session_state["is_logged_in"] = False
        st.rerun()
    
    prompt = st.chat_input("Posez votre question...")
    if prompt:
        st.write(f"Vous avez dit : {prompt}")