Update modules/database/discourse_mongo_db.py

modules/database/discourse_mongo_db.py

@@ -30,37 +30,35 @@ def fig_to_bytes(fig):
         logger.error(f"Error en fig_to_bytes: {str(e)}")
         return None
 
+
+#################################################################################
 def store_student_discourse_result(username, text1, text2, analysis_result):
     """
     Guarda el resultado del análisis de discurso comparativo en MongoDB.
     """
     try:
-        # Convertir figuras a bytes antes de codificar
-        graph1_data = None
-        graph2_data = None
-        combined_graph_data = None
-
-        if 'graph1' in analysis_result and analysis_result['graph1'] is not None:
+        # Primero convertir las figuras a bytes
+        if 'graph1' in analysis_result:
             try:
-                # Convertir figura a bytes
+                # Convertir y guardar bytes del grafo 1
                 graph1_bytes = fig_to_bytes(analysis_result['graph1'])
                 if graph1_bytes:
-                    graph1_data = base64.b64encode(graph1_bytes).decode('utf-8')
-                    # Guardar los bytes también para el botón de descarga
                     analysis_result['graph1_bytes'] = graph1_bytes
+                    analysis_result['graph1_base64'] = base64.b64encode(graph1_bytes).decode('utf-8')
+                plt.close(analysis_result['graph1'])  # Cerrar la figura
             except Exception as e:
-                logger.error(f"Error al codificar gráfico 1: {str(e)}")
+                logger.error(f"Error al procesar gráfico 1: {str(e)}")
 
-        if 'graph2' in analysis_result and analysis_result['graph2'] is not None:
+        if 'graph2' in analysis_result:
             try:
-                # Convertir figura a bytes
+                # Convertir y guardar bytes del grafo 2
                 graph2_bytes = fig_to_bytes(analysis_result['graph2'])
                 if graph2_bytes:
-                    graph2_data = base64.b64encode(graph2_bytes).decode('utf-8')
-                    # Guardar los bytes también para el botón de descarga
                     analysis_result['graph2_bytes'] = graph2_bytes
+                    analysis_result['graph2_base64'] = base64.b64encode(graph2_bytes).decode('utf-8')
+                plt.close(analysis_result['graph2'])  # Cerrar la figura
             except Exception as e:
-                logger.error(f"Error al codificar gráfico 2: {str(e)}")
+                logger.error(f"Error al procesar gráfico 2: {str(e)}")
 
         # Crear documento para MongoDB
         analysis_document = {
@@ -71,27 +69,23 @@ def store_student_discourse_result(username, text1, text2, analysis_result):
             'analysis_type': 'discourse',
             'key_concepts1': analysis_result.get('key_concepts1', []),
             'key_concepts2': analysis_result.get('key_concepts2', []),
-            'graph1': graph1_data,
-            'graph2': graph2_data,
-            'graph1_bytes': analysis_result.get('graph1_bytes'),
-            'graph2_bytes': analysis_result.get('graph2_bytes')
+            'graph1_base64': analysis_result.get('graph1_base64'),
+            'graph2_base64': analysis_result.get('graph2_base64'),
         }
 
         # Insertar en MongoDB
         result = insert_document(COLLECTION_NAME, analysis_document)
-        if result:
-            logger.info(f"Análisis del discurso guardado con ID: {result} para el usuario: {username}")
-            return True
+        if not result:
+            logger.error("No se pudo insertar el documento en MongoDB")
+            return False
 
-        logger.error("No se pudo insertar el documento en MongoDB")
-        return False
+        logger.info(f"Análisis del discurso guardado con ID: {result} para el usuario: {username}")
+        return True
 
     except Exception as e:
         logger.error(f"Error al guardar el análisis del discurso: {str(e)}")
         return False
 
-
-
 #################################################################################
 def get_student_discourse_analysis(username, limit=10):
     """