Update modules/database/discourse_mongo_db.py

modules/database/discourse_mongo_db.py

@@ -8,39 +8,40 @@ import logging
 logger = logging.getLogger(__name__)
 COLLECTION_NAME = 'student_discourse_analysis'
 
-def store_student_discourse_result(username, text, analysis_result):
+def store_student_discourse_result(username, text1, text2, analysis_result):
     """
-    Guarda el resultado del análisis de discurso en MongoDB.
+    Guarda el resultado del análisis de discurso comparativo en MongoDB.
     Args:
         username: Nombre del usuario
-        text: Texto analizado
+        text1: Primer texto analizado (patrón)
+        text2: Segundo texto analizado (comparación)
         analysis_result: Resultado del análisis que incluye:
-            - key_concepts: Lista de conceptos clave y sus frecuencias
-            - concept_graph: Gráfico de relaciones conceptuales en bytes
-            - entities: Diccionario de entidades identificadas
-            - entity_graph: Gráfico de relaciones entre entidades en bytes
+            - key_concepts1: Lista de conceptos clave y frecuencias del primer texto
+            - key_concepts2: Lista de conceptos clave y frecuencias del segundo texto
+            - graph1: Gráfico del primer texto
+            - graph2: Gráfico del segundo texto
     """
     try:
         analysis_document = {
             'username': username,
             'timestamp': datetime.now(timezone.utc).isoformat(),
-            'text': text,
-            'analysis_type': 'semantic',
-            'key_concepts': analysis_result['key_concepts'],
-            'concept_graph': analysis_result['concept_graph'],
-            'entities': analysis_result['entities'],
-            'entity_graph': analysis_result['entity_graph']
+            'text1': text1,
+            'text2': text2,
+            'analysis_type': 'discourse',
+            'key_concepts1': analysis_result['key_concepts1'],
+            'key_concepts2': analysis_result['key_concepts2'],
+            'graph1': analysis_result['graph1'],
+            'graph2': analysis_result['graph2']
         }
         
         result = insert_document(COLLECTION_NAME, analysis_document)
-        
         if result:
-            logger.info(f"Análisis semántico guardado con ID: {result} para el usuario: {username}")
+            logger.info(f"Análisis del discurso guardado con ID: {result} para el usuario: {username}")
             return True
         return False
         
     except Exception as e:
-        logger.error(f"Error al guardar el análisis semántico: {str(e)}")
+        logger.error(f"Error al guardar el análisis del discurso: {str(e)}")
         return False
 
 def get_student_discourse_analysis(username, limit=10):