Update modules/discourse/discourse_live_interface.py

modules/discourse/discourse_live_interface.py

@@ -5,6 +5,8 @@ from streamlit_float import *
 from streamlit_antd_components import *
 import pandas as pd
 import logging
+import io
+import matplotlib.pyplot as plt
 
 # Configuración del logger
 logger = logging.getLogger(__name__)
@@ -15,6 +17,17 @@ from ..utils.widget_utils import generate_unique_key
 from ..database.discourse_mongo_db import store_student_discourse_result
 from ..database.chat_mongo_db import store_chat_history, get_chat_history
 
+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 display_discourse_live_interface(lang_code, nlp_models, discourse_t):
     """
     Interfaz para el análisis del discurso en vivo
@@ -92,6 +105,12 @@ def display_discourse_live_interface(lang_code, nlp_models, discourse_t):
                         )
 
                         if result['success']:
+                            # Convertir gráficos a bytes antes de guardar
+                            if 'graph1' in result:
+                                result['graph1_bytes'] = fig_to_bytes(result['graph1'])
+                            if 'graph2' in result:
+                                result['graph2_bytes'] = fig_to_bytes(result['graph2'])
+
                             st.session_state.discourse_live_state['last_result'] = result
                             st.session_state.discourse_live_state['analysis_count'] += 1
                             st.session_state.discourse_live_state['text_changed'] = False
@@ -110,6 +129,7 @@ def display_discourse_live_interface(lang_code, nlp_models, discourse_t):
                     logger.error(f"Error en análisis: {str(e)}")
                     st.error(discourse_t.get('error_processing', 'Error al procesar el texto'))
 
+                    
         # Columna derecha: Visualización de resultados
         with result_col:
             st.subheader(discourse_t.get('live_results', 'Resultados en vivo'))