from flask import Flask, jsonify, request
import pickle
import pandas as pd

from utils import predict


app = Flask(__name__)


@app.route('/')
def index():
    return "/model/text – predicts tag for string sample"


@app.route('/model/text', methods=['POST'])
def parse_string():
    sample = request.get_data()
    model = pickle.load(open('models/tags_model.pkl', 'rb'))
    labelencoder = pickle.load(open('models/tags_encoder.pkl', 'rb'))
    return predict(sample, model, labelencoder)


if __name__ == '__main__':
    app.run(debug=True)