Update modules/__init__.py

modules/__init__.py

@@ -47,7 +47,15 @@ def load_database_functions():
         delete_student_semantic_analysis,
         get_student_semantic_data
     )
-    
+
+    from .database.discourse_mongo_db import (
+        store_student_discourse_result',
+        get_student_discourse_analysis',
+        update_student_discourse_analysis',
+        delete_student_discourse_analysis',
+        get_student_discourse_data'
+    )
+
     from .database.chat_db import store_chat_history, get_chat_history
 
     return {
@@ -74,7 +82,12 @@ def load_database_functions():
         'delete_student_semantic_analysis': delete_student_semantic_analysis,
         'get_student_semantic_data': get_student_semantic_data,
         'store_chat_history': store_chat_history,
-        'get_chat_history': get_chat_history
+        'get_chat_history': get_chat_history, 
+        'store_student_discourse_result': store_student_discourse_result,
+        'get_student_discourse_analysis': get_student_discourse_analysis,
+        'update_student_discourse_analysis': update_student_discourse_analysis,
+        'delete_student_discourse_analysis'; delete_student_discourse_analysis,
+        'get_student_discourse_data': get_student_discourse_data
     }
 
 def load_ui_functions():
@@ -124,6 +137,23 @@ def load_semantic_functions():
     }
 
 
+def load_discourse_functions():
+    from modules.discourse.discourse_interface import (
+        display_discourse_interface, 
+        display_discourse_results
+    )
+    from modules.discourse.discourse_process import (
+        process_discourse_input, 
+        format_discourse_results
+    )
+
+    return {
+        'display_discourse_interface': display_semantic_interface,
+        'display_discourse_results': display_semantic_results,
+        'process_discourse_input': process_semantic_input,
+        'format_discourse_results': format_analysis_results,
+    }
+
 def load_admin_functions():
     from .admin.admin_ui import admin_page
     return {
@@ -152,6 +182,7 @@ def load_all_functions():
         **load_admin_functions(),
         **load_morphosyntax_functions(),
         **load_semantic_functions(),
+        **load_discourse_functions(),
         **load_utils_functions(),
         **load_chatbot_functions(),
         **load_student_activities_functions()