Update modules/discourse/discourse_interface.py

modules/discourse/discourse_interface.py

@@ -10,16 +10,16 @@ from ..database.chat_mongo_db import store_chat_history
 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
+    Args:
+        lang_code: Código del idioma actual
+        nlp_models: Modelos de spaCy cargados
+        discourse_t: Diccionario de traducciones
     """
     try:
-        # Activar estado
-        st.session_state.tab_states['discourse_active'] = True
-        
         # 1. Inicializar estado si no existe
         if 'discourse_state' not in st.session_state:
             st.session_state.discourse_state = {
@@ -67,9 +67,11 @@ def display_discourse_interface(lang_code, nlp_models, discourse_t):
         if analyze_button and uploaded_file1 and uploaded_file2:
             try:
                 with st.spinner(discourse_t.get('processing', 'Procesando análisis...')):
+                    # Leer contenido de archivos
                     text1 = uploaded_file1.getvalue().decode('utf-8')
                     text2 = uploaded_file2.getvalue().decode('utf-8')
 
+                    # Realizar análisis
                     result = perform_discourse_analysis(
                         text1, 
                         text2, 
@@ -78,6 +80,7 @@ def display_discourse_interface(lang_code, nlp_models, discourse_t):
                     )
 
                     if result['success']:
+                        # Guardar estado
                         st.session_state.discourse_result = result
                         st.session_state.discourse_state['analysis_count'] += 1
                         st.session_state.discourse_state['current_files'] = (
@@ -85,6 +88,7 @@ def display_discourse_interface(lang_code, nlp_models, discourse_t):
                             uploaded_file2.name
                         )
 
+                        # Guardar en base de datos
                         if store_student_discourse_result(
                             st.session_state.username,
                             text1,
@@ -92,6 +96,8 @@ def display_discourse_interface(lang_code, nlp_models, discourse_t):
                             result
                         ):
                             st.success(discourse_t.get('success_message', 'Análisis guardado correctamente'))
+                            
+                            # Mostrar resultados
                             display_discourse_results(result, lang_code, discourse_t)
                         else:
                             st.error(discourse_t.get('error_message', 'Error al guardar el análisis'))
@@ -99,7 +105,6 @@ def display_discourse_interface(lang_code, nlp_models, discourse_t):
                         st.error(discourse_t.get('analysis_error', 'Error en el análisis'))
 
             except Exception as e:
-                st.session_state.tab_states['discourse_active'] = False
                 logger.error(f"Error en análisis del discurso: {str(e)}")
                 st.error(discourse_t.get('error_processing', f'Error procesando archivos: {str(e)}'))
 
@@ -117,69 +122,17 @@ def display_discourse_interface(lang_code, nlp_models, discourse_t):
             )
 
     except Exception as e:
-        st.session_state.tab_states['discourse_active'] = False
-        logger.error(f"Error general en interfaz discursiva: {str(e)}")
-        st.error(discourse_t.get('general_error', "Se produjo un error. Por favor, intente de nuevo."))
-
-##########################################################################################
+        logger.error(f"Error general en interfaz del discurso: {str(e)}")
+        st.error(discourse_t.get('general_error', 'Se produjo un error. Por favor, intente de nuevo.'))
 
 def display_discourse_results(result, lang_code, discourse_t):
     """
-    Muestra los resultados del análisis del discurso con conceptos en formato horizontal
-    y botones de control consistentes
+    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
 
-    # Estilo CSS unificado
-    st.markdown("""
-        <style>
-        .concepts-container {
-            display: flex;
-            flex-wrap: nowrap;
-            gap: 8px;
-            padding: 12px;
-            background-color: #f8f9fa;
-            border-radius: 8px 8px 0 0;
-            overflow-x: auto;
-            margin-bottom: 0;
-            white-space: nowrap;
-        }
-        .concept-item {
-            background-color: white;
-            border-radius: 4px;
-            padding: 6px 10px;
-            display: inline-flex;
-            align-items: center;
-            gap: 4px;
-            box-shadow: 0 1px 2px rgba(0,0,0,0.1);
-            flex-shrink: 0;
-        }
-        .concept-name {
-            font-weight: 500;
-            color: #1f2937;
-            font-size: 0.85em;
-        }
-        .concept-freq {
-            color: #6b7280;
-            font-size: 0.75em;
-        }
-        .graph-container {
-            background-color: white;
-            border-radius: 0 0 8px 8px;
-            padding: 20px;
-            box-shadow: 0 2px 4px rgba(0,0,0,0.1);
-            margin-top: 0;
-        }
-        .controls-container {
-            display: flex;
-            gap: 10px;
-            margin-top: 10px;
-        }
-        </style>
-    """, unsafe_allow_html=True)
-
     col1, col2 = st.columns(2)
 
     # Documento 1
@@ -187,42 +140,12 @@ def display_discourse_results(result, lang_code, discourse_t):
         with st.expander(discourse_t.get('doc1_title', 'Documento 1'), expanded=True):
             st.subheader(discourse_t.get('key_concepts', 'Conceptos Clave'))
             if 'key_concepts1' in result:
-                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)
+                df1 = pd.DataFrame(result['key_concepts1'], columns=['Concepto', 'Frecuencia'])
+                df1['Frecuencia'] = df1['Frecuencia'].round(2)
+                st.table(df1)
 
                 if 'graph1' in result:
-                    with st.container():
-                        st.markdown('<div class="graph-container">', unsafe_allow_html=True)
-                        st.pyplot(result['graph1'])
-                        
-                        # Añadir botones de control para el grafo 1
-                        button_col1, spacer_col1 = st.columns([1,4])
-                        with button_col1:
-                            st.download_button(
-                                label="📥 " + discourse_t.get('download_graph', "Download"),
-                                data=result['graph1_bytes'] if 'graph1_bytes' in result else None,
-                                file_name="discourse_graph1.png",
-                                mime="image/png",
-                                use_container_width=True
-                            )
-                        
-                        with st.expander("📊 " + discourse_t.get('graph_help', "Graph Interpretation")):
-                            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)
+                    st.pyplot(result['graph1'])
                 else:
                     st.warning(discourse_t.get('graph_not_available', 'Gráfico no disponible'))
             else:
@@ -233,42 +156,12 @@ def display_discourse_results(result, lang_code, discourse_t):
         with st.expander(discourse_t.get('doc2_title', 'Documento 2'), expanded=True):
             st.subheader(discourse_t.get('key_concepts', 'Conceptos Clave'))
             if 'key_concepts2' in result:
-                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_concepts2']
-                        ])}
-                    </div>
-                """
-                st.markdown(concepts_html, unsafe_allow_html=True)
+                df2 = pd.DataFrame(result['key_concepts2'], columns=['Concepto', 'Frecuencia'])
+                df2['Frecuencia'] = df2['Frecuencia'].round(2)
+                st.table(df2)
 
                 if 'graph2' in result:
-                    with st.container():
-                        st.markdown('<div class="graph-container">', unsafe_allow_html=True)
-                        st.pyplot(result['graph2'])
-                        
-                        # Añadir botones de control para el grafo 2
-                        button_col2, spacer_col2 = st.columns([1,4])
-                        with button_col2:
-                            st.download_button(
-                                label="📥 " + discourse_t.get('download_graph', "Download"),
-                                data=result['graph2_bytes'] if 'graph2_bytes' in result else None,
-                                file_name="discourse_graph2.png",
-                                mime="image/png",
-                                use_container_width=True
-                            )
-                        
-                        with st.expander("📊 " + discourse_t.get('graph_help', "Graph Interpretation")):
-                            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)
+                    st.pyplot(result['graph2'])
                 else:
                     st.warning(discourse_t.get('graph_not_available', 'Gráfico no disponible'))
             else:
@@ -276,4 +169,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.'))