import streamlit as st
from transformers import pipeline, PegasusForConditionalGeneration, PegasusTokenizer
from fill_in_summary import FillInSummary
from paraphrase import PegasusParaphraser

def paraphrase(text):
    return text


st.title('Question Generator by Eddevs')

select = st.selectbox('Type',  ['Question Generator', 'Paraphrasing', 'Summarization', 'Fill in the blank'])

if select == "Summarization":
    with st.form("summarization"):
        # left_column, right_column = st.columns(2)
        # left_column.selectbox('Type',  ['Question Generator', 'Paraphrasing'])
        #st.selectbox('Model',  ['T5', 'GPT Neo-X'])

        text_input = st.text_area("Input Text")

        submitted = st.form_submit_button("Generate")

        if submitted:
            with st.spinner('Wait for it...'):
                result = FillInSummary().summarize(text_input)
            st.write(text_input)


if select == "Fill in the blank":
    with st.form("summarization"):
        text_input = st.text_area("Input Text")

        submitted = st.form_submit_button("Generate")

        if submitted:
            with st.spinner('Wait for it...'):
                fill = FillInSummary()
                result = fill.summarize(text_input)
                result = fill.blank_ne_out(result)
            st.write(result)


if select == "Paraphrasing":
    with st.form("paraphrasing"):
        # st.selectbox('Model',  ['T5', 'GPT Neo-X'])

        text_input = st.text_area("Input Text")

        submitted = st.form_submit_button("Generate")

        if submitted:
            with st.spinner('Wait for it...'):
                paraphrase_model = PegasusParaphraser()
                result = paraphrase_model.paraphrase(text_input)
            st.write(result)





#if st.button('Generate'):
    # st.write(input)
    #st.success("We have generated 105 Questions for you")
    # st.snow()
##else:
    ##nothing here