Update modules/text_analysis/discourse_analysis.py

modules/text_analysis/discourse_analysis.py

@@ -91,13 +91,11 @@ ENTITY_LABELS = {
 
 #################
 
-def fig_to_bytes(fig):
+def fig_to_bytes(fig, dpi=100):
     """Convierte una figura de matplotlib a bytes."""
     try:
         buf = io.BytesIO()
-        # Reducir DPI y usar compression
-        fig.savefig(buf, format='png', dpi=100, bbox_inches='tight', 
-                   compression=9)  # Máxima compresión
+        fig.savefig(buf, format='png', dpi=dpi, bbox_inches='tight')  # Sin compression
         buf.seek(0)
         return buf.getvalue()
     except Exception as e:
@@ -220,6 +218,12 @@ def perform_discourse_analysis(text1, text2, nlp, lang):
         logger.info(f"Figura 1 convertida a {len(graph1_bytes) if graph1_bytes else 0} bytes")
         logger.info(f"Figura 2 convertida a {len(graph2_bytes) if graph2_bytes else 0} bytes")
 
+        # Verificar que las conversiones fueron exitosas antes de continuar
+        if not graph1_bytes or not graph2_bytes:
+            logger.error("Error al convertir figuras a bytes - obteniendo 0 bytes")
+            # Opción 1: Devolver error
+            raise ValueError("No se pudieron convertir las figuras a bytes")
+
         # Crear tablas de resultados
         table1 = create_concept_table(key_concepts1)
         table2 = create_concept_table(key_concepts2)