Update modules/__init__.py

modules/__init__.py

@@ -1,18 +1,12 @@
 # modules/__init__.py
 
-#from ..modules.auth.auth import authenticate_user, register_user
 def load_auth_functions():
-    from modules.auth.auth import(
-        authenticate_user,
-        register_user
-    )
+    from modules.auth.auth import authenticate_user, register_user
     return {
-        'main' : main,
         'authenticate_user': authenticate_user,
         'register_user': register_user
     }
 
-
 def load_database_function():
     from modules.database.database import (
         initialize_mongodb_connection,
@@ -26,19 +20,17 @@ def load_database_function():
         create_student_user
     )
     return {
-        'main' : main,
-        '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_admin_user
+        '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,
@@ -53,24 +45,22 @@ def load_ui_functions():
         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   
+        '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  # Añade esta línea
-    )
+    from modules.admin.admin_ui import admin_page
     return {
-        'admin_page' : admin_page
+        'admin_page': admin_page
     }
     
 def morpho_analysis_functions():
@@ -81,10 +71,10 @@ def morpho_analysis_functions():
         POS_TRANSLATIONS
     )
     return {
-        'get_repeated_words_colors' : get_repeated_words_colors,
-        'highlight_repeated_words'  : highlight_repeated_words,
-        'highlight_repeated_words'  : POS_COLORS,
-        'highlight_repeated_words'  : POS_TRANSLATIONS
+        'get_repeated_words_colors': get_repeated_words_colors,
+        'highlight_repeated_words': highlight_repeated_words,
+        'POS_COLORS': POS_COLORS,
+        'POS_TRANSLATIONS': POS_TRANSLATIONS
     }
 
 def semantic_analysis_text_functions():
@@ -94,38 +84,49 @@ def semantic_analysis_text_functions():
         create_semantic_graph
     )
     return {
-        'visualize_semantic_relations' : visualize_semantic_relations,
-        'perform_semantic_analysis' : perform_semantic_analysis,
-        'create_semantic_graph' : create_semantic_graph
+        '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_anaysis' : compare_semantic_analysis
+        '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
-    )
+    from modules.utils.spacy_utils import load_spacy_models
     return {
-        'load_spacy_models' : load_spacy_models
+        'load_spacy_models': load_spacy_models
     }
 
 def chatbot_functions():
     from modules.chatbot.chatbot import (
         initialize_chatbot,
         get_chatbot_response,
-        ClaudeAPIChat  # Nueva clase
+        ClaudeAPIChat
     )
     return {
-        'initialize_chatbot' : initialize_chatbot,
-        'get_chatbot_response' : get_chatbot_response,
-        'ClaudeAPIChat' : ClaudeAPIChat  # Nueva clase
+        '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()
     }