Update modules/database/semantic_mongo_db.py

modules/database/semantic_mongo_db.py

@@ -25,37 +25,16 @@ COLLECTION_NAME = 'student_semantic_analysis'
 def store_student_semantic_result(username, text, analysis_result):
     """
     Guarda el resultado del análisis semántico en MongoDB.
-    Args:
-        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 conceptual a formato base64
+        # El gráfico ya viene en bytes, solo necesitamos codificarlo a base64
         concept_graph_data = None
         if 'concept_graph' in analysis_result and analysis_result['concept_graph'] is not None:
-            buf = io.BytesIO()
-            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 and analysis_result['entity_graph'] is not None:
-            buf = io.BytesIO()
             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
+                # Ya está en bytes, solo codificar a base64
+                concept_graph_data = base64.b64encode(analysis_result['concept_graph']).decode('utf-8')
             except Exception as e:
-                logger.error(f"Error al convertir gráfico de entidades: {str(e)}")
+                logger.error(f"Error al codificar gráfico conceptual: {str(e)}")
 
         # Crear documento para MongoDB
         analysis_document = {
@@ -64,9 +43,7 @@ def store_student_semantic_result(username, text, analysis_result):
             'text': text,
             'analysis_type': 'semantic',
             'key_concepts': analysis_result.get('key_concepts', []),
-            'concept_graph': concept_graph_data,
-            'entities': analysis_result.get('entities', {}),
-            'entity_graph': entity_graph_data
+            'concept_graph': concept_graph_data
         }
 
         # Insertar en MongoDB
@@ -81,9 +58,6 @@ 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):