Refactor GET_DEFAULT_CHECKBOX function in app.py

app.py

@@ -79,11 +79,13 @@ def GET_DEFAULT_TEXTBOX():
 
 
 def GET_DEFAULT_CHECKBOX():
-    print("Choices:", [c.name for c in fields(AutoEvalColumn) if not c.hidden])
+    choices = list(LEADERBOARD_DF.columns)
+    choices.remove("Model Name")
+    # print("Choices:", [c.name for c in fields(AutoEvalColumn) if not c.hidden])
     return gr.CheckboxGroup(
-        choices=[c.name for c in fields(AutoEvalColumn) if not c.hidden],
+        choices=choices,
         label="Select Columns to Display",
-        value=[c.name for c in fields(AutoEvalColumn) if c.displayed_by_default],
+        value=choices,
     )
 
 
@@ -98,13 +100,15 @@ def init_leaderboard(dataframes):
     data = dataframes[NEWEST_VERSION]
 
     with gr.Row():
-        datatype = [c.type for c in fields(AutoEvalColumn)]
-        df = gr.Dataframe(data, datatype=datatype, type="pandas")
+        # datatype = 
+        df = gr.Dataframe(data, type="pandas")
 
     def refresh(subset):
         global LEADERBOARD_DF
         LEADERBOARD_DF = get_leaderboard_df(RESULTS_REPO)
-        default_columns = [c.name for c in fields(AutoEvalColumn) if c.displayed_by_default]
+        # default_columns = [c.name for c in fields(AutoEvalColumn) if c.displayed_by_default]
+        default_columns = list(LEADERBOARD_DF.columns)
+        default_columns.remove("Model Name")
 
         return update_data(subset, None, default_columns), GET_DEFAULT_TEXTBOX(), GET_DEFAULT_CHECKBOX()
 
@@ -116,12 +120,12 @@ def init_leaderboard(dataframes):
         if search_term:
             filtered_data = filtered_data[dataframes[subset]["Model Name"].str.contains(search_term, case=False)]
         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_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],
+            # datatype=[c.type for c in fields(AutoEvalColumn) if c.name in selected_columns],
         )
 
     with gr.Row():