Update app.py

app.py

@@ -135,17 +135,59 @@ def evaluate_amount_predictions(y_true, y_pred, relaxed_tol=0.01):
 # CORD v2 GT
 # ============================================================
 def get_gt_total_from_cord_item(item):
+    """
+    從 CORD v2 的 ground_truth 中取出 total amount (total_price).
+    CORD v2 結構大致為:
+      {
+        "gt_parse": {
+          "menu": [...],
+          "sub_total": {"subtotal_price": "..."},
+          "total": {"total_price": "...", ...}
+        },
+        "meta": {...},
+        "valid_line": [...],
+        ...
+      }
+    """
     gt = item["ground_truth"]
+
+    # ground_truth 在 HF 裡是 JSON string，要先 parse 成 dict
     if not isinstance(gt, dict):
         try:
             gt = json.loads(gt)
         except Exception:
             return None
+
+    # 1) 優先走官方路徑 gt_parse.total.total_price
     try:
-        val = gt.get("total", {}).get("total_price")
+        val = (
+            gt.get("gt_parse", {})
+              .get("total", {})
+              .get("total_price", None)
+        )
+        if val is not None:
+            return val
     except Exception:
-        val = None
-    return val
+        pass
+
+    # 2) 保險：如果結構有變，遞迴搜尋任何名為 "total_price" 的欄位
+    def _find_total_price(node):
+        if isinstance(node, dict):
+            for k, v in node.items():
+                if k == "total_price" and isinstance(v, str):
+                    return v
+                found = _find_total_price(v)
+                if found is not None:
+                    return found
+        elif isinstance(node, list):
+            for item_ in node:
+                found = _find_total_price(item_)
+                if found is not None:
+                    return found
+        return None
+
+    return _find_total_price(gt)
+
 
 
 # ============================================================