Update modules/database/semantic_mongo_db.py

modules/database/semantic_mongo_db.py

@@ -1,12 +1,24 @@
 #/modules/database/semantic_mongo_db.py
-import matplotlib.pyplot as plt  # Añadir esta importación al inicio
+
+# Importaciones estándar
 import io
 import base64
-from .mongo_db import insert_document, find_documents, update_document, delete_document
 from datetime import datetime, timezone
 import logging
 
-logger = logging.getLogger(__name__)
+# Importaciones de terceros
+import matplotlib.pyplot as plt
+
+# Importaciones locales
+from .mongo_db import (
+    insert_document, 
+    find_documents, 
+    update_document, 
+    delete_document
+)
+
+# Configuración del logger
+logger = logging.getLogger(__name__)  # Cambiado de name a __name__
 COLLECTION_NAME = 'student_semantic_analysis'
 
 def store_student_semantic_result(username, text, analysis_result):
@@ -16,22 +28,33 @@ def store_student_semantic_result(username, text, analysis_result):
         username: Nombre del usuario
         text: Texto analizado
         analysis_result: Resultado del análisis
+    Returns:
+        bool: True si se guardó correctamente, False en caso contrario
     """
     try:
-        # Convertir gráfico a formato base64
+        # Convertir gráfico conceptual a formato base64
         concept_graph_data = None
-        if 'concept_graph' in analysis_result:
+        if 'concept_graph' in analysis_result and analysis_result['concept_graph'] is not None:
             buf = io.BytesIO()
-            analysis_result['concept_graph'].savefig(buf, format='png')
-            buf.seek(0)
-            concept_graph_data = base64.b64encode(buf.getvalue()).decode('utf-8')
+            try:
+                analysis_result['concept_graph'].savefig(buf, format='png', dpi=300, bbox_inches='tight')
+                buf.seek(0)
+                concept_graph_data = base64.b64encode(buf.getvalue()).decode('utf-8')
+                plt.close(analysis_result['concept_graph'])  # Cerrar la figura
+            except Exception as e:
+                logger.error(f"Error al convertir gráfico conceptual: {str(e)}")
 
+        # Convertir gráfico de entidades a formato base64
         entity_graph_data = None
-        if 'entity_graph' in analysis_result:
+        if 'entity_graph' in analysis_result and analysis_result['entity_graph'] is not None:
             buf = io.BytesIO()
-            analysis_result['entity_graph'].savefig(buf, format='png')
-            buf.seek(0)
-            entity_graph_data = base64.b64encode(buf.getvalue()).decode('utf-8')
+            try:
+                analysis_result['entity_graph'].savefig(buf, format='png', dpi=300, bbox_inches='tight')
+                buf.seek(0)
+                entity_graph_data = base64.b64encode(buf.getvalue()).decode('utf-8')
+                plt.close(analysis_result['entity_graph'])  # Cerrar la figura
+            except Exception as e:
+                logger.error(f"Error al convertir gráfico de entidades: {str(e)}")
 
         # Crear documento para MongoDB
         analysis_document = {
@@ -57,9 +80,9 @@ def store_student_semantic_result(username, text, analysis_result):
     except Exception as e:
         logger.error(f"Error al guardar el análisis semántico: {str(e)}")
         return False
-
-
-
+    finally:
+        # Asegurarnos de cerrar cualquier figura pendiente
+        plt.close('all')
 
 def get_student_semantic_analysis(username, limit=10):
     """