Update modules/database/discourse_mongo_db.py

modules/database/discourse_mongo_db.py

@@ -19,46 +19,33 @@ logger = logging.getLogger(__name__)
 
 COLLECTION_NAME = 'student_discourse_analysis'
 
-def fig_to_bytes(fig):
-    """Convierte una figura de matplotlib a bytes."""
-    try:
-        buf = io.BytesIO()
-        fig.savefig(buf, format='png', dpi=300, bbox_inches='tight')
-        buf.seek(0)
-        return buf.getvalue()
-    except Exception as e:
-        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:
-        # Primero convertir las figuras a bytes
-        if 'graph1' in analysis_result:
+        # Los gráficos ya vienen en bytes, solo necesitamos codificar a base64
+        graph1_data = None
+        graph2_data = None
+        combined_graph_data = None
+
+        if 'graph1' in analysis_result and analysis_result['graph1'] is not None:
+            try:
+                graph1_data = base64.b64encode(analysis_result['graph1']).decode('utf-8')
+            except Exception as e:
+                logger.error(f"Error al codificar gráfico 1: {str(e)}")
+
+        if 'graph2' in analysis_result and analysis_result['graph2'] is not None:
             try:
-                # Convertir y guardar bytes del grafo 1
-                graph1_bytes = fig_to_bytes(analysis_result['graph1'])
-                if graph1_bytes:
-                    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
+                graph2_data = base64.b64encode(analysis_result['graph2']).decode('utf-8')
             except Exception as e:
-                logger.error(f"Error al procesar gráfico 1: {str(e)}")
+                logger.error(f"Error al codificar gráfico 2: {str(e)}")
 
-        if 'graph2' in analysis_result:
+        if 'combined_graph' in analysis_result and analysis_result['combined_graph'] is not None:
             try:
-                # Convertir y guardar bytes del grafo 2
-                graph2_bytes = fig_to_bytes(analysis_result['graph2'])
-                if graph2_bytes:
-                    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
+                combined_graph_data = base64.b64encode(analysis_result['combined_graph']).decode('utf-8')
             except Exception as e:
-                logger.error(f"Error al procesar gráfico 2: {str(e)}")
+                logger.error(f"Error al codificar gráfico combinado: {str(e)}")
 
         # Crear documento para MongoDB
         analysis_document = {
@@ -69,23 +56,26 @@ 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_base64': analysis_result.get('graph1_base64'),
-            'graph2_base64': analysis_result.get('graph2_base64'),
+            'graph1': graph1_data,
+            'graph2': graph2_data,
+            'combined_graph': combined_graph_data
         }
 
         # Insertar en MongoDB
         result = insert_document(COLLECTION_NAME, analysis_document)
-        if not result:
-            logger.error("No se pudo insertar el documento en MongoDB")
-            return False
+        if result:
+            logger.info(f"Análisis del discurso guardado con ID: {result} para el usuario: {username}")
+            return True
 
-        logger.info(f"Análisis del discurso guardado con ID: {result} para el usuario: {username}")
-        return True
+        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 del discurso: {str(e)}")
         return False
 
+
+
 #################################################################################
 def get_student_discourse_analysis(username, limit=10):
     """
@@ -130,6 +120,10 @@ def get_student_discourse_analysis(username, limit=10):
         logger.error(f"Error recuperando análisis del discurso: {str(e)}")
         return []
 #####################################################################################
+
+
+
+
         
 def get_student_discourse_data(username):
     """