Update app.py

app.py

@@ -4,25 +4,35 @@ 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("/", methods=["GET", "POST"])
+@app.route("/")
 def index():
-    if request.method == "POST":
-        text = request.form["text"]
-        library = request.form["library"]
-
-        if library == "transformers":
-            model = transformers.AutoModelWithLMHead.from_pretrained("gpt2")
-            paraphrased_text = transformers.generate(model, text)
-        elif library == "parrot":
-            paraphrased_text = parrot.augment(input_phrase=text, use_gpu=False)
-        else:
-            return "Error: Invalid library selected"
-
-        return render_template("index.html", original_text=text, paraphrased_text=paraphrased_text)
-
     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)
+    app.run(host="0.0.0.0",port=7860)