Update modules/database/discourse_mongo_db.py

modules/database/discourse_mongo_db.py

@@ -22,54 +22,30 @@ COLLECTION_NAME = 'student_discourse_analysis'
 def store_student_discourse_result(username, text1, text2, analysis_result):
     """
     Guarda el resultado del análisis de discurso comparativo en MongoDB.
-    Args:
-        username: Nombre del usuario
-        text1: Primer texto analizado (patrón)
-        text2: Segundo texto analizado (comparación)
-        analysis_result: Resultado del análisis
     """
     try:
-        # Convertir gráficos individuales a base64
+        # Los gráficos ya vienen en bytes, solo necesitamos codificar a base64
         graph1_data = None
         graph2_data = None
         combined_graph_data = None
 
-        # Convertir primer gráfico
-        if 'graph1' in analysis_result:
-            buf = io.BytesIO()
-            analysis_result['graph1'].savefig(buf, format='png')
-            buf.seek(0)
-            graph1_data = base64.b64encode(buf.getvalue()).decode('utf-8')
-
-        # Convertir segundo gráfico
-        if 'graph2' in analysis_result:
-            buf = io.BytesIO()
-            analysis_result['graph2'].savefig(buf, format='png')
-            buf.seek(0)
-            graph2_data = base64.b64encode(buf.getvalue()).decode('utf-8')
-
-        # Crear gráfico combinado
-        if graph1_data and graph2_data:
-            fig, (ax1, ax2) = plt.subplots(1, 2, figsize=(20, 10))
-            
-            # Convertir base64 a imagen para el gráfico combinado
-            img1 = plt.imread(io.BytesIO(base64.b64decode(graph1_data)))
-            img2 = plt.imread(io.BytesIO(base64.b64decode(graph2_data)))
-            
-            ax1.imshow(img1)
-            ax1.axis('off')
-            ax1.set_title("Documento 1: Relaciones Conceptuales")
-            
-            ax2.imshow(img2)
-            ax2.axis('off')
-            ax2.set_title("Documento 2: Relaciones Conceptuales")
-            
-            # Guardar gráfico combinado en base64
-            buf = io.BytesIO()
-            fig.savefig(buf, format='png')
-            buf.seek(0)
-            combined_graph_data = base64.b64encode(buf.getvalue()).decode('utf-8')
-            plt.close(fig)
+        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:
+                graph2_data = base64.b64encode(analysis_result['graph2']).decode('utf-8')
+            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 = {