Update modules/database/database.py

modules/database/database.py

@@ -222,11 +222,15 @@ def store_morphosyntax_result(username, text, repeated_words, arc_diagrams, pos_
         return False
 
 ################################################################################################################
-def store_semantic_result(username, text, network_diagram):
+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 a una imagen base64
+        # Convertir la figura de relaciones semánticas a una imagen base64
         buf = io.BytesIO()
-        network_diagram.savefig(buf, format='png')
+        analysis_result['relations_graph'].savefig(buf, format='png')
         buf.seek(0)
         img_str = base64.b64encode(buf.getvalue()).decode('utf-8')
 
@@ -234,12 +238,13 @@ def store_semantic_result(username, text, network_diagram):
             'username': username,
             'timestamp': datetime.utcnow(),
             'text': text,
-            'network_diagram': img_str,  # Guardar la imagen como string base64
+            'entities': analysis_result['entities'],
+            'key_concepts': analysis_result['key_concepts'],
+            'relations_graph': img_str,  # Guardar la imagen como string base64
             '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}")
         return True
     except Exception as e: