from paddleocr import PaddleOCR, draw_ocr
import numpy as np
from flask import Flask, request, jsonify, render_template
from PIL import Image
import requests
import base64
import io
import cv2
import time
import json

# # Paddleocr目前支持的多语言语种可以通过修改lang参数进行切换
# # 例如`ch`, `en`, `fr`, `german`, `korean`, `japan`
# ocr = PaddleOCR(use_angle_cls=True, lang="ch", ocr_version='PP-OCRv4')  # need to run only once to download and load model into memory
# # img_path = './imgs/11.jpg'
# img_path = './id_cards/card0.png'
# result = ocr.ocr(img_path, cls=True)
# print(result)
# # for idx in range(len(result)):
# #     res = result[idx]
# #     for line in res:
# #         print(line)
#
# # 显示结果
# result = result[0]
# print(len(result))
# for line in result:
#     print(line[1][0])
# image = Image.open(img_path).convert('RGB')
# boxes = [line[0] for line in result]
# txts = [line[1][0] for line in result]
# scores = [line[1][1] for line in result]
# im_show = draw_ocr(image, boxes, txts, scores, font_path='./fonts/simfang.ttf')
# im_show = Image.fromarray(im_show)
# im_show.save('result.jpg')

app = Flask(__name__)
ocr = PaddleOCR(use_angle_cls=True, lang="ch", ocr_version='PP-OCRv4')

@app.route('/')
def hello_world():  # put application's code here
    # return 'Hello World!'
    return render_template('index.html')


@app.route('/processing', methods=['POST'])
def process():
    file = request.files['image']  # 获取图像

    img_pil = Image.open(file.stream)  # PIL读取图流
    img_cv = cv2.cvtColor(np.array(img_pil), cv2.COLOR_RGB2BGR)  # PIL->cv2
    result = ocr.ocr(img=img_cv, cls=True)
    txts = [line[1][0] for line in result[0]]
    json_demo = {'result': txts, 'len': len(result[0])}

    # 返回，需要看实际情况返回字段
    return json_demo  # 这个是把图片展示在前端

# @app.route('/show')
# def show_image():
#     r = requests.get('https://s.cn.bing.net/th?id=OHR.CliffsEtretat_ZH-CN9911283373_UHD.jpg')
#     # image = base64.b64encode(r.content).decode('ascii')
#     image = base64.b64encode(r.content).decode()
#     return render_template('show_image.html', img_stream=image)


if __name__ == '__main__':
    app.run(host='127.0.1.3', port=5555)