diff --git a/flows/label_ocr.py b/flows/label_ocr.py index 5514006..ff8d92a 100644 --- a/flows/label_ocr.py +++ b/flows/label_ocr.py @@ -28,6 +28,9 @@ _OCR_FIELD_MAP = [ def _result_to_fields(r) -> dict[str, str | None]: if not r.ok: + err = (r.error or "").lower() + if "reject" in err or "not a recognized layout" in err: + return {"ocrLayout": "UNSUPPORTED"} return {"ocrLayout": "ERROR"} return {col: (getattr(r, attr) or "") for attr, col in _OCR_FIELD_MAP}