Keep column selector for base columns, use dataset/metric filters for data columns

app.py

@@ -99,6 +99,15 @@ def build_tab(df, name):
                     elem_id="search-bar"
                 )
 
+                # Column selector for base columns only (not dataset+metric combos)
+                base_cols = ["Model", "Organization", "First Author"]
+                col_selector = gr.CheckboxGroup(
+                    choices=base_cols,
+                    value=base_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:
@@ -148,7 +157,7 @@ def build_tab(df, name):
                     elem_classes="filter-group"
                 )
 
-        def update(search, md, org, dset, metrics):
+        def update(search, md, org, dset, metrics, cols):
             filtered = pivoted_df.copy()
 
             if search:
@@ -160,8 +169,13 @@ def build_tab(df, name):
             if org:
                 filtered = filtered[filtered["Organization"].isin(org)]
 
-            # Filter by dataset and metric - hide columns that don't match
-            display_cols = ["Model", "Organization", "First Author"]
+            # Build display columns based on selected base columns and dataset/metric filters
+            display_cols = []
+
+            # Add selected base columns
+            for col in cols:
+                if col in base_cols:
+                    display_cols.append(col)
 
             # Add metric columns that match selected datasets and metrics
             for col in metric_combo_cols:
@@ -175,11 +189,12 @@ def build_tab(df, name):
 
             return filtered
 
-        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)
+        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)
 
 
 custom_css = """
@@ -261,6 +276,21 @@ 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;
+}
+
+.column-select label > span {
+    display: inline-flex;
+    align-items: center;
+}
 """
 
 TITLE = """<h1 align="center" id="space-title">MarsBench Leaderboard</h1>"""