update

app.py

@@ -120,7 +120,6 @@ def get_folders_matching_format(directory):
 
 
 def get_unique_column_names(all_data):
-
     # column_names = {}
     #
     # for folder_name, files in all_data.items():
@@ -132,7 +131,12 @@ def get_unique_column_names(all_data):
     #
     # return list(column_names.keys())
 
-    return ['ao3_\u200benglish', 'bbc_\u200bnews', 'wikipedia_\u200benglish', 'arxiv_\u200bcomputer_\u200bscience', 'arxiv_\u200bphysics', 'github_\u200bcpp', 'github_\u200bpython']
+    return ['ao3_\u200benglish', 'bbc_\u200bnews', 'wikipedia_\u200benglish', 'arxiv_\u200bcomputer_\u200bscience',
+            'arxiv_\u200bphysics', 'github_\u200bcpp', 'github_\u200bpython']
+
+
+def color_cell(value):
+    return 'background-color: #fffdd0' if pd.notna(value) else 'default'
 
 
 def update_table(period: str,
@@ -177,7 +181,8 @@ def update_table(period: str,
         if 'Average' in color_columns:
             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)']])
+            target_color_columns.extend([col for col in filtered_data.columns if
+                                         col not in ['Name', 'Parameters Count (B)', 'Average (lower=better)']])
 
         # styler = filtered_data.style.format(formatter).background_gradient(
         #     cmap=cmap,
@@ -187,28 +192,26 @@ def update_table(period: str,
         # ).apply(color_column, subset=['Parameters Count (B)'])
 
         # for better visualization
-        second_largest = filtered_data['Average (lower=better)'].nlargest(2).iloc[-1]
-        min_value = filtered_data['Average (lower=better)'].min().min()
-        if len(models) == 1 and '~7B' in models:
-            max_value = second_largest
-        else:
-            max_value = filtered_data['Average (lower=better)'].max().max()
-
-        specific_column = 'Average (lower=better)'
+        vmin = {}
+        vmax = {}
+        for column in filtered_data.columns:
+            if column in ['Name', 'Parameters Count (B)']:
+                continue
+            col_values = filtered_data[column]
+            second_largest = col_values.nlargest(2).iloc[-1]
+            vmin[column] = col_values.min()
+            vmax[column] = second_largest
 
-        styler_specific = filtered_data.style.background_gradient(
-            cmap=cmap,
-            subset=[specific_column],
-            vmin=min_value,
-            vmax=max_value
-        )
-
-        styler_other = filtered_data.style.background_gradient(
-            cmap=cmap,
-            subset=[col for col in target_color_columns if col != specific_column]
-        )
+        target_color_columns = []
+        if 'Average' in color_columns:
+            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)']])
 
-        styler = styler_specific.use(styler_other.export()).format(formatter).apply(color_column, subset=['Parameters Count (B)'])
+        styler = filtered_data.style.format(formatter).map(color_cell, subset=['Parameters Count (B)'])
+        for column in target_color_columns:
+            styler = styler.background_gradient(cmap=cmap, subset=[column], vmin=vmin[column], vmax=vmax[column])
 
         return styler
     else:
@@ -307,7 +310,8 @@ 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=[125, 50, 50, 35, 35, 35, 35, 35, 35, 35], wrap=True)
+            table = gr.Dataframe(initial_data, column_widths=[125, 50, 50, 35, 35, 35, 35, 35, 35, 35], wrap=True,
+                                 height=750)
 
             period_selector.change(update_table,
                                    inputs=[period_selector, model_selector, metric_selector, colfilter, color_selector],