import logging.config

from flask import Flask, request
from paddleocr import PaddleOCR

from log import LOGGING_CONFIG
from utils import process_request

app = Flask(__name__)
# 如果不希望识别出空格，可以设置use_space_char=False。做此项设置一定要测试，2.7.3版本此项设置有bug，会导致识别失败
OCR = PaddleOCR(use_angle_cls=False, show_log=False, det_db_thresh=0.1, det_db_box_thresh=0.3, det_limit_side_len=1248,
                drop_score=0.3)


@app.route('/', methods=['POST'])
@process_request
def main():
    img_path = request.form.get('img_path')
    return OCR.ocr(img_path, cls=False)


if __name__ == '__main__':
    logging.config.dictConfig(LOGGING_CONFIG)
    app.run('0.0.0.0', 5001)