Update modules/discourse/discourse_interface.py

modules/discourse/discourse_interface.py

@@ -12,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
@@ -129,9 +130,10 @@ 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"""
+    """
+    Muestra los resultados del análisis del discurso
+    """
     if not result.get('success'):
         st.warning(discourse_t.get('no_results', 'No hay resultados disponibles'))
         return
@@ -186,7 +188,16 @@ def display_discourse_results(result, lang_code, discourse_t):
         st.subheader(discourse_t.get('doc1_title', 'Documento 1'))
         st.markdown(discourse_t.get('key_concepts', 'Conceptos Clave'))
         if 'key_concepts1' in result:
-            # [conceptos_html se mantiene igual]...
+            concepts_html = f"""
+                <div class="concepts-container">
+                    {''.join([
+                        f'<div class="concept-item"><span class="concept-name">{concept}</span>'
+                        f'<span class="concept-freq">({freq:.2f})</span></div>'
+                        for concept, freq in result['key_concepts1']
+                    ])}
+                </div>
+            """
+            st.markdown(concepts_html, unsafe_allow_html=True)
 
             if 'graph1' in result:
                 st.markdown('<div class="graph-container">', unsafe_allow_html=True)
@@ -200,6 +211,33 @@ def display_discourse_results(result, lang_code, discourse_t):
                     st.image(result['graph1'])
                 else:
                     st.warning("Formato de gráfico no reconocido")
+                
+                # Botones y controles
+                button_col1, spacer_col1 = st.columns([1,4])
+                with button_col1:
+                    if 'graph1_bytes' in result:
+                        st.download_button(
+                            label="📥 " + discourse_t.get('download_graph', "Download"),
+                            data=result['graph1_bytes'],
+                            file_name="discourse_graph1.png",
+                            mime="image/png",
+                            use_container_width=True
+                        )
+
+                # Interpretación como texto normal sin expander
+                st.markdown("**📊 Interpretación del grafo:**")
+                st.markdown("""
+                    - 🔀 Las flechas indican la dirección de la relación entre conceptos
+                    - 🎨 Los colores más intensos indican conceptos más centrales en el texto
+                    - ⭕ El tamaño de los nodos representa la frecuencia del concepto
+                    - ↔️ El grosor de las líneas indica la fuerza de la conexión
+                """)
+                
+                st.markdown('</div>', unsafe_allow_html=True)
+            else:
+                st.warning(discourse_t.get('graph_not_available', 'Gráfico no disponible'))
+        else:
+            st.warning(discourse_t.get('concepts_not_available', 'Conceptos no disponibles'))
 
     # Documento 2
     with col2:
@@ -219,7 +257,16 @@ def display_discourse_results(result, lang_code, discourse_t):
 
             if 'graph2' in result:
                 st.markdown('<div class="graph-container">', unsafe_allow_html=True)
-                st.pyplot(result['graph2'])
+                
+                # 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
+                    st.image(result['graph2'])
+                else:
+                    st.warning("Formato de gráfico no reconocido")
                 
                 # Botones y controles
                 button_col2, spacer_col2 = st.columns([1,4])
@@ -250,4 +297,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.'))