Update modules/database/discourse_mongo_db.py

modules/database/discourse_mongo_db.py

@@ -1,11 +1,12 @@
-#/modules/database/discourse_mongo_db.py
-
-#/modules/database/discoruse_mongo_db.py
+# modules/database/discourse_mongo_db.py
 from .mongo_db import insert_document, find_documents, update_document, delete_document
 from datetime import datetime, timezone
 import logging
+import io
+import base64
 
 logger = logging.getLogger(__name__)
+
 COLLECTION_NAME = 'student_discourse_analysis'
 
 def store_student_discourse_result(username, text1, text2, analysis_result):
@@ -15,29 +16,46 @@ def store_student_discourse_result(username, text1, text2, analysis_result):
         username: Nombre del usuario
         text1: Primer texto analizado (patrón)
         text2: Segundo texto analizado (comparación)
-        analysis_result: Resultado del análisis que incluye:
-            - 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
+        analysis_result: Resultado del análisis
     """
     try:
+        # Convertir gráficos a formato base64 si existen
+        graph1_data = None
+        graph2_data = None
+        
+        if 'graph1' in analysis_result:
+            buf = io.BytesIO()
+            analysis_result['graph1'].savefig(buf, format='png')
+            buf.seek(0)
+            graph1_data = base64.b64encode(buf.getvalue()).decode('utf-8')
+            
+        if 'graph2' in analysis_result:
+            buf = io.BytesIO()
+            analysis_result['graph2'].savefig(buf, format='png')
+            buf.seek(0)
+            graph2_data = base64.b64encode(buf.getvalue()).decode('utf-8')
+
+        # Crear documento para MongoDB
         analysis_document = {
             'username': username,
             'timestamp': datetime.now(timezone.utc).isoformat(),
             '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']
+            'key_concepts1': analysis_result.get('key_concepts1', []),
+            'key_concepts2': analysis_result.get('key_concepts2', []),
+            'graph1': graph1_data,
+            'graph2': graph2_data
         }
-        
+
+        # Insertar en MongoDB
         result = insert_document(COLLECTION_NAME, analysis_document)
+        
         if result:
             logger.info(f"Análisis del discurso guardado con ID: {result} para el usuario: {username}")
             return True
+            
+        logger.error("No se pudo insertar el documento en MongoDB")
         return False
         
     except Exception as e:
@@ -46,66 +64,61 @@ def store_student_discourse_result(username, text1, text2, analysis_result):
 
 def get_student_discourse_analysis(username, limit=10):
     """
-    Recupera los análisis semánticos de un estudiante.
-    Args:
-        username: Nombre del usuario
-        limit: Número máximo de análisis a retornar
-    Returns:
-        list: Lista de análisis semánticos
+    Recupera los análisis del discurso de un estudiante.
     """
-    query = {"username": username, "analysis_type": "semantic"}
-    return find_documents(COLLECTION_NAME, query, sort=[("timestamp", -1)], limit=limit)
+    try:
+        query = {
+            "username": username,
+            "analysis_type": "discourse"
+        }
+        return find_documents(COLLECTION_NAME, query, sort=[("timestamp", -1)], limit=limit)
+    except Exception as e:
+        logger.error(f"Error al recuperar análisis del discurso: {str(e)}")
+        return []
 
-def update_student_discourse_analysis(analysis_id, update_data):
+def get_student_discourse_data(username):
     """
-    Actualiza un análisis semántico existente.
-    Args:
-        analysis_id: ID del análisis a actualizar
-        update_data: Datos a actualizar
+    Obtiene un resumen de los análisis del discurso de un estudiante.
     """
-    query = {"_id": analysis_id}
-    update = {"$set": update_data}
-    return update_document(COLLECTION_NAME, query, update)
+    try:
+        analyses = get_student_discourse_analysis(username, limit=None)
+        formatted_analyses = []
+        
+        for analysis in analyses:
+            formatted_analysis = {
+                'timestamp': analysis['timestamp'],
+                'text1': analysis.get('text1', ''),
+                'text2': analysis.get('text2', ''),
+                'key_concepts1': analysis.get('key_concepts1', []),
+                'key_concepts2': analysis.get('key_concepts2', [])
+            }
+            formatted_analyses.append(formatted_analysis)
+            
+        return {'entries': formatted_analyses}
+        
+    except Exception as e:
+        logger.error(f"Error al obtener datos del discurso: {str(e)}")
+        return {'entries': []}
 
-def delete_student_discourse_analysis(analysis_id):
+def update_student_discourse_analysis(analysis_id, update_data):
     """
-    Elimina un análisis semántico.
-    Args:
-        analysis_id: ID del análisis a eliminar
+    Actualiza un análisis del discurso existente.
     """
-    query = {"_id": analysis_id}
-    return delete_document(COLLECTION_NAME, query)
+    try:
+        query = {"_id": analysis_id}
+        update = {"$set": update_data}
+        return update_document(COLLECTION_NAME, query, update)
+    except Exception as e:
+        logger.error(f"Error al actualizar análisis del discurso: {str(e)}")
+        return False
 
-def get_student_discourse_data(username):
+def delete_student_discourse_analysis(analysis_id):
     """
-    Obtiene todos los análisis semánticos de un estudiante.
-    Args:
-        username: Nombre del usuario
-    Returns:
-        dict: Diccionario con todos los análisis del estudiante
+    Elimina un análisis del discurso.
     """
-    analyses = get_student_semantic_analysis(username, limit=None)
-    
-    formatted_analyses = []
-    for analysis in analyses:
-        formatted_analysis = {
-            'timestamp': analysis['timestamp'],
-            'text': analysis['text'],
-            'key_concepts': analysis['key_concepts'],
-            'entities': analysis['entities']
-            # No incluimos los gráficos en el resumen general
-        }
-        formatted_analyses.append(formatted_analysis)
-    
-    return {
-        'entries': formatted_analyses
-    }
-
-# Exportar las funciones necesarias
-__all__ = [
-    'store_student_discourse_result',
-    'get_student_discourse_analysis',
-    'update_student_discourse_analysis',
-    'delete_student_discourse_analysis',
-    'get_student_discourse_data'
-]
+    try:
+        query = {"_id": analysis_id}
+        return delete_document(COLLECTION_NAME, query)
+    except Exception as e:
+        logger.error(f"Error al eliminar análisis del discurso: {str(e)}")
+        return False