'''
Author: error: error: git config user.name & please set dead value or install git && error: git config user.email & please set dead value or install git & please set dead value or install git
Date: 2023-03-29 20:22:34
LastEditors: error: error: git config user.name & please set dead value or install git && error: git config user.email & please set dead value or install git & please set dead value or install git
LastEditTime: 2023-04-05 15:27:13
FilePath: /luyuetong-data/yolox-pytorch/flask_on.py
Description: 这是默认设置,请设置`customMade`, 打开koroFileHeader查看配置 进行设置: https://githaub.com/OBKoro1/koro1FileHeader/wiki/%E9%85%8D%E7%BD%AE
'''
import os

from flask import Flask, request, jsonify
from gevent import pywsgi

from predict import PREDICT

app = Flask(__name__)
# 设置未处理文件路径以及处理保存路径
# dataPath = '/home/DiatomRecognition/upload'
# resultPath = '/home/DiatomRecognition/result'
# localhost 测试
dataPath = 'E:/Document/project/temp/data'
resultPath = r'E:\Document\project\temp/save'


@app.route('/predict', methods=['POST'])
def predict_one():
    if not os.path.exists(resultPath):
        # 如果不存在，创建文件夹及其所有父文件夹
        os.makedirs(resultPath)
    # 从请求参数中提取文件名
    fileName = request.form['fileName']
    # 将文件路径传递给模型进行预测
    result = PREDICT(dataPath, resultPath, fileName)  # predicted_class，score如果需要再处理

    return result


if __name__ == '__main__':
    # app.debug = False
    server = pywsgi.WSGIServer(('0.0.0.0', 2000), app)
    app.config["JSON_AS_ASCII"] = False
    server.serve_forever()