added arrow colors

app.py

@@ -426,6 +426,8 @@ def build_leaderboard_tab(elo_results_file, leaderboard_table_file, show_plot=Fa
         elo_subset_results = category_elo_results[category]
         arena_df = arena_dfs["Total"]
         arena_values = get_arena_table(arena_df, model_table_df, arena_subset_df)
+        if category != "Total":
+            arena_values = update_leaderboard_df(arena_values)
         p1 = elo_subset_results["win_fraction_heatmap"]
         p2 = elo_subset_results["battle_count_heatmap"]
         p3 = elo_subset_results["bootstrap_elo_rating"]
@@ -434,6 +436,18 @@ def build_leaderboard_tab(elo_results_file, leaderboard_table_file, show_plot=Fa
         """
         leaderboard_md = make_category_arena_leaderboard_md(arena_df, arena_subset_df, name=category)
         return arena_values, p1, p2, p3, p4, more_stats_md, leaderboard_md
+    
+    def update_leaderboard_df(arena_table_vals):
+        elo_datarame = pd.DataFrame(arena_table_vals, columns=["Rank", "Model", "Arena Elo", "95% CI", "Votes", "Organization", "License", "Knowledge Cutoff"])
+
+        # goal: color the rows based on the rank with styler
+        def highlight_max(s):
+            # all items in S which contain up arrow should be green, down arrow should be red, otherwise black
+            return ["color: green" if "\u2191" in v else "color: red" if "\u2193" in v else "" for v in s]
+        
+        styled_df = elo_datarame.style.apply(highlight_max, subset=["Rank"])
+
+        return styled_df
                 
     category_dropdown.change(update_leaderboard_and_plots, inputs=[category_dropdown], outputs=[elo_display_df, plot_1, plot_2, plot_3, plot_4, more_stats_md, category_deets])