import streamlit as st
from transformers import pipeline, set_seed

pipe = pipeline("text-generation", model="google/flan-t5-base")

st.title("Paraphrase Generator")
user_word = st.text_input("Enter a line:")

if st.button("Generate Paraphrase"):
    if user_word:
        paraphrase_prompt= f"Write a Paraphrase on '{user_word}'.\n"
        set_seed(42)
        paraphrase = pipe(paraphrase_prompt, max_length=100, do_sample=True, num_return_sequences=1)[0]["generated_text"]
        st.markdown("**Paraphrase:**")
        st.markdown(paraphrase)
    else:
        st.warning("Please enter a line.")