from flask import Flask, jsonify, request
from waitress import serve
import logging

logger = logging.getLogger('waitress')
logger.setLevel(logging.INFO)


app = Flask(__name__)


@app.route('/', methods=["GET"])
def home():
    logger.info("serving home")
    return jsonify(
        app_name="test",
    )
  
  

@app.route('/tagger', methods=["POST"])
def get_tagger():
    logger.info("serving tagger")
    req_data = request.get_json()
    if req_data is None:
        return jsonify({"text": ""})
    return jsonify(predict(model, req_data['text']))
  

if __name__ == '__main__':
    serve(app, host='0.0.0.0', port=5000)