style theme

app.py

@@ -13,6 +13,7 @@ import plotly.graph_objects as go
 from apscheduler.schedulers.background import BackgroundScheduler
 import numpy as np
 from gradio.themes.utils import fonts, colors
+from dataclasses import fields
 
 from src.about import (
     CITATION_BUTTON_LABEL,
@@ -254,6 +255,12 @@ def init_leaderboard(dataframe, visible_columns=None):
 
         datatypes.append(col_type)
 
+    # Create a mapping from internal column names to display names
+    all_columns_map = {field.name: field.display_name for field in fields(GUARDBENCH_COLUMN)}
+
+    # Create the list of headers using display names
+    display_headers = [all_columns_map.get(col, col) for col in visible_columns]
+
     # Create a dummy column for search functionality if it doesn't exist
     if 'search_dummy' not in dataframe.columns:
         dataframe['search_dummy'] = dataframe.apply(
@@ -279,7 +286,7 @@ def init_leaderboard(dataframe, visible_columns=None):
 
     return gr.Dataframe(
         value=display_df,
-        headers=visible_columns,
+        headers=display_headers,
         datatype=datatypes,  # Now using the correct format
         interactive=False,
         wrap=True,