Update modules/database.py

modules/database.py

@@ -247,17 +247,51 @@ def store_semantic_result(username, text, network_diagram):
         return False
 
 ###############################################################################################################
-def store_discourse_analysis_result(username, text, graph1, graph2):
+import io
+import base64
+import matplotlib.pyplot as plt
+from matplotlib.figure import Figure
+
+def store_discourse_analysis_result(username, text1, text2, graph1, graph2):
     try:
+        # Crear una nueva figura combinada
+        fig, (ax1, ax2) = plt.subplots(1, 2, figsize=(20, 10))
+
+        # Añadir la primera imagen con título
+        ax1.imshow(graph1.get_figure().canvas.renderer.buffer_rgba())
+        ax1.set_title("Documento Patrón: Relaciones semánticas relevantes")
+        ax1.axis('off')
+
+        # Añadir la segunda imagen con título
+        ax2.imshow(graph2.get_figure().canvas.renderer.buffer_rgba())
+        ax2.set_title("Documento Comparado con el documento patrón: Relaciones semánticas relevantes")
+        ax2.axis('off')
+
+        # Ajustar el diseño
+        plt.tight_layout()
+
+        # Convertir la figura combinada a una imagen base64
+        buf = io.BytesIO()
+        fig.savefig(buf, format='png')
+        buf.seek(0)
+        img_str = base64.b64encode(buf.getvalue()).decode('utf-8')
+
+        # Cerrar las figuras para liberar memoria
+        plt.close(fig)
+        plt.close(graph1.get_figure())
+        plt.close(graph2.get_figure())
+
         analysis_document = {
             'username': username,
             'timestamp': datetime.utcnow(),
-            'text': text,
-            'graph1': graph1,
-            'graph2': graph2,
+            'text1': text1,
+            'text2': text2,
+            'combined_graph': img_str,
             'analysis_type': 'discourse'
         }
+
         result = analysis_collection.insert_one(analysis_document)
+
         logger.info(f"Análisis discursivo guardado con ID: {result.inserted_id} para el usuario: {username}")
         return True
     except Exception as e: