Include Max and Min Distances in Table

app.py

@@ -127,11 +127,27 @@ def create_table(df_distances):
     df_table = df_distances.copy()
     df_table.reset_index(inplace=True)
     df_table.rename(columns={'index': 'Synthetic'}, inplace=True)
+    
+    # Calcular las filas de medias, máximos y mínimos para cada columna numérica
+    min_row = {"Synthetic": "Min."}
+    mean_row = {"Synthetic": "Mean"}
+    max_row = {"Synthetic": "Max."}
+    
+    for col in df_table.columns:
+        if col != "Synthetic":
+            min_row[col] = df_table[col].min()
+            mean_row[col] = df_table[col].mean()
+            max_row[col] = df_table[col].max()
+    
+    # Agregar las filas de medias, máximos y mínimos al final del DataFrame
+    df_table = pd.concat([df_table, pd.DataFrame([min_row, mean_row, max_row])], ignore_index=True)
+
     source_table = ColumnDataSource(df_table)
     columns = [TableColumn(field='Synthetic', title='Synthetic')]
     for col in df_table.columns:
         if col != 'Synthetic':
             columns.append(TableColumn(field=col, title=col))
+    
     row_height = 28
     header_height = 30
     total_height = header_height + len(df_table) * row_height
@@ -139,6 +155,8 @@ def create_table(df_distances):
     data_table = DataTable(source=source_table, columns=columns, sizing_mode='stretch_width', height=total_height)
     return data_table, df_table, source_table
 
+
+
 # Función que ejecuta todo el proceso para un modelo determinado
 def run_model(model_name):
     embeddings = load_embeddings(model_name)
@@ -278,7 +296,7 @@ def run_model(model_name):
 
     # Agregar un botón de descarga en Streamlit
     st.download_button(
-        label="Exportar Table",
+        label="Export Table",
         data=buffer,
         file_name=f"cluster_distances_{model_name}.xlsx",
         mime="application/vnd.openxmlformats-officedocument.spreadsheetml.sheet",
@@ -293,11 +311,11 @@ def main():
     tabs = st.tabs(["Donut", "Idefics2"])
     
     with tabs[0]:
-        st.markdown('<h2 class="sub-title">Modelo Donut 🤗</h2>', unsafe_allow_html=True)
+        st.markdown('<h2 class="sub-title">Donut 🤗</h2>', unsafe_allow_html=True)
         run_model("Donut")
     
     with tabs[1]:
-        st.markdown('<h2 class="sub-title">Modelo Idefics2 🤗</h2>', unsafe_allow_html=True)
+        st.markdown('<h2 class="sub-title">Idefics2 🤗</h2>', unsafe_allow_html=True)
         run_model("Idefics2")
 
 if __name__ == "__main__":