Refactor GET_DEFAULT_CHECKBOX function to remove "Model Name" column and handle exceptions in app.py

app.py

@@ -6,6 +6,7 @@ from huggingface_hub import snapshot_download
 from gradio.components.textbox import Textbox
 from gradio.components.dataframe import Dataframe
 from gradio.components.checkboxgroup import CheckboxGroup
+import copy
 
 # from fastchat.serve.monitor.monitor import build_leaderboard_tab, build_basic_stats_tab, basic_component_values, leader_component_values
 
@@ -123,17 +124,16 @@ def init_leaderboard(dataframes):
         if old_version != subset:
             search_term = None
             selected_columns = GET_DEFAULT_CHECKBOX(subset)
+        bak_selected_columns = copy.deepcopy(selected_columns)
         old_version = subset
         filtered_data = dataframes[subset]
         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.append("Model Name")
-        try:
+        if isinstance(selected_columns, CheckboxGroup):
             selected_columns = selected_columns.choices
-        except Exception as e:
-            print(e)
-            selected_columns = selected_columns
+        print("Selected Columns:", selected_columns)
         selected_columns = [
             c for c in filtered_data.columns if c in selected_columns or c == "Model Name"
         ]
@@ -143,7 +143,7 @@ def init_leaderboard(dataframes):
             selected_data,
             type="pandas",
             # datatype=[c.type for c in fields(AutoEvalColumn) if c.name in selected_columns],
-        ), research_textbox, selected_columns
+        ), research_textbox, bak_selected_columns
 
     with gr.Row():
         refresh_button = gr.Button("Refresh")