Update modules/database/database.py

modules/database/database.py

@@ -7,6 +7,7 @@ from pymongo import MongoClient
 import certifi
 from datetime import datetime
 import io
+from io import BytesIO
 import base64
 import matplotlib.pyplot as plt
 from matplotlib.figure import Figure
@@ -294,16 +295,31 @@ def store_discourse_analysis_result(username, text1, text2, analysis_result):
     
     try:
         # Convertir los grafos individuales a imágenes base64
-        buf1 = io.BytesIO()
+        buf1 = BytesIO()
         analysis_result['graph1'].savefig(buf1, format='png')
         buf1.seek(0)
         img_str1 = base64.b64encode(buf1.getvalue()).decode('utf-8')
 
-        buf2 = io.BytesIO()
+        buf2 = BytesIO()
         analysis_result['graph2'].savefig(buf2, format='png')
         buf2.seek(0)
         img_str2 = base64.b64encode(buf2.getvalue()).decode('utf-8')
 
+        # Crear una imagen combinada
+        fig, (ax1, ax2) = plt.subplots(1, 2, figsize=(20, 10))
+        ax1.imshow(plt.imread(BytesIO(base64.b64decode(img_str1))))
+        ax1.axis('off')
+        ax1.set_title("Documento 1: Relaciones Conceptuales")
+        ax2.imshow(plt.imread(BytesIO(base64.b64decode(img_str2))))
+        ax2.axis('off')
+        ax2.set_title("Documento 2: Relaciones Conceptuales")
+        
+        buf_combined = BytesIO()
+        fig.savefig(buf_combined, format='png')
+        buf_combined.seek(0)
+        img_str_combined = base64.b64encode(buf_combined.getvalue()).decode('utf-8')
+        plt.close(fig)
+
         # Crear el documento para guardar
         analysis_document = {
             'username': username,
@@ -312,6 +328,7 @@ def store_discourse_analysis_result(username, text1, text2, analysis_result):
             'text2': text2,
             'graph1': img_str1,
             'graph2': img_str2,
+            'combined_graph': img_str_combined,
             'analysis_type': 'discourse'
         }