Refactor init_leaderboard function to update data outputs to dataframe and improve dropdown UI

app.py

@@ -71,9 +71,11 @@ except Exception:
 
 LEADERBOARD_DF = get_leaderboard_df(RESULTS_REPO)
 
+
 def GET_DEFAULT_TEXTBOX():
     return gr.Textbox("", placeholder="🔍 Search Models... [press enter]", label="Filter Models by Name")
 
+
 def GET_DEFAULT_CHECKBOX():
     print("Choices:", [c.name for c in fields(AutoEvalColumn) if not c.hidden])
     return gr.CheckboxGroup(
@@ -86,7 +88,6 @@ def GET_DEFAULT_CHECKBOX():
 def init_leaderboard(dataframes):
     subsets = list(dataframes.keys())
 
-
     with gr.Row():
         selected_subset = gr.Dropdown(choices=subsets, label="Select Dataset Subset", value=subsets[-1])
         research_textbox = GET_DEFAULT_TEXTBOX()
@@ -102,7 +103,7 @@ def init_leaderboard(dataframes):
         global LEADERBOARD_DF
         LEADERBOARD_DF = get_leaderboard_df(RESULTS_REPO)
         default_columns = [c.name for c in fields(AutoEvalColumn) if c.displayed_by_default]
-        
+
         return update_data(subset, None, default_columns), GET_DEFAULT_TEXTBOX(), GET_DEFAULT_CHECKBOX()
 
     def update_data(subset, search_term, selected_columns):
@@ -111,11 +112,15 @@ def init_leaderboard(dataframes):
         print("Selected Columns:", selected_columns)
         filtered_data = dataframes[subset]
         if search_term:
-            filtered_data = filtered_data[dataframes[subset]['Model Name'].str.contains(search_term, case=False)]
-        selected_data.sort_values(by="Total", ascending=False, inplace=True)
+            filtered_data = filtered_data[dataframes[subset]["Model Name"].str.contains(search_term, case=False)]
+        filtered_data = filtered_data.sort_values(by="Total", ascending=False, inplace=True)
         selected_columns = [c.name for c in fields(AutoEvalColumn) if c.name in selected_columns]
         selected_data = filtered_data[selected_columns]
-        return gr.DataFrame(selected_data, type="pandas", datatype=[c.type for c in fields(AutoEvalColumn) if c.name in selected_columns])
+        return gr.DataFrame(
+            selected_data,
+            type="pandas",
+            datatype=[c.type for c in fields(AutoEvalColumn) if c.name in selected_columns],
+        )
 
     with gr.Row():
         refresh_button = gr.Button("Refresh")

src/display/utils.py

@@ -26,7 +26,9 @@ class ColumnContent:
 auto_eval_column_dict = []
 # Init
 # auto_eval_column_dict.append(["model_type_symbol", ColumnContent, ColumnContent("T", "str", True, never_hidden=True)])
-auto_eval_column_dict.append(["model", ColumnContent, ColumnContent("Model Name", "str", True, True, never_hidden=True)])
+auto_eval_column_dict.append(
+    ["model", ColumnContent, ColumnContent("Model Name", "str", True, True, never_hidden=True)]
+)
 # Scores
 auto_eval_column_dict.append(["Overall", ColumnContent, ColumnContent("Total", "number", True)])
 for task in Tasks: