# ===============================================
# =============== OCR 插件接口管理 ===============
# ===============================================

from umi_log import logger

ApiDict = {}
AllDict = {}


# TODO: 静态插件
# 由插件控制器调用，初始化OCR插件的接口。传入动态插件
def initOcrPlugins(plugins):
    global ApiDict, AllDict
    for p in plugins:
        ApiDict[p] = plugins[p]["api_class"]
        AllDict[p] = plugins[p]


# 生成一个ocr api实例，成功返回对象，失败返回 [Error] 开头的字符串
def getApiOcr(apiKey, argd):
    # 检测argd，恢复int类型
    for k in argd:
        n = argd[k]
        if isinstance(n, float):
            rounded = round(n)
            if abs(n - rounded) <= 1e-7:
                argd[k] = rounded
    if apiKey in ApiDict:
        try:
            return ApiDict[apiKey](argd)  # 实例化后返回
        except Exception as e:
            logger.error(f"生成api实例{apiKey}失败。", exc_info=True, stack_info=True)
            return f"[Error] Failed to generate API instance {apiKey}: {e}"
    return f'[Error] "{apiKey}" not in ApiDict.'


# 返回一个API的局部配置字典
def getLocalOptions(apiKey):
    if apiKey in AllDict:
        return AllDict[apiKey]["local_options"]
    return {}