from typing import TYPE_CHECKING

from configs.basic_config import logger

# if TYPE_CHECKING:
#     try:
#         from rapidocr_paddle import RapidOCR
#     except ImportError:
#         from rapidocr_onnxruntime import RapidOCR


def get_ocr(use_cuda: bool = True) -> "RapidOCR":
    try:
        from rapidocr_paddle import RapidOCR
        logger.info(f"from rapidocr_paddle import RapidOCR")
        ocr = RapidOCR(det_use_cuda=use_cuda, cls_use_cuda=use_cuda, rec_use_cuda=use_cuda)
    except ImportError:
        from rapidocr_onnxruntime import RapidOCR
        logger.info(f"from rapidocr_onnxruntime import RapidOCR")
        ocr = RapidOCR()
    return ocr