from flask import Flask, render_template, request, jsonify
from corssencode_inference import get_range_answers, get_best_answer

app = Flask(__name__)


@app.route('/', methods=['GET', 'POST'])
def index():
    if request.method == 'POST':
        if request.form.get('get_answer') == 'One answer':
            one_answer = get_best_answer(request.form.get('context'), request.form.get('question'))

            return render_template('index.html', answer=one_answer)

            # return jsonify(
            #     {
            #         "response_code": "200",
            #         "request": f"{request.form.get('context')} [Cont_token] {request.form.get('question')}",
            #         "response": one_answer
            #     }
            # )
        elif request.form.get('get_answer_corpus') == 'Five answer':
            many_answer = get_range_answers(request.form.get('context'), request.form.get('question'))

            return render_template('index.html', answer=many_answer)

            # return jsonify(
            #     {
            #         "response_code": "200",
            #         "request": f"{request.form.get('context')} [Cont_token] {request.form.get('question')}",
            #         "response": many_answer
            #     }
            # )
    elif request.method == 'GET':
        return render_template('index.html')


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