Update modules/__init__.py

modules/__init__.py

@@ -1,17 +1,61 @@
 # 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
-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
+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  # 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
+)
+from .discourse_analysis import (  # Nuevo módulo
+    perform_discourse_analysis,
+    compare_semantic_analysis
+)
 from .spacy_utils import load_spacy_models
 
 __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',
+    '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',  # Nueva función
     'display_student_progress',
-    'get_repeated_words_colors', 'highlight_repeated_words', 'POS_COLORS', 'POS_TRANSLATIONS',
-    'visualize_semantic_relations', 'perform_semantic_analysis', 'load_spacy_models'
+    'get_repeated_words_colors',
+    'highlight_repeated_words',
+    'POS_COLORS',
+    'POS_TRANSLATIONS',
+    'visualize_semantic_relations',
+    'perform_semantic_analysis',
+    'perform_discourse_analysis',  # Nueva función
+    'compare_semantic_analysis',  # Nueva función
+    'load_spacy_models'
 ]