trying to fix

app.py

@@ -39,9 +39,26 @@ def process_mc_df(df, shot=None):
     df = df.rename(columns={"name": "Model"})
     # 将zero_naive, zero_self_con, zero_cot, zero_cot_self_con, few_naive, few_self_con, few_cot, few_cot_self_con列重新组织成MultiIndex，一层为Zeroshot, Fewshot,一层为Naive, Self-Consistency, CoT, CoT+Self-Consistency
     df = df.set_index("Model")
+
+    columns = [col for col in df.columns if col != "Model"]
+
+    col_to_multi_index = {
+        "zero_naive": ("Zeroshot", "Naive"),
+        "zero_self_con": ("Zeroshot", "SC"),
+        "zero_cot": ("Zeroshot", "CoT"),
+        "zero_cot_self_con": ("Zeroshot", "CoT+SC"),
+        "few_naive": ("Fewshot", "Naive"),
+        "few_self_con": ("Fewshot", "SC"),
+        "few_cot": ("Fewshot", "CoT"),
+        "few_cot_self_con": ("Fewshot", "CoT+SC"),
+    }
+
+    columns = [col_to_multi_index[col] for col in df.columns]
+    
+
     # df = df.stack().unstack()
     try:
-        df.columns = pd.MultiIndex.from_tuples([("Zeroshot", "Naive"), ("Zeroshot", "SC"), ("Zeroshot", "CoT"), ("Zeroshot", "CoT+SC"), ("Fewshot", "Naive"), ("Fewshot", "SC"), ("Fewshot", "CoT"), ("Fewshot", "CoT+SC")])
+        df.columns = pd.MultiIndex.from_tuples(columns)
     except:
         print(df)
         raise