Update modules/__init__.py

modules/__init__.py

@@ -1,85 +1,134 @@
 # modules/__init__.py
 
-from .auth import authenticate_user, register_user
-from .database import (
-    initialize_mongodb_connection,
-    get_student_data,
-    store_application_request,
-    store_morphosyntax_result,
-    store_semantic_result,
-    store_discourse_analysis_result,
-    store_chat_history,
-    create_admin_user,
-    create_student_user
-)
-from .ui import (
-    main,
-    login_register_page,
-    login_form,
-    register_form,
-    user_page,
-    display_student_progress,
-    display_morphosyntax_analysis_interface,
-    display_semantic_analysis_interface,
-    display_discourse_analysis_interface,
-    display_chatbot_interface
-)
-from .admin_ui import admin_page  # Añade esta línea
-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 (
-    initialize_chatbot,
-    get_chatbot_response,
-    ClaudeAPIChat  # Nueva clase
-)
+def load_auth_functions():
+    from modules.auth.auth import authenticate_user, register_user
+    return {
+        'authenticate_user': authenticate_user,
+        'register_user': register_user
+    }
 
-__all__ = [
-    'authenticate_user',
-    'register_user',
-    'initialize_mongodb_connection',
-    'get_student_data',
-    'store_morphosyntax_result',
-    'store_semantic_result',
-    'store_discourse_analysis_result',
-    'store_chat_history',
-    'create_admin_user',
-    'create_student_user',
-    'main',
-    'login_register_page',
-    'login_form',
-    'register_form',
-    'user_page',
-    'admin_page',  # Añade esta línea
-    'display_morphosyntax_analysis_interface',
-    'display_semantic_analysis_interface',
-    'display_discourse_analysis_interface',
-    'display_chatbot_interface',
-    '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',
-    'get_chatbot_response',
-    'ClaudeAPIChat'
-]
+def load_database_function():
+    from modules.database.database import (
+        initialize_mongodb_connection,
+        get_student_data,
+        store_application_request,
+        store_morphosyntax_result,
+        store_semantic_result,
+        store_discourse_analysis_result,
+        store_chat_history,
+        create_admin_user,
+        create_student_user
+    )
+    return {
+        'initialize_mongodb_connection': initialize_mongodb_connection, 
+        'get_student_data': get_student_data, 
+        'store_application_request': store_application_request, 
+        'store_morphosyntax_result': store_morphosyntax_result,
+        'store_semantic_result': store_semantic_result, 
+        'store_discourse_analysis_result': store_discourse_analysis_result, 
+        'store_chat_history': store_chat_history, 
+        'create_admin_user': create_admin_user, 
+        'create_student_user': create_student_user
+    }
+
+def load_ui_functions():
+    from modules.ui.ui import (
+        main,
+        login_register_page,
+        login_form,
+        register_form,
+        user_page,
+        display_student_progress,
+        display_morphosyntax_analysis_interface,
+        display_semantic_analysis_interface,
+        display_discourse_analysis_interface,
+        display_chatbot_interface
+    )
+    return {
+        'main': main, 
+        'login_register_page': login_register_page,
+        'login_form': login_form,
+        'register_form': register_form,
+        'user_page': user_page,
+        'display_student_progress': display_student_progress,
+        'display_morphosyntax_analysis_interface': display_morphosyntax_analysis_interface,
+        'display_semantic_analysis_interface': display_semantic_analysis_interface,
+        'display_discourse_analysis_interface': display_discourse_analysis_interface,
+        'display_chatbot_interface': display_chatbot_interface   
+    }
+
+def load_admin_functions():
+    from modules.admin.admin_ui import admin_page
+    return {
+        'admin_page': admin_page
+    }
+    
+def morpho_analysis_functions():
+    from modules.analysis_text.morpho_analysis import (
+        get_repeated_words_colors,
+        highlight_repeated_words,
+        POS_COLORS,
+        POS_TRANSLATIONS,
+        perform_advance_morphosyntax_analysis
+    )
+    return {
+        'get_repeated_words_colors': get_repeated_words_colors,
+        'highlight_repeated_words': highlight_repeated_words,
+        'POS_COLORS': POS_COLORS,
+        'POS_TRANSLATIONS': POS_TRANSLATIONS,
+        'perform_advance_morphosyntax_analysis' : perform_advance_morphosyntax_analysis
+    }
+
+def semantic_analysis_text_functions():
+    from modules.analysis_text.semantic_analysis import (
+        visualize_semantic_relations,
+        perform_semantic_analysis,
+        create_semantic_graph
+    )
+    return {
+        'visualize_semantic_relations': visualize_semantic_relations,
+        'perform_semantic_analysis': perform_semantic_analysis,
+        'create_semantic_graph': create_semantic_graph
+    }
+
+def discourse_analysis_text_functions():
+    from modules.analysis_text.discourse_analysis import (
+        perform_discourse_analysis,
+        compare_semantic_analysis
+    )
+    return {
+        'perform_discourse_analysis': perform_discourse_analysis,
+        'compare_semantic_analysis': compare_semantic_analysis
+    }
+
+def spacy_utils_functions():
+    from modules.utils.spacy_utils import load_spacy_models
+    return {
+        'load_spacy_models': load_spacy_models
+    }
+
+def chatbot_functions():
+    from modules.chatbot.chatbot import (
+        initialize_chatbot,
+        get_chatbot_response,
+        ClaudeAPIChat
+    )
+    return {
+        'initialize_chatbot': initialize_chatbot,
+        'get_chatbot_response': get_chatbot_response,
+        'ClaudeAPIChat': ClaudeAPIChat
+    }
+
+# Opcional: función para cargar todas las funciones
+def load_all_functions():
+    return {
+        **load_auth_functions(),
+        **load_database_function(),
+        **load_ui_functions(),
+        **load_admin_functions(),
+        **morpho_analysis_functions(),
+        **semantic_analysis_text_functions(),
+        **discourse_analysis_text_functions(),
+        **spacy_utils_functions(),
+        **chatbot_functions()
+    }