Update modules/discourse/discourse_interface.py

modules/discourse/discourse_interface.py

@@ -199,18 +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)
                 
-                # Verificar el tipo de graph1
-                if isinstance(result['graph1'], plt.Figure):
-                    # Es una figura de matplotlib directamente
-                    st.pyplot(result['graph1'])
-                elif isinstance(result['graph1'], bytes):
-                    # Es bytes, mostrar como imagen
+                # 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:
-                    st.warning("Formato de gráfico no reconocido")
+                    # 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])
@@ -255,18 +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)
                 
-                # Verificar el tipo de graph2
-                if isinstance(result['graph2'], plt.Figure):
-                    # Es una figura de matplotlib directamente
-                    st.pyplot(result['graph2'])
-                elif isinstance(result['graph2'], bytes):
-                    # Es bytes, mostrar como imagen
+                # 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:
-                    st.warning("Formato de gráfico no reconocido")
+                    # 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])