#---------------------AI Paraphraser - iFrame code --------------


import transformers
import gradio as gr
import torch

from transformers import AutoTokenizer, AutoModelForSeq2SeqLM

tokenizer = AutoTokenizer.from_pretrained("humarin/chatgpt_paraphraser_on_T5_base")
model = AutoModelForSeq2SeqLM.from_pretrained("humarin/chatgpt_paraphraser_on_T5_base")

def paraphrase(
    Content_to_Rephrase,
    num_beams: 3,
    num_return_sequences=3,
):
    input_ids = tokenizer(
        f'paraphrase: {Content_to_Rephrase}',
        return_tensors="pt", padding="longest",

    ).input_ids
    
    outputs = model.generate(input_ids, num_beams=num_beams, num_return_sequences=num_return_sequences)

    res = tokenizer.batch_decode(outputs, skip_special_tokens=True)
    res1 = res [0]
    res2 = res [1]
    res3 = res [2]
    
    
    return res1, res2, res3

output1 = gr.Textbox(label="Rephrased: Option 1")
output2 = gr.Textbox(label="Rephrased: Option 2")
output3 = gr.Textbox(label="Rephrased: Option 3")

input = gr.Textbox(lines=10)

iface = gr.Interface(fn=paraphrase, 
                     inputs=[input],
                     outputs=[output1, output2, output3],
                     title="Free AI Sentence Rephraser",
                     description="<ul><li>Paste text in the input box and press 'Submit'.</li><li>Max length: ~35 words (larger content is summarized)</li><li>The rephrased sentences *may not* be better than the original input.</li><li>Model 'humarin' pre-trained by ChatGPT. Temp = 0.7</li></ul>",
                     examples=[
                                ["With the humble is wisdom."],
                                ["Hatred stirs up strife."],
                                ["The way of a fool is right in his own eyes."],
                                ["Righteousness leads to life."],
                              ],
                     cache_examples=True,
                    )

iface.launch()