fix: add error handling and logging to OCR endpoint
All checks were successful
CI / nodejs-lint (push) Has been skipped
CI / go-lint (push) Has been skipped
CI / python-lint (push) Has been skipped
CI / test-go-consent (push) Successful in 31s
CI / test-python-voice (push) Successful in 32s
CI / test-bqas (push) Successful in 33s
CI / Deploy (push) Successful in 2s
All checks were successful
CI / nodejs-lint (push) Has been skipped
CI / go-lint (push) Has been skipped
CI / python-lint (push) Has been skipped
CI / test-go-consent (push) Successful in 31s
CI / test-python-voice (push) Successful in 32s
CI / test-bqas (push) Successful in 33s
CI / Deploy (push) Successful in 2s
Return detailed error message instead of generic 500, and handle empty OCR results gracefully. Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
This commit is contained in:
@@ -88,10 +88,17 @@ async def ocr(
|
|||||||
img = Image.open(io.BytesIO(img_bytes)).convert("RGB")
|
img = Image.open(io.BytesIO(img_bytes)).convert("RGB")
|
||||||
img_np = np.array(img)
|
img_np = np.array(img)
|
||||||
|
|
||||||
|
try:
|
||||||
result = _engine.ocr(img_np)
|
result = _engine.ocr(img_np)
|
||||||
|
except Exception as e:
|
||||||
|
logger.error(f"OCR failed: {e}", exc_info=True)
|
||||||
|
raise HTTPException(status_code=500, detail=f"OCR processing failed: {e}")
|
||||||
|
|
||||||
|
if not result or not result[0]:
|
||||||
|
return {"words": [], "image_width": img_np.shape[1], "image_height": img_np.shape[0]}
|
||||||
|
|
||||||
words = []
|
words = []
|
||||||
for line in result[0] or []:
|
for line in result[0]:
|
||||||
box, (text, conf) = line[0], line[1]
|
box, (text, conf) = line[0], line[1]
|
||||||
x_min = min(p[0] for p in box)
|
x_min = min(p[0] for p in box)
|
||||||
y_min = min(p[1] for p in box)
|
y_min = min(p[1] for p in box)
|
||||||
|
|||||||
Reference in New Issue
Block a user