Update modules/database/discourse_mongo_db.py

modules/database/discourse_mongo_db.py

@@ -19,22 +19,48 @@ def store_student_discourse_result(username, text1, text2, analysis_result):
         analysis_result: Resultado del análisis
     """
     try:
-        # Convertir gráficos a formato base64 si existen
+        # Convertir gráficos individuales 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)
+
         # Crear documento para MongoDB
         analysis_document = {
             'username': username,
@@ -45,19 +71,19 @@ def store_student_discourse_result(username, text1, text2, analysis_result):
             'key_concepts1': analysis_result.get('key_concepts1', []),
             'key_concepts2': analysis_result.get('key_concepts2', []),
             'graph1': graph1_data,
-            'graph2': graph2_data
+            'graph2': graph2_data,
+            'combined_graph': combined_graph_data
         }
 
         # 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
-            
+
         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