Update app.py

app.py

@@ -104,26 +104,12 @@ def update_metrics(selected_benchmarks):
     return list(updated_metrics)
 
 def update_leaderboard(selected_methods, selected_metrics):
-    df = get_baseline_df(selected_methods, selected_metrics).round(4)
-    styler = (
-        df
-        .style
-        .apply(highlight_top5, axis=0) # green shades
-        .applymap(style_method, subset=["Method"])  # text colours
-        .format(precision=4)                        # keep tidy numbers
-    )
-
-    return styler
-
+    return make_leaderboard_df(selected_methods, selected_metrics)
 
-def format_method(name: str) -> str:
-    """Wrap a model name in a coloured <span> so Gradio renders it."""
-    colour = color_dict.get(name, "black")
-    return f"<span style='color:{colour}; font-weight:bold;'>{name}</span>"
 
-# ――― text colour for the Method column ―――
-def style_method(val: str) -> str:
-    return f"color:{color_dict.get(val, 'black')}; font-weight:bold;"
+def colour_method(name: str) -> str:
+    """Wrap model name in a coloured <span>."""
+    return f"<span style='color:{color_dict.get(name, 'black')}; font-weight:bold;'>{name}</span>"
 
 # ――― background shading for the top-5 of a numeric column ―――
 #    darkest → lightest
@@ -144,6 +130,24 @@ def highlight_top5(col: pd.Series) -> list[str]:
             styles.append("")
     return styles
 
+def make_leaderboard_df(selected_methods=None, selected_metrics=None):
+    """Fetch, tidy, colour, and return a Styler that Gradio will render."""
+    df = get_baseline_df(selected_methods, selected_metrics).round(4)
+
+    # 1️⃣ colour method names via inline HTML
+    df["Method"] = df["Method"].apply(colour_method)
+
+    numeric_cols = [c for c in df.columns if c != "Method"]
+
+    # 2️⃣ build Styler: shade numeric top-5, keep HTML intact
+    styler = (
+        df.style
+        .apply(shade_top5, axis=0, subset=numeric_cols)
+        .format(escape=False, precision=4)   # escape=False keeps <span> alive
+    )
+    return styler
+
+
 # ------- Visualisation helpers ---------------------------------------------
 
 
@@ -253,13 +257,7 @@ with block:
             baseline_header  = ["Method"] + metric_names
             baseline_datatype = ['markdown'] + ['number'] * len(metric_names)            
 
-            styler = (
-                baseline_value
-                .style
-                .apply(highlight_top5, axis=0) # green shades
-                .applymap(style_method, subset=["Method"])  # text colours
-                .format(precision=4)                        # keep tidy numbers
-            )
+            styler = make_leaderboard_df()
         
             with gr.Row(show_progress=True, variant='panel'):
                 data_component = gr.Dataframe(