from flask import Flask, render_template, request, jsonify
# from flask_cors import CORS, cross_origin
from modules.model import summarize
import __main__

app = Flask(__name__)
# shortTokenizer = BartTokenizer.from_pretrained('sshleifer/distilbart-xsum-12-6')
# shortModel = BartForConditionalGeneration.from_pretrained('sshleifer/distilbart-xsum-12-6')

# longTokenizer = BartTokenizer.from_pretrained('sshleifer/distilbart-cnn-12-6')
# longModel = BartForConditionalGeneration.from_pretrained('sshleifer/distilbart-cnn-12-6')

@app.route("/")
def home():
    return render_template('index.html')


@app.route("/summarize")
def recommend():
    # Get form data
    # request_data = request.args.get("input").get_json()
    # input_text = request_data['input_text']
    input_text = request.args.get("input")
    print(input_text)

    # Call the function summarize to run the text summarization
    try:
        short_output_summary, long_output_summary = summarize(input_text)
        response = jsonify(message={'short': short_output_summary.strip(), 'long': long_output_summary.strip()})
        response.headers.add("Access-Control-Allow-Origin", "*")
        # Pass output summary to the output template
        return response

    except Exception as e:
        return render_template('index.html', query=e)

    pass

if __name__ == '__main__':
    print("Loading BART model and tokenzier . . .")
    app.run(host="0.0.0.0", port=7860)