from flask import Flask, jsonify, request
from util import decode_image, handle_img

app = Flask(__name__)


@app.route("/img_receive", methods=["POST"])
def img_receive():
    try:
        body_json = request.get_json()
        print(f"接收到参数:{body_json}")
        images = []
        for index in range(0, 10):
            img = body_json[f'image_{index}']
            if not img is None:
                images.append(decode_image(img, False, None, None))
        # 开始处理
        bol = handle_img(images)
        return return_body(True, "成功", bol)
    except Exception as e:
        print(f"接受处理图片失败：{e}")
        return return_body(False, "接受处理图片失败", None)


def return_body(success, message, data):
    result = {"success": success, "message": message, "data": data}
    return jsonify(result)


if __name__ == "__main__":
    app.config['JSON_AS_ASCII'] = False
    app.run(host="0.0.0.0", port=5001, debug=True)