Update app.py

app.py

@@ -473,12 +473,37 @@ with block:
     gr.Markdown(
         LEADERBORAD_INTRODUCTION
     )
-    with gr.Tabs(elem_classes="tab-buttons") as tabs:
+    with gr.Tabs(elem_classes="tab-buttons", id=1) as tabs:
         # Table 0
-        df = pd.read_csv("https://huggingface.co/spaces/V-STaR-Bench/V-STaR-LeaderBoard/resolve/main/results.csv")
-        all_columns = df.columns.tolist()[1:]
+        df_raw = pd.read_csv(
+            "https://huggingface.co/spaces/V-STaR-Bench/V-STaR-LeaderBoard/resolve/main/results.csv", 
+            header=[0, 1]  # 告诉 pandas 前两行为表头
+        )
+        
+        # 2) 将 MultiIndex 列名转换为单层列名，例如 "Animals-mAM"
+        new_columns = []
+        for col_tuple in df_raw.columns:
+            # col_tuple 是形如 ("Animals", "mAM") 或 ("Model", nan) 的二元元组
+            domain = str(col_tuple[0]).strip() if pd.notnull(col_tuple[0]) else ""
+            metric = str(col_tuple[1]).strip() if pd.notnull(col_tuple[1]) else ""
+            if domain and metric:
+                new_columns.append(f"{domain}-{metric}")
+            else:
+                # 如果某一层为空，就只使用非空的那层
+                new_columns.append(domain or metric)
+        
+        df_raw.columns = new_columns
+        
+        # 如果第一列是 "Model-" 这种情况，可以进行一下修正
+        if df_raw.columns[0].endswith("-"):
+            df_raw.rename(columns={df_raw.columns[0]: "Model"}, inplace=True)
+        
+        # 3) 用前面处理过的列来构建 checkbox 选项
+        #    假设第一列 "Model" 不需要放到 checkbox 里
+        all_columns = df_raw.columns.tolist()[1:]
         choices_from_csv = [col.strip() for col in all_columns if col.strip()]
 
+
         with gr.Tabs(elem_classes="tab-buttons") as tabs:
             with gr.TabItem("📊 V-STaR"):
                 with gr.Row():
@@ -488,33 +513,33 @@ with block:
                             label=CITATION_BUTTON_LABEL,
                             lines=14,
                         )
+        
                 gr.Markdown(TABLE_INTRODUCTION)
                 
-                # 添加一个 CheckboxGroup，用 CSV 中的类别作为选项
+                # 复选框
                 checkbox_group = gr.CheckboxGroup(
                     choices=choices_from_csv,
-                    value=choices_from_csv,  # 默认全部选中，可根据需要调整
+                    value=choices_from_csv,  # 默认全选
                     label="Evaluation Dimension",
                     interactive=True,
                 )
                 
-                # 将 checkbox_group 放在一行中展示（可选）
                 with gr.Row():
                     checkbox_group
-                
-                # 显示 CSV 数据的 Dataframe 组件
-                data_component = gr.components.Dataframe(
-                    value=df, 
-                    type="pandas", 
+        
+                # 显示 DataFrame
+                data_component = gr.Dataframe(
+                    value=df_raw,
+                    type="pandas",
                     interactive=False,
                     visible=True,
                 )
             
         # table info
-        with gr.TabItem("📝 About", elem_id="mvbench-tab-table", id=5):
+        with gr.TabItem("📝 About", elem_id="mvbench-tab-table", id=3):
             gr.Markdown(LEADERBORAD_INFO, elem_classes="markdown-text")
         
-        with gr.TabItem("🚀 Submit here! ", elem_id="mvbench-i2v-tab-table", id=7):
+        with gr.TabItem("🚀 Submit here! ", elem_id="mvbench-i2v-tab-table", id=5):
             gr.Markdown(LEADERBORAD_INTRODUCTION, elem_classes="markdown-text")
 
             with gr.Row():