#app.py
import streamlit as st
from modules.database import initialize_mongodb_connection
from modules.auth import authenticate_user, get_user_role, register_user
from modules.ui import login_register_page, display_student_progress, display_text_analysis_interface
from modules.spacy_utils import load_spacy_models
import time

st.set_page_config(page_title="AIdeaText", layout="wide", page_icon="random")

def main():
    if not initialize_mongodb_connection():
        st.warning("La conexión a la base de datos MongoDB no está disponible. Algunas funciones pueden no estar operativas.")

    if 'logged_in' not in st.session_state:
        st.session_state.logged_in = False

    if not st.session_state.logged_in:
        login_register_page()
    else:
        logged_in_interface()

def logged_in_interface():
    nlp_models = load_spacy_models()
    languages = {'Español': 'es', 'English': 'en', 'Français': 'fr'}
    selected_lang = st.sidebar.selectbox("Select Language / Seleccione el idioma / Choisissez la langue", list(languages.keys()))
    lang_code = languages[selected_lang]

    st.sidebar.title(f"Bienvenido, {st.session_state.username}")
    if st.sidebar.button("Cerrar Sesión"):
        st.session_state.logged_in = False
        st.experimental_rerun()

    tab1, tab2, tab3, tab4 = st.tabs(["Análisis morfosintáctico", "Análisis semántico", "Análisis semántico discursivo", "Mi Progreso"])

    with tab1:
        display_text_analysis_interface(nlp_models, lang_code)

    with tab2:
        st.header("Análisis semántico")
        st.write("Esta función aún no está implementada.")

    with tab3:
        st.header("Análisis semántico discursivo")
        st.write("Esta función aún no está implementada.")

    with tab4:
        display_student_progress(st.session_state.username, lang_code)

if __name__ == "__main__":
    main()