from paddleocr import PaddleOCR
from flask import request, Flask
from wsgiref.simple_server import make_server
import time
import base64
import json


class OpenOcr:
    _ocr = None

    def __int__(self):
        self._ocr = PaddleOCR(use_angle_cls=True, lang="ch", use_gpu=False
                              , rec_model_dir='models/', det_model_dir='models/')
                # , rec_model_dir='./models/ch_ppocr_server_v2.0_rec_infer/'
                # , cls_model_dir='./models/ch_ppocr_mobile_v2.0_cls_infer/'
                # , det_model_dir='./models/ch_ppocr_server_v2.0_det_infer/')
        print('ocr create success!')

    def start_up(self):
        self._ocr = PaddleOCR(use_angle_cls=True, lang="ch", use_gpu=False)
        print('start_up: _ocr created!')

    def recognize(self, image_path):
        text_block_list = []
        result = self._ocr.ocr(image_path, cls=True)
        for item in result:
            data = {"name": item[1][0], "x1": item[0][0][0], "y1": item[0][0][1], "x2": item[0][1][0], "y2": item[0][1][1], "score": item[1][1]}
            text_block_list.append(data)
            # print(item[1][0])
            # print(item[0][0])
            # print(item[0][1])
            # print(item[1][1])
        return text_block_list


# 1.启动OCR模型
openOcr = OpenOcr()
openOcr.start_up()

# 2.启动web服务
app = Flask(__name__)
# app.config['PORT'] = 8801
# save_path = '/opt/soft/fire_images/'
save_path = 'd:/dev_tools/ai/'


@app.route('/')
def hello_world():
    return 'Hello World!'


@app.route("/ai/ocr/table", methods=['post'])
def request_ocr_table():
    start_time = time.time()
    data = json.loads(request.get_data(as_text=True))
    if 'img' not in data or 'id' not in data:
        return '参数不正确'
    request_id = data['id']
    img_base64 = data['img']
    if not img_base64 or img_base64 == '':
        return 'file is needed!'
    try:
        # print('')
        imgdata = base64.b64decode(img_base64)
        source_file_id = save_path + 'ocr_' + request_id + '.jpg'
        file = open(source_file_id, 'wb')
        file.write(imgdata)
        file.close()
        # detect_img(source_file_id, score=score, id=requestId)
        return openOcr.recognize(image_path=source_file_id)

    except Exception as e:
        print(e)
        return 'error:' + source_file_id
    # return 'success'


if __name__ == '__main__':
    # openOcr = OpenOcr()
    # openOcr.start_up()
    # openOcr.recognize(image_path='../doc/imgs/00015504.jpg')
    # openOcr.recognize(image_path='D:/dev_tools/ai/demo_table_02.jpg')
    # app.run(host='127.0.0.1', port=8801)
    server = make_server('127.0.0.1', 8801, app)
    server.serve_forever()