Update app.py

This commit is contained in:
2026-03-10 16:59:53 +08:00
parent 252bc49307
commit ee38f71f40

9
app.py
View File

@@ -41,16 +41,19 @@ async def predict(file: UploadFile = File(...)):
image = Image.open(io.BytesIO(image_bytes)).convert("RGB")
except Exception:
raise HTTPException(status_code=400, detail="无法识别的图像文件,请上传有效的图像")
logger.info(f"处理图像: {file.filename}, 尺寸: {image.size}")
logger.info(f"{datetime.now()} - 处理图像: {file.filename}, 尺寸: {image.size}")
result = ocr_engine.sample_inference(image)
if isinstance(result, dict) and "error" in result:
logger.error(f"{datetime.now()} - 推理返回错误: {result}")
raise HTTPException(status_code=500, detail=result)
return result
except HTTPException:
# 直接抛出 HTTP 异常,避免被通用异常捕获导致状态码错误
raise
except Exception as e:
logger.exception("推理过程中发生错误")
logger.exception(f"{datetime.now()} - 推理过程中发生错误: {e}")
raise HTTPException(status_code=500, detail=str(e))
@app.get("/health")