import streamlit as st
from oauth.user_manager import UserManager
from sollai.assistentes import Assistente

def main():
    st.title("🔐 Login")
    
    with st.form("login_form"):
        email = st.text_input("Email")
        password = st.text_input("Senha", type="password")
        
        if st.form_submit_button("Entrar"):
            try:
                user_manager = UserManager()
                user = user_manager.authenticate_user(email, password)
                
                if user:
                    # Guardar dados do usuário
                    st.session_state['user'] = user
                    st.session_state['is_authenticated'] = True
                    
                    # Carregar assistente usando o nome
                    assistant_name = user.get('assistant_name')
                    if assistant_name:
                        try:
                            assistente = Assistente(assistant_name)
                            st.session_state['assistant'] = assistente
                            st.success("✅ Login realizado com sucesso!")
                            st.rerun()
                        except Exception as e:
                            st.error(f"❌ Erro ao carregar assistente: {str(e)}")
                    else:
                        st.error("❌ Você não possui um assistente configurado.")
                else:
                    st.error("❌ Email ou senha inválidos!")
                    
            except Exception as e:
                st.error(f"❌ Erro ao fazer login: {str(e)}")