import streamlit as st
from oauth.login import main as login_main
from oauth.cadastro import main as cadastro_main
from sollai.chat import main as chat_main

# Configuração inicial do Streamlit
st.set_page_config(
    page_title="Sollai Assistant",
    page_icon="🤖",
    layout="wide"
)

# Inicializar variáveis de sessão
if 'is_authenticated' not in st.session_state:
    st.session_state['is_authenticated'] = False

if 'user' not in st.session_state:
    st.session_state['user'] = None

if 'assistant' not in st.session_state:
    st.session_state['assistant'] = None

# Função principal
def main():
    if not st.session_state['is_authenticated']:
        login_main()
    else:
        user = st.session_state['user']
        assistant = st.session_state.get('assistant')
        
        if not assistant:
            st.warning("⚠️ Você não possui um assistente configurado.")
            if st.button("Sair"):
                for key in st.session_state.keys():
                    del st.session_state[key]
                st.rerun()
        else:
            chat_main()

# Execução principal
if __name__ == "__main__":
    main()