from fastapi import FastAPI
from pydantic import BaseModel
from util  import ocr_recognition
from configs.basic_config import *

app = FastAPI()

class IdentifyRecognitionParams(BaseModel):
    type: int
    recognitionFrontData: str = ""
    recognitionBackData: str = ""

@app.post("/recognition")
async def recognition(params: IdentifyRecognitionParams):
    logger.info(f"python recognition里的参数，{params.type}, {params.recognitionFrontData}, {params.recognitionBackData}")
    returnStr = ocr_recognition.OCRRecognition.extractIdCardInfo(params.type, params.recognitionFrontData, params.recognitionBackData)
    return returnStr


# 运行命令： uvicorn webApp:app --reload
# if __name__ == "__main__":
#     unicorn.run("webApp:app", port="8013", reload=True)