import logging.config

from flask import Flask, request
from paddlenlp import Taskflow

from log import LOGGING_CONFIG
from utils import process_request

app = Flask(__name__)
schema = ['基本医保结算单', '出院记录', '费用清单']
CLAS = Taskflow('zero_shot_text_classification', model='utc-xbase', schema=schema,
                task_path='model/text_classification', precision='fp16')


@app.route('/', methods=['POST'])
@process_request
def main():
    text = request.form.get('text')
    cls_result = CLAS(text)
    cls_result = cls_result[0].get('predictions')[0]
    if cls_result['score'] < 0.8:
        raise Exception(f'识别结果置信度过低！text: {text}')
    return cls_result['label']


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