Update modules/ui/ui.py

modules/ui/ui.py

@@ -6,6 +6,7 @@ from io import BytesIO
 import base64
 import matplotlib.pyplot as plt
 import pandas as pd
+import numpy as np
 import time
 from datetime import datetime
 from streamlit_player import st_player  # Necesitarás instalar esta librería: pip install streamlit-player
@@ -1018,13 +1019,25 @@ def display_discourse_results(result, lang_code, t):
                 st.warning(t.get('concepts_not_available', "Los conceptos clave no están disponibles."))
                 
     # Comparación de conceptos clave (ahora fuera de las columnas para ocupar todo el ancho)
-    st.subheader(t.get('comparison', "Comparación de Conceptos Clave"))
+    st.subheader(t.get('comparison', "Relación de conceptos entre ambos documentos"))
     if 'key_concepts1' in result and 'key_concepts2' in result:
-        df1 = pd.DataFrame(result['key_concepts1'], columns=['Concepto', 'Frecuencia']).set_index('Concepto')
-        df2 = pd.DataFrame(result['key_concepts2'], columns=['Concepto', 'Frecuencia']).set_index('Concepto')
+        df1 = pd.DataFrame(result['key_concepts1'], columns=['Concepto', 'Frecuencia'])
+        df2 = pd.DataFrame(result['key_concepts2'], columns=['Concepto', 'Frecuencia'])
+        
+        # Asegurarse de que las frecuencias sean numéricas
+        df1['Frecuencia'] = pd.to_numeric(df1['Frecuencia'], errors='coerce').fillna(0)
+        df2['Frecuencia'] = pd.to_numeric(df2['Frecuencia'], errors='coerce').fillna(0)
+        
+        df1 = df1.set_index('Concepto')
+        df2 = df2.set_index('Concepto')
         df_comparison = pd.concat([df1, df2], axis=1, keys=[t.get('file_uploader1', "Documento 1"), t.get('file_uploader2', "Documento 2")])
         df_comparison = df_comparison.fillna(0)
-        st.dataframe(df_comparison, width=1000)  # Ajusta el ancho según sea necesario
+        
+        # Asegurarse de que todas las columnas sean del tipo float
+        for col in df_comparison.columns:
+            df_comparison[col] = df_comparison[col].astype(float)
+        
+        st.dataframe(df_comparison.style.format("{:.2f}"), width=1000)  # Ajusta el ancho según sea necesario
     else:
         st.warning(t.get('comparison_not_available', "La comparación no está disponible."))