Update app.py

app.py

@@ -286,7 +286,7 @@ with gr.Blocks() as demo_leaderboard:
     with gr.Row():
         shown_columns_dict = {}
         checkboxes = []
-        with gr.Row():  # 制御ボタン用のRowをアコーディオンの前に配置
+        with gr.Row():  
             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)
@@ -445,33 +445,38 @@ with gr.Blocks() as demo_leaderboard:
 def toggle_all_categories(action: str, shown_columns_dict: dict):
     """全カテゴリーのチェックボックスを一括制御する関数"""
     results = []
-    for task_type in TaskType:
+    for task_type in TaskType:  # NotTaskも含めて処理
         if task_type == TaskType.NotTask:
-            continue
-            
-        if action == "all":
-            # 全選択: そのカテゴリーの全ての選択肢を返す
+            # Model detailsカテゴリーの処理
             results.append([
                 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
-            ])
-        elif action == "none":
-            # 全解除: 空のリストを返す
-            results.append([])
-        elif action == "avg_only":
-            # AVGのみ: AVGに関連する選択肢のみを返す
-            results.append([
-                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)
-                )
-            ])
-    return results  # リストとして全ての結果を返す
+            ] if action == "all" else [])
+        else:
+            if action == "all":
+                # 全選択
+                results.append([
+                    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
+                ])
+            elif action == "none":
+                # 全解除
+                results.append([])
+            elif action == "avg_only":
+                # AVGのみ
+                results.append([
+                    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)
+                    )
+                ])
+    return results
 
 # Submission demo