from flask import Flask, request
from paddleclas import PaddleClas

from util.common_util import process_request

app = Flask(__name__)
CLAS = PaddleClas(model_name="text_image_orientation")


@app.route('/clas/orientation', methods=['POST'])
@process_request
def orientation():
    """
    判断图片旋转角度，逆时针旋转该角度后为正。可能值["0", "90", "180", "270"]
    :return: 最有可能的两个角度
    """
    img_path = request.form.get('img_path')
    clas_result = CLAS.predict(input_data=img_path)
    clas_result = next(clas_result)[0]
    if clas_result["scores"][0] < 0.5:
        return ['0', '90']
    return clas_result["label_names"]


if __name__ == '__main__':
    app.run('0.0.0.0', 5002)