Update app.py

app.py

@@ -285,47 +285,41 @@ with gr.Blocks() as demo_leaderboard:
         )
     with gr.Row():
         shown_columns_dict = {}
+        checkboxes = []
         for task_type in TaskType:
             if task_type == TaskType.NotTask:
                 label = "Model details"
             else:
                 label = task_type.value
             with gr.Accordion(label, open=True, elem_classes="accordion"):
-                with gr.Row(height=110):
-                    shown_column = gr.CheckboxGroup(
-                        show_label=False,
-                        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 not c.hidden and not c.never_hidden 
-                            and c.task_type == task_type
-                            and (
-                                (task_type == TaskType.AVG) or  # AVGカテゴリの場合は全て
-                                (task_type != TaskType.AVG and c.average)  # その他のカテゴリはAVGのみ
-                            )
-                        ],
-                        elem_id="column-select",
-                        container=False,
-                    )
-                    shown_columns_dict[task_type.name] = shown_column
+                    with gr.Row(height=110):
+                        shown_column = gr.CheckboxGroup(
+                            show_label=False,
+                            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 not c.hidden and not c.never_hidden 
+                                and c.task_type == task_type
+                                and (
+                                    (task_type == TaskType.AVG) or
+                                    (task_type != TaskType.AVG and c.average)
+                                )
+                            ],
+                            elem_id="column-select",
+                            container=False,
+                        )
+                        shown_columns_dict[task_type.name] = shown_column
+                        checkboxes.append(shown_column)
+                    
                 with gr.Row():
-                    gr.Button("全選択", size="sm").click(
-                        fn=lambda: toggle_all_categories("all", shown_columns_dict),
-                        outputs=[shown_columns_dict[task_type.name] for task_type in TaskType if task_type != TaskType.NotTask]
-                    )
-                    gr.Button("全解除", size="sm").click(
-                        fn=lambda: toggle_all_categories("none", shown_columns_dict),
-                        outputs=[shown_columns_dict[task_type.name] for task_type in TaskType if task_type != TaskType.NotTask]
-                    )
-                    gr.Button("AVGのみ", size="sm").click(
-                        fn=lambda: toggle_all_categories("avg_only", shown_columns_dict),
-                        outputs=[shown_columns_dict[task_type.name] for task_type in TaskType if task_type != TaskType.NotTask]
-                    )
+                    gr.Button("全選択", size="sm").click(fn=lambda: toggle_all_categories("all", shown_columns_dict), outputs=checkboxes)
+                    gr.Button("全解除", size="sm").click(fn=lambda: toggle_all_categories("none", shown_columns_dict), outputs=checkboxes)
+                    gr.Button("AVGのみ", size="sm").click(fn=lambda: toggle_all_categories("avg_only", shown_columns_dict), outputs=checkboxes)
                 # with gr.Row(height=110):
                 #     shown_column = gr.CheckboxGroup(
                 #         show_label=False,