Function ocr_recognition

ddddocr/api/routes.py:80–118

执行OCR识别

(request: OCRRequest)

    
    @app.post("/ocr", response_model=APIResponse)
    async def ocr_recognition(request: OCRRequest):
        """执行OCR识别"""
        try:
            if not service.ocr_instance:
                raise HTTPException(status_code=400, detail="OCR功能未初始化，请先调用 /initialize 接口")
            
            if "ocr" not in service.enabled_features:
                raise HTTPException(status_code=400, detail="OCR功能已禁用")
            
            # 解码base64图片
            try:
                image_data = base64.b64decode(request.image)
            except Exception:
                raise HTTPException(status_code=400, detail="图片base64解码失败")
            
            # 设置字符集范围
            if request.charset_range is not None:
                service.ocr_instance.set_ranges(request.charset_range)
            
            # 执行OCR识别
            result = service.ocr_instance.classification(
                image_data,
                png_fix=request.png_fix,
                probability=request.probability,
                color_filter_colors=request.color_filter_colors,
                color_filter_custom_ranges=request.color_filter_custom_ranges
            )
            
            if request.probability:
                response_data = OCRResponse(text=None, probability=result)
            else:
                response_data = OCRResponse(text=result, probability=None)
            
            return APIResponse(success=True, message="OCR识别成功", data=response_data.dict())
            
        except HTTPException:
            raise
        except Exception as e:
            return APIResponse(success=False, message=f"OCR识别失败: {str(e)}")
    
    @app.post("/detect", response_model=APIResponse)
    async def object_detection(request: DetectionRequest):