from flask import Flask, request, render_template
import transformers
from parrot import Parrot
import torch
parrot = Parrot(model_tag="prithivida/parrot_paraphraser_on_T5")

#from flask import Flask, request, render_template
#import parrot
#import transformers
import time

app = Flask(__name__)

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

@app.route("/paraphrase", methods=["POST"])
def paraphrase():
    text = request.form["text"]
    model = request.form["model"]
    
    if model == "parrot":
        start_time = time.time()
        paraphrased_text = parrot.augment(input_phrase=text, use_gpu=False)
        processing_time = time.time() - start_time
        
    elif model == "transformers":
        start_time = time.time()
        paraphrased_text = transformers.paraphrase(text)
        processing_time = time.time() - start_time
    
    processing_percentage = 100 * processing_time / 30
    
    return render_template("paraphrase.html", original_text=text, paraphrased_text=paraphrased_text, processing_percentage=processing_percentage)

if __name__ == "__main__":
    app.run(host="0.0.0.0",port=7860)