Remove column selector, use dataset and metric filters instead

app.py

@@ -99,13 +99,6 @@ def build_tab(df, name):
                     elem_id="search-bar"
                 )
 
-                col_selector = gr.CheckboxGroup(
-                    choices=all_cols,
-                    value=all_cols,
-                    label="Select Columns to Display:",
-                    elem_classes="column-select"
-                )
-
                 # Set datatype to 'markdown' for First Author column to enable clickable links
                 datatypes = []
                 for col in pivoted_df.columns:
@@ -155,7 +148,7 @@ def build_tab(df, name):
                     elem_classes="filter-group"
                 )
 
-        def update(search, md, org, dset, metrics, cols):
+        def update(search, md, org, dset, metrics):
             filtered = pivoted_df.copy()
 
             if search:
@@ -168,32 +161,25 @@ def build_tab(df, name):
                 filtered = filtered[filtered["Organization"].isin(org)]
 
             # Filter by dataset and metric - hide columns that don't match
-            if cols:
-                display_cols = []
-                # Always include base columns
-                for col in ["Model", "Organization", "First Author"]:
-                    if col in cols and col in filtered.columns:
-                        display_cols.append(col)
-
-                # Add metric columns that match selected datasets and metrics
-                for col in cols:
-                    if col in metric_combo_cols:
-                        # Check if this column matches selected datasets and metrics
-                        col_dataset = col.split(" (")[0]
-                        col_metric = col.split(" (")[1].rstrip(")")
-                        if col_dataset in dset and col_metric in metrics:
-                            display_cols.append(col)
-
-                filtered = filtered[display_cols]
+            display_cols = ["Model", "Organization", "First Author"]
+
+            # Add metric columns that match selected datasets and metrics
+            for col in metric_combo_cols:
+                # Check if this column matches selected datasets and metrics
+                col_dataset = col.split(" (")[0]
+                col_metric = col.split(" (")[1].rstrip(")")
+                if col_dataset in dset and col_metric in metrics:
+                    display_cols.append(col)
+
+            filtered = filtered[display_cols]
 
             return filtered
 
-        search_bar.change(update, [search_bar, model_filter, org_filter, dataset_filter, metric_filter, col_selector], table)
-        model_filter.change(update, [search_bar, model_filter, org_filter, dataset_filter, metric_filter, col_selector], table)
-        org_filter.change(update, [search_bar, model_filter, org_filter, dataset_filter, metric_filter, col_selector], table)
-        dataset_filter.change(update, [search_bar, model_filter, org_filter, dataset_filter, metric_filter, col_selector], table)
-        metric_filter.change(update, [search_bar, model_filter, org_filter, dataset_filter, metric_filter, col_selector], table)
-        col_selector.change(update, [search_bar, model_filter, org_filter, dataset_filter, metric_filter, col_selector], table)
+        search_bar.change(update, [search_bar, model_filter, org_filter, dataset_filter, metric_filter], table)
+        model_filter.change(update, [search_bar, model_filter, org_filter, dataset_filter, metric_filter], table)
+        org_filter.change(update, [search_bar, model_filter, org_filter, dataset_filter, metric_filter], table)
+        dataset_filter.change(update, [search_bar, model_filter, org_filter, dataset_filter, metric_filter], table)
+        metric_filter.change(update, [search_bar, model_filter, org_filter, dataset_filter, metric_filter], table)
 
 
 custom_css = """
@@ -275,22 +261,6 @@ custom_css = """
 .filter-group label {
     font-size: 14px;
 }
-
-.column-select {
-    margin-bottom: 1.5em;
-}
-
-.column-select label {
-    display: flex;
-    flex-wrap: wrap;
-    gap: 0.5em;
-}
-
-/* Styling for the column select checkboxes to display in rows */
-.column-select label > span {
-    display: inline-flex;
-    align-items: center;
-}
 """
 
 TITLE = """<h1 align="center" id="space-title">MarsBench Leaderboard</h1>"""