Update modules/text_analysis/discourse_analysis.py

modules/text_analysis/discourse_analysis.py

@@ -21,8 +21,7 @@ logger = logging.getLogger(__name__)
 from .semantic_analysis import (
     create_concept_graph,
     visualize_concept_graph,
-    identify_key_concepts,
-    fig_to_bytes
+    identify_key_concepts
 )
 
 
@@ -91,7 +90,16 @@ ENTITY_LABELS = {
 }
 
 #################
-
+def fig_to_bytes(fig):
+    """Convierte una figura de matplotlib a bytes."""
+    try:
+        buf = io.BytesIO()
+        fig.savefig(buf, format='png', dpi=300, bbox_inches='tight')
+        buf.seek(0)
+        return buf.getvalue()
+    except Exception as e:
+        logger.error(f"Error en fig_to_bytes: {str(e)}")
+        return None
 #################
 def compare_semantic_analysis(text1, text2, nlp, lang):
     """