better visualization

app.py

@@ -164,7 +164,8 @@ def update_table(period: str,
         if len(combined_data) > 0:
             sorted_data = combined_data.sort_values(by=sort_by, ascending=ascending)
             sorted_data = sorted_data.rename(columns={'Average (The lower the better)': 'Average (lower=better)'})
-            visible_columns = ['Name', 'Parameters Count (B)', 'Average (lower=better)'] + visible_columns
+            sorted_data = sorted_data.rename(columns={'Parameters Count (B)': 'Params (B)'})
+            visible_columns = ['Name', 'Params (B)', 'Average (lower=better)'] + visible_columns
             filtered_data = sorted_data[visible_columns]
 
             filtered_data.columns = [col.replace('_', ' ') for col in filtered_data.columns]
@@ -178,7 +179,7 @@ def update_table(period: str,
             vmin = {}
             vmax = {}
             for column in filtered_data.columns:
-                if column in ['Name', 'Parameters Count (B)']:
+                if column in ['Name', 'Params (B)']:
                     continue
                 col_values = filtered_data[column]
                 if len(col_values) > 1:
@@ -191,9 +192,10 @@ def update_table(period: str,
                 target_color_columns.append('Average (lower=better)')
             if 'Individual Tests' in color_columns:
                 target_color_columns.extend([col for col in filtered_data.columns if
-                                             col not in ['Name', 'Parameters Count (B)', 'Average (lower=better)']])
+                                             col not in ['Name', 'Params (B)', 'Average (lower=better)']])
 
-            styler = filtered_data.style.format(formatter)
+
+            styler = filtered_data.style.format(formatter).applymap(color_cell, subset=['Params (B)'])
 
             for column in target_color_columns:
                 if column in vmin and column in vmax:  # Ensure that the vmin and vmax dicts contain the column
@@ -395,7 +397,7 @@ with gr.Blocks(css=css) as demo:
                                                  choices=get_unique_column_names(all_data),
                                                  value=get_unique_column_names(all_data))
 
-            table = gr.Dataframe(initial_data, column_widths=[130, 60, 60, 35, 35, 35, 35, 35, 35, 35],
+            table = gr.Dataframe(initial_data, column_widths=[130, 50, 50, 35, 35, 35, 35, 35, 35, 35],
                                  wrap=True,
                                  height=800,
                                  )