Update modules/database/database.py

modules/database/database.py

@@ -226,31 +226,24 @@ def store_semantic_result(username, text, analysis_result):
     if analysis_collection is None:
         logger.error("La conexión a MongoDB no está inicializada")
         return False
-
     try:
-        # Convertir la figura de relaciones semánticas a una imagen base64
         buf = io.BytesIO()
         analysis_result['relations_graph'].savefig(buf, format='png')
         buf.seek(0)
         img_str = base64.b64encode(buf.getvalue()).decode('utf-8')
-
-        logger.info(f"Longitud de la imagen antes de guardar: {len(img_str)}") 
-        
         analysis_document = {
             'username': username,
             'timestamp': datetime.utcnow(),
             'text': text,
             'entities': analysis_result['entities'],
             'key_concepts': analysis_result['key_concepts'],
-            'relations_graph': img_str,  # Guardar la imagen como string base64
+            'network_diagram': img_str,  # Cambiado de 'relations_graph' a 'network_diagram'
             'analysis_type': 'semantic'
         }
-
         result = analysis_collection.insert_one(analysis_document)
         logger.info(f"Análisis semántico guardado con ID: {result.inserted_id} para el usuario: {username}")
-        logger.info(f"Claves en el documento guardado: {analysis_document.keys()}")
-        return True
         logger.info(f"Longitud de la imagen guardada: {len(img_str)}")
+        return True
     except Exception as e:
         logger.error(f"Error al guardar el análisis semántico para el usuario {username}: {str(e)}")
         return False