Update modules/discourse/discourse_interface.py

modules/discourse/discourse_interface.py

@@ -2,6 +2,7 @@
 
 import streamlit as st
 import pandas as pd
+import matplotlib.pyplot as plt
 import plotly.graph_objects as go
 import logging
 from ..utils.widget_utils import generate_unique_key
@@ -11,6 +12,7 @@ from ..database.discourse_mongo_db import store_student_discourse_result
 
 logger = logging.getLogger(__name__)
 
+#############################################################################################
 def display_discourse_interface(lang_code, nlp_models, discourse_t):
     """
     Interfaz para el análisis del discurso
@@ -29,7 +31,7 @@ def display_discourse_interface(lang_code, nlp_models, discourse_t):
             }
 
         # 2. Título y descripción
-        #st.subheader(discourse_t.get('discourse_title', 'Análisis del Discurso'))
+        # st.subheader(discourse_t.get('discourse_title', 'Análisis del Discurso'))
         st.info(discourse_t.get('initial_instruction', 
             'Cargue dos archivos de texto para realizar un análisis comparativo del discurso.'))
 
@@ -128,7 +130,6 @@ def display_discourse_interface(lang_code, nlp_models, discourse_t):
 
 
 #####################################################################################################################
-
 def display_discourse_results(result, lang_code, discourse_t):
     """
     Muestra los resultados del análisis del discurso
@@ -198,9 +199,27 @@ def display_discourse_results(result, lang_code, discourse_t):
             """
             st.markdown(concepts_html, unsafe_allow_html=True)
 
+            # Verificar el tipo de graph1 de manera más robusta
             if 'graph1' in result:
                 st.markdown('<div class="graph-container">', unsafe_allow_html=True)
-                st.pyplot(result['graph1'])
+                
+                # Más información para depuración
+                graph_type = type(result['graph1']).__name__
+                graph_size = len(result['graph1']) if isinstance(result['graph1'], bytes) else "N/A"
+                logger.info(f"Tipo de graph1: {graph_type}, Tamaño: {graph_size}")
+                
+                if isinstance(result['graph1'], bytes) and len(result['graph1']) > 0:
+                    # Es bytes válidos
+                    st.image(result['graph1'])
+                elif isinstance(result['graph1'], plt.Figure):
+                    # Es una figura de matplotlib
+                    st.pyplot(result['graph1'])
+                elif result['graph1'] is None:
+                    # Es None
+                    st.warning("Gráfico no disponible")
+                else:
+                    # Otro tipo o bytes vacíos
+                    st.warning(f"Formato de gráfico no reconocido: {graph_type}")
                 
                 # Botones y controles
                 button_col1, spacer_col1 = st.columns([1,4])
@@ -245,9 +264,27 @@ def display_discourse_results(result, lang_code, discourse_t):
             """
             st.markdown(concepts_html, unsafe_allow_html=True)
 
-            if 'graph2' in result:
+            # Verificar el tipo de graph1 de manera más robusta
+            if 'graph1' in result:
                 st.markdown('<div class="graph-container">', unsafe_allow_html=True)
-                st.pyplot(result['graph2'])
+                
+                # Más información para depuración
+                graph_type = type(result['graph2']).__name__
+                graph_size = len(result['graph2']) if isinstance(result['graph2'], bytes) else "N/A"
+                logger.info(f"Tipo de graph2: {graph_type}, Tamaño: {graph_size}")
+                
+                if isinstance(result['graph2'], bytes) and len(result['graph2']) > 0:
+                    # Es bytes válidos
+                    st.image(result['graph2'])
+                elif isinstance(result['graph2'], plt.Figure):
+                    # Es una figura de matplotlib
+                    st.pyplot(result['graph2'])
+                elif result['graph2'] is None:
+                    # Es None
+                    st.warning("Gráfico no disponible")
+                else:
+                    # Otro tipo o bytes vacíos
+                    st.warning(f"Formato de gráfico no reconocido: {graph_type}")
                 
                 # Botones y controles
                 button_col2, spacer_col2 = st.columns([1,4])
@@ -278,4 +315,4 @@ def display_discourse_results(result, lang_code, discourse_t):
 
     # Nota informativa sobre la comparación
     st.info(discourse_t.get('comparison_note', 
-        'La funcionalidad de comparación detallada estará disponible en una próxima actualización.'))
+        'La funcionalidad de comparación detallada estará disponible en una próxima actualización.'))