# modules/__init__.py

from .auth import authenticate_user, register_user, get_user_role
from .database import (
    initialize_mongodb_connection, 
    get_student_data, 
    store_morphosyntax_result, 
    store_semantic_result
)
from .ui import (
    login_register_page, 
    login_form, 
    register_form, 
    display_chat_interface, 
    display_student_progress, 
    display_morphosyntax_analysis_interface, 
    display_semantic_analysis_interface,
    display_discourse_analysis_interface,
    display_chatbot_interface  # Nueva función
)
from .morpho_analysis import (
    get_repeated_words_colors, 
    highlight_repeated_words, 
    POS_COLORS, 
    POS_TRANSLATIONS
)
from .semantic_analysis import (
    visualize_semantic_relations, 
    perform_semantic_analysis,
    create_semantic_graph
)
from .discourse_analysis import (
    perform_discourse_analysis,
    compare_semantic_analysis
)
from .spacy_utils import load_spacy_models
from .chatbot import (  # Nuevo módulo
    initialize_chatbot,
    get_chatbot_response,
    Llama2Chatbot
)

__all__ = [
    'authenticate_user',
    'register_user',
    'get_user_role',
    'initialize_mongodb_connection',
    'get_student_data',
    'store_morphosyntax_result',
    'store_semantic_result',
    'login_register_page',
    'login_form',
    'register_form',
    'display_chat_interface',
    'display_morphosyntax_analysis_interface',
    'display_semantic_analysis_interface',
    'display_discourse_analysis_interface',
    'display_chatbot_interface',  # Nueva función
    'display_student_progress',
    'get_repeated_words_colors',
    'highlight_repeated_words',
    'POS_COLORS',
    'POS_TRANSLATIONS',
    'visualize_semantic_relations',
    'perform_semantic_analysis',
    'create_semantic_graph',
    'perform_discourse_analysis',
    'compare_semantic_analysis',
    'load_spacy_models',
    'initialize_chatbot',  # Nueva función
    'get_chatbot_response',  # Nueva función
    'Llama2Chatbot'  # Nueva clase
]