Update app.py

app.py

@@ -22,18 +22,22 @@ def load_data(file_path):
 # Function to style the DataFrame
 def style_dataframe(df: pd.DataFrame):
     df[RESULTS_COLUMN_NAME] = df.apply(lambda row: [row[SENTIMENT_COLUMN_NAME], row[UNDERSTANDING_COLUMN_NAME], row[PHRASEOLOGY_COLUMN_NAME]], axis=1)
-
     # Insert the new column after the 'Average' column
     cols = list(df.columns)
     cols.insert(cols.index(AVERAGE_COLUMN_NAME) + 1, cols.pop(cols.index(RESULTS_COLUMN_NAME)))
     df = df[cols]
-
     # Create a color ramp using Seaborn
     return df
 
 def styler(df: pd.DataFrame):
     palette = sns.color_palette("RdYlGn", as_cmap=True)
-    styled_df = df.style.background_gradient(cmap=palette, subset=["Params",AVERAGE_COLUMN_NAME, SENTIMENT_COLUMN_NAME, PHRASEOLOGY_COLUMN_NAME, UNDERSTANDING_COLUMN_NAME]).format(precision=2)
+    # Apply reverse color gradient to the "Params" column
+    params_palette = sns.color_palette("RdYlGn_r", as_cmap=True)  # Reversed RdYlGn palette
+    styled_df = df.style.background_gradient(cmap=palette, subset=[AVERAGE_COLUMN_NAME, SENTIMENT_COLUMN_NAME, PHRASEOLOGY_COLUMN_NAME, UNDERSTANDING_COLUMN_NAME]
+                                            ).background_gradient(cmap=params_palette, subset=["Params"]
+                                            ).set_properties(**{'text-align': 'center'}, subset=[AVERAGE_COLUMN_NAME, SENTIMENT_COLUMN_NAME, PHRASEOLOGY_COLUMN_NAME, UNDERSTANDING_COLUMN_NAME]
+                                            ).format("{:.2f}".center(10), subset=[AVERAGE_COLUMN_NAME, SENTIMENT_COLUMN_NAME, PHRASEOLOGY_COLUMN_NAME, UNDERSTANDING_COLUMN_NAME]
+                                            ).format("{:.1f}".center(10), subset=["Params"])
     return styled_df
 
 
@@ -136,14 +140,15 @@ with tab1:
 
     # Display the styled DataFrame
     data = load_data('data.json')
-    data['Params'] = data['Params'].str.replace('B', '')
+    
+    data['Params'] = data['Params'].str.replace('B', '').astype(float)
     data = data.sort_values(by=AVERAGE_COLUMN_NAME, ascending=False)
     styled_df_show = style_dataframe(data)
     styled_df_show = styler(styled_df_show)
 
     st.data_editor(styled_df_show, column_config={
         "Model": st.column_config.TextColumn("Model", help="Model name", width="large"),
-        "Params": st.column_config.NumberColumn("Params [B]", format="%.1f"),
+        "Params": st.column_config.NumberColumn("Params [B]"),
         AVERAGE_COLUMN_NAME: st.column_config.NumberColumn(AVERAGE_COLUMN_NAME),
         RESULTS_COLUMN_NAME: st.column_config.BarChartColumn(
             "Bar chart of results", help="Summary of the results of each task",