Split the group of checkboxes for selecting columns to display

app.py

@@ -15,6 +15,7 @@ from src.about import (
     INTRODUCTION_TEXT_JP,
     LLM_BENCHMARKS_TEXT,
     TITLE,
+    TaskType,
 )
 from src.display.css_html_js import custom_css
 from src.display.utils import (
@@ -181,7 +182,6 @@ def select_columns(df: pd.DataFrame, columns: list) -> pd.DataFrame:
 
 def update_table(
     hidden_df: pd.DataFrame,
-    columns: list,
     type_query: list,
     precision_query: str,
     size_query: list,
@@ -190,7 +190,9 @@ def update_table(
     version_query: list,
     backend_query: list,
     query: str,
+    *columns,
 ):
+    columns = [item for column in columns for item in column]
     print(
         f"Update table called with: type_query={type_query}, precision_query={precision_query}, size_query={size_query}"
     )
@@ -284,18 +286,30 @@ with gr.Blocks() as demo_leaderboard:
                     elem_id="search-bar",
                 )
             with gr.Row():
-                shown_columns = gr.CheckboxGroup(
-                    label="Select columns to show",
-                    choices=[
-                        c.name for c in fields(AutoEvalColumn) if not c.hidden and not c.never_hidden and not c.dummy
-                    ],
-                    value=[
-                        c.name
-                        for c in fields(AutoEvalColumn)
-                        if c.displayed_by_default and not c.hidden and not c.never_hidden
-                    ],
-                    elem_id="column-select",
-                )
+                shown_columns_dict = {}
+                for task_type in TaskType:
+                    if task_type == TaskType.NotTask:
+                        label = "Select columns to show"
+                    else:
+                        label = f"{task_type.name}(Select columns to show)"
+                    shown_column = gr.CheckboxGroup(
+                        label=label,
+                        choices=[
+                            c.name
+                            for c in fields(AutoEvalColumn)
+                            if not c.hidden and not c.never_hidden and not c.dummy and c.task_type == task_type
+                        ],
+                        value=[
+                            c.name
+                            for c in fields(AutoEvalColumn)
+                            if c.displayed_by_default
+                            and not c.hidden
+                            and not c.never_hidden
+                            and c.task_type == task_type
+                        ],
+                        elem_id="column-select",
+                    )
+                    shown_columns_dict[task_type.name] = shown_column
         with gr.Column(min_width=320):
             filter_columns_type = gr.CheckboxGroup(
                 label="Model types",
@@ -364,7 +378,6 @@ with gr.Blocks() as demo_leaderboard:
     gr.on(
         triggers=[
             hidden_search_bar.change,
-            shown_columns.change,
             filter_columns_type.change,
             filter_columns_precision.change,
             filter_columns_size.change,
@@ -373,11 +386,11 @@ with gr.Blocks() as demo_leaderboard:
             filter_columns_version.change,
             filter_columns_backend.change,
             search_bar.submit,
-        ],
+        ]
+        + [shown_columns.change for shown_columns in shown_columns_dict.values()],
         fn=update_table,
         inputs=[
             hidden_leaderboard_table_for_search,
-            shown_columns,
             filter_columns_type,
             filter_columns_precision,
             filter_columns_size,
@@ -386,7 +399,8 @@ with gr.Blocks() as demo_leaderboard:
             filter_columns_version,
             filter_columns_backend,
             search_bar,
-        ],
+        ]
+        + [shown_columns for shown_columns in shown_columns_dict.values()],
         outputs=leaderboard_table,
     )