Update modules/__init__.py

modules/__init__.py

@@ -15,7 +15,8 @@ from .ui import (
     display_student_progress, 
     display_morphosyntax_analysis_interface, 
     display_semantic_analysis_interface,
-    display_discourse_analysis_interface  # Nueva función
+    display_discourse_analysis_interface,
+    display_chatbot_interface  # Nueva función
 )
 from .morpho_analysis import (
     get_repeated_words_colors, 
@@ -25,13 +26,19 @@ from .morpho_analysis import (
 )
 from .semantic_analysis import (
     visualize_semantic_relations, 
-    perform_semantic_analysis
+    perform_semantic_analysis,
+    create_semantic_graph
 )
-from .discourse_analysis import (  # Nuevo módulo
+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',
@@ -47,7 +54,8 @@ __all__ = [
     'display_chat_interface',
     'display_morphosyntax_analysis_interface',
     'display_semantic_analysis_interface',
-    'display_discourse_analysis_interface',  # Nueva función
+    'display_discourse_analysis_interface',
+    'display_chatbot_interface',  # Nueva función
     'display_student_progress',
     'get_repeated_words_colors',
     'highlight_repeated_words',
@@ -55,7 +63,12 @@ __all__ = [
     'POS_TRANSLATIONS',
     'visualize_semantic_relations',
     'perform_semantic_analysis',
-    'perform_discourse_analysis',  # Nueva función
-    'compare_semantic_analysis',  # Nueva función
-    'load_spacy_models'
-]
+    '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
+]
+