fix: refact logic
This commit is contained in:
@@ -28,24 +28,15 @@ async def process_image_ocr(
|
||||
- Otherwise: use PaddleOCR-VL with formula prompt
|
||||
4. Convert output to LaTeX, Markdown, and MathML formats
|
||||
"""
|
||||
try:
|
||||
# 1. Load and preprocess image
|
||||
image = image_processor.preprocess(
|
||||
image_url=request.image_url,
|
||||
image_base64=request.image_base64,
|
||||
)
|
||||
except ValueError as e:
|
||||
raise HTTPException(status_code=400, detail=str(e))
|
||||
|
||||
try:
|
||||
# 2. Detect layout
|
||||
layout_info = layout_detector.detect(image)
|
||||
except RuntimeError as e:
|
||||
raise HTTPException(status_code=500, detail=f"Layout detection failed: {e}")
|
||||
image = image_processor.preprocess(
|
||||
image_url=request.image_url,
|
||||
image_base64=request.image_base64,
|
||||
)
|
||||
|
||||
try:
|
||||
# 3. Perform OCR based on layout
|
||||
ocr_result = ocr_service.recognize(image, layout_info)
|
||||
ocr_result = ocr_service.recognize(image)
|
||||
except RuntimeError as e:
|
||||
raise HTTPException(status_code=503, detail=str(e))
|
||||
|
||||
@@ -54,6 +45,4 @@ async def process_image_ocr(
|
||||
latex=ocr_result.get("latex", ""),
|
||||
markdown=ocr_result.get("markdown", ""),
|
||||
mathml=ocr_result.get("mathml", ""),
|
||||
layout_info=layout_info,
|
||||
recognition_mode=ocr_result.get("recognition_mode", ""),
|
||||
)
|
||||
|
||||
Reference in New Issue
Block a user