Update app.py

app.py

@@ -104,41 +104,40 @@ def update_metrics(selected_benchmarks):
     return list(updated_metrics)
 
 def update_leaderboard(selected_methods, selected_metrics):
-    return make_leaderboard_df(selected_methods, selected_metrics)
+    return make_leaderboard_styler(selected_methods, selected_metrics)
 
 
-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>"
+def style_method(val: str) -> str:
+    """Return CSS for the model name cell."""
+    return f"color:{color_dict.get(val, 'black')}; font-weight:bold;"
 
-# ――― background shading for the top-5 of a numeric column ―――
-#    darkest → lightest
+# darkest → lightest green for ranks 1-5
 TOP5_GREENS = ["#006400", "#228B22", "#32CD32", "#7CFC00", "#ADFF2F"]
 
 def shade_top5(col: pd.Series) -> list[str]:
-    """Return background‐colour CSS for the best 5 numeric cells of a column."""
+    """Background colours for the best 5 numeric values of one column."""
     if not pd.api.types.is_numeric_dtype(col):
-        return [""] * len(col)          # leave non-numeric cols untouched
+        return [""] * len(col)
     ranks = col.rank(ascending=False, method="first")
     return [
         f"background-color:{TOP5_GREENS[int(r)-1]};" if r <= 5 else ""
         for r in ranks
     ]
 
-def make_leaderboard_df(selected_methods=None, selected_metrics=None):
-    """Fetch, tidy, colour, and return a Styler that Gradio will render."""
+def make_leaderboard_styler(selected_methods=None, selected_metrics=None):
+    # pull / slice the raw dataframe
     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 columns only (everything except 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
+        # 1. shade top-5 in numeric columns
         .apply(shade_top5, axis=0, subset=numeric_cols)
-        .format(escape=False, precision=4)   # escape=False keeps <span> alive
+        # 2. colour the Method text
+        .applymap(style_method, subset=["Method"])
+        .format(precision=4)          # nice numbers
     )
     return styler
 
@@ -252,7 +251,7 @@ with block:
             baseline_header  = ["Method"] + metric_names
             baseline_datatype = ['markdown'] + ['number'] * len(metric_names)            
 
-            styler = make_leaderboard_df()
+            styler = make_leaderboard_styler()
         
             with gr.Row(show_progress=True, variant='panel'):
                 data_component = gr.Dataframe(