finetune header formats

app.py

@@ -82,7 +82,7 @@ with gr.Blocks() as block:
                 interactive=False,
                 elem_classes="custom-dataframe",
                 max_height=2400,
-                column_widths=["100px", "240px"] + ["160px"] * 3 + ["200px"] * (len(initial_headers) - 5),
+                column_widths=["100px", "240px"] + ["160px"] * 3 + ["210px"] * (len(initial_headers) - 5),
             )
 
             def update_table_and_caption(table_type, super_group, model_group):
@@ -99,7 +99,7 @@ with gr.Blocks() as block:
                         headers=headers,
                         datatype=["number", "html"] + ["number"] * (len(headers) - 2),
                         interactive=False,
-                        column_widths=["100px", "240px"] + ["160px"] * 3 + ["200px"] * (len(headers) - 5),
+                        column_widths=["100px", "240px"] + ["160px"] * 3 + ["210px"] * (len(headers) - 5),
                     ),
                     caption,
                     f"<style>{base_css}\n{table_css}</style>"

static/css/style.css

@@ -84,9 +84,9 @@
 /* Dimension columns */
 .custom-dataframe td:nth-child(n+6),
 .custom-dataframe th:nth-child(n+6) {
-    width: 200px !important;
+    width: 210px !important;
     min-width: 160px !important;
-    max-width: 200px !important;
+    max-width: 210px !important;
     text-align: center !important;
 }
 

utils.py

@@ -128,13 +128,13 @@ class MEGABenchEvalDataLoader:
         total_open_tasks = self.SUMMARY_DATA[sample_model]["open"]["num_eval_tasks"]
         total_tasks = total_core_tasks + total_open_tasks
 
-        # Define headers with task counts
+        # Define headers with task counts on new line using Unicode line break
         column_headers = {
             "Rank": "Rank",
             "Models": "Models",
-            "Overall": f"Overall({total_tasks})",
-            "Core": f"Core({total_core_tasks})",
-            "Open-ended": f"Open-ended({total_open_tasks})"
+            "Overall": f"Overall\n({total_tasks})",
+            "Core": f"Core\n({total_core_tasks})",
+            "Open-ended": f"Open-ended\n({total_open_tasks})"
         }
         
         # Add rank column to DataFrame
@@ -144,13 +144,20 @@ class MEGABenchEvalDataLoader:
         # Rename the columns in DataFrame to match headers
         df = df.rename(columns=column_headers)
         
+        # For dimension columns, add task counts on new line
+        dimension_headers = []
+        for display_name in self.SUPER_GROUPS[selected_super_group]:
+            task_count = display_name.split('(')[1].rstrip(')')
+            base_name = display_name.split('(')[0]
+            dimension_headers.append(f"{base_name}\n({task_count})")
+        
         headers = [
             column_headers["Rank"],
             column_headers["Models"],
             column_headers["Overall"],
             column_headers["Core"],
             column_headers["Open-ended"]
-        ] + self.SUPER_GROUPS[selected_super_group]
+        ] + dimension_headers
         
         data = df[[
             column_headers["Rank"],