Update app.py

app.py

@@ -109,19 +109,6 @@ CATEGORY_MAPPING = {
     "Cinematography": ["camera_movement", "camera_angle", "shot_size"]
 }
 
-def save_class_accuracy_to_csv(model_name, class_accuracy):
-    """ 将模型的 class 正确率追加到 CSV，按固定顺序存储 """
-    data = {"Model Name": model_name}
-    for cls in CLASS_LIST:
-        data[cls] = class_accuracy.get(cls, 0)  # 确保所有列都有数据
-
-    df = pd.DataFrame([data])
-
-    # 如果 CSV 文件不存在，则创建并写入表头，否则追加数据
-    if not os.path.exists(CSV_FILE):
-        df.to_csv(CSV_FILE, index=False)
-    else:
-        df.to_csv(CSV_FILE, mode='a', header=False, index=False)  # 追加模式 
 
 def load_id_answer_mapping():
     id_answer_mapping = os.getenv("ID_ANSWER_MAPPING") 
@@ -160,7 +147,7 @@ def evaluate_uploaded_json(user_file, model_name):
             class_correct[question_class] += 1
             correct += 1
 
-    # 计算每个 class 的正确率
+    # 计算每个二级子类的正确率
     class_accuracy = {
         cls: class_correct[cls] / class_total[cls] if class_total[cls] > 0 else 0
         for cls in CLASS_LIST[:-5]  # 只计算二级子类，最后5个是一级分类
@@ -177,10 +164,13 @@ def evaluate_uploaded_json(user_file, model_name):
     # **保存 class 正确率到 CSV**
     save_class_accuracy_to_csv(model_name, class_accuracy)
 
-    # 格式化输出
+    # **修复 NameError 并避免 KeyError**
     class_accuracy_str = "\n".join(
-        [f"- {cls}: {acc:.2%} ({class_correct[cls]}/{class_total[cls]} correct)"
-         for cls in CLASS_LIST]
+        [
+            f"- {cls}: {class_accuracy[cls]:.2%} "
+            f"({class_correct.get(cls, 0)}/{class_total.get(cls, 0)} correct)"
+            for cls in CLASS_LIST
+        ]
     )
 
     return (
@@ -189,6 +179,20 @@ def evaluate_uploaded_json(user_file, model_name):
         f"Class-wise Accuracy:\n{class_accuracy_str}"
     )
 
+def save_class_accuracy_to_csv(model_name, class_accuracy):
+    """ 将模型的 class 正确率追加到 CSV，按固定顺序存储 """
+    data = {"Model Name": model_name}
+    for cls in CLASS_LIST:
+        data[cls] = class_accuracy.get(cls, 0)  # 确保所有列都有数据
+
+    df = pd.DataFrame([data])
+
+    # 如果 CSV 文件不存在，则创建并写入表头，否则追加数据
+    if not os.path.exists(CSV_FILE):
+        df.to_csv(CSV_FILE, index=False)
+    else:
+        df.to_csv(CSV_FILE, mode='a', header=False, index=False)  # 追加模式
+
 
 
 demo = gr.Blocks(css=custom_css)