import streamlit as st
import generator
from generator import *
import string

st.info("Enter short paragraphs, please, to get better answers.\n")
text_ar = st.text_area("Enter text:")
col1, col2, col3 = st.columns([2,1,2])

if(col2.button("Generate!")):
    text_ar = text_ar.replace("\n", " ")
    text_ar = text_ar.split('.')
    for text in text_ar :
        if text.isspace() or len(text) == 0:
            continue
        pairs = creator(text)
        for pair in pairs:
            st.text(pair)
    st.balloons()