Update modules/database/discourse_mongo_db.py

modules/database/discourse_mongo_db.py

@@ -19,34 +19,49 @@ 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:
-        # Los gráficos ya vienen en bytes, solo necesitamos codificar a base64
+        # 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:
             try:
-                graph1_data = base64.b64encode(analysis_result['graph1']).decode('utf-8')
+                # Convertir figura a bytes
+                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
             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:
-                graph2_data = base64.b64encode(analysis_result['graph2']).decode('utf-8')
+                # Convertir figura a bytes
+                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
             except Exception as e:
                 logger.error(f"Error al codificar gráfico 2: {str(e)}")
 
-        if 'combined_graph' in analysis_result and analysis_result['combined_graph'] is not None:
-            try:
-                combined_graph_data = base64.b64encode(analysis_result['combined_graph']).decode('utf-8')
-            except Exception as e:
-                logger.error(f"Error al codificar gráfico combinado: {str(e)}")
-
         # Crear documento para MongoDB
         analysis_document = {
             'username': username,
@@ -58,7 +73,8 @@ def store_student_discourse_result(username, text1, text2, analysis_result):
             'key_concepts2': analysis_result.get('key_concepts2', []),
             'graph1': graph1_data,
             'graph2': graph2_data,
-            'combined_graph': combined_graph_data
+            'graph1_bytes': analysis_result.get('graph1_bytes'),
+            'graph2_bytes': analysis_result.get('graph2_bytes')
         }
 
         # Insertar en MongoDB
@@ -120,10 +136,6 @@ 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):
     """