Refactor GET_DEFAULT_CHECKBOX function to remove "Model Name" column

app.py

@@ -80,7 +80,7 @@ def GET_DEFAULT_TEXTBOX():
 
 def GET_DEFAULT_CHECKBOX(subset):
     choices = list(LEADERBOARD_DF[subset].columns)
-    # choices.remove("Model Name")
+    choices.remove("Model Name")
     # print("Choices:", [c.name for c in fields(AutoEvalColumn) if not c.hidden])
     return gr.CheckboxGroup(
         choices=choices,
@@ -102,6 +102,8 @@ def init_leaderboard(dataframes):
     with gr.Row():
         # datatype = 
         df = gr.Dataframe(data, type="pandas")
+    
+    old_version = NEWEST_VERSION
 
     def refresh(subset):
         global LEADERBOARD_DF
@@ -116,23 +118,25 @@ def init_leaderboard(dataframes):
         print("Subset:", subset)
         print("Search Term:", search_term)
         print("Selected Columns:", selected_columns)
-        try:
-            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 = [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],
-            )
-        except Exception as e:
-            print("====================================")
-            print("Error:", e)
-            print("====================================")
-            return refresh(subset)
+        if old_version != subset:
+            search_term = None
+            selected_columns = GET_DEFAULT_CHECKBOX(subset)
+        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")
+        selected_columns = [
+            c for c in filtered_data.columns if c 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],
+        )
 
     with gr.Row():
         refresh_button = gr.Button("Refresh")