Update modules/database/semantic_mongo_db.py

modules/database/semantic_mongo_db.py

@@ -14,35 +14,52 @@ def store_student_semantic_result(username, text, analysis_result):
     Args:
         username: Nombre del usuario
         text: Texto analizado
-        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
+        analysis_result: Resultado del análisis
     """
     try:
+        # Convertir gráfico a formato base64
+        concept_graph_data = None
+        if 'concept_graph' in analysis_result:
+            buf = io.BytesIO()
+            analysis_result['concept_graph'].savefig(buf, format='png')
+            buf.seek(0)
+            concept_graph_data = base64.b64encode(buf.getvalue()).decode('utf-8')
+
+        entity_graph_data = None
+        if 'entity_graph' in analysis_result:
+            buf = io.BytesIO()
+            analysis_result['entity_graph'].savefig(buf, format='png')
+            buf.seek(0)
+            entity_graph_data = base64.b64encode(buf.getvalue()).decode('utf-8')
+
+        # Crear documento para MongoDB
         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']
+            'key_concepts': analysis_result.get('key_concepts', []),
+            'concept_graph': concept_graph_data,
+            'entities': analysis_result.get('entities', {}),
+            'entity_graph': entity_graph_data
         }
-        
+
+        # Insertar en MongoDB
         result = insert_document(COLLECTION_NAME, analysis_document)
-        
         if result:
             logger.info(f"Análisis semántico guardado con ID: {result} para el usuario: {username}")
             return True
-        return False
         
+        logger.error("No se pudo insertar el documento en MongoDB")
+        return False
+
     except Exception as e:
         logger.error(f"Error al guardar el análisis semántico: {str(e)}")
         return False
 
+
+
+
 def get_student_semantic_analysis(username, limit=10):
     """
     Recupera los análisis semánticos de un estudiante.