from utils import get_resources, simplify

import streamlit as st

st.set_page_config(
    page_title="Text Simplification in Dutch",
    page_icon="🏃"
)

st.title("🏃 Text Simplification in Dutch")

with st.form("input data"):
    text = st.text_area(label="Input text", value="Met het naderen van de zonovergoten middaghemel op deze betoverende dag, waarbij de atmosferische omstandigheden een onbelemmerde convergentie van cumulusbewolking en uitgestrekte stratosferische azuurblauwe wijdheid faciliteren, lijken de geaggregeerde weersverschijnselen van vandaag, die variëren van sporadische plensbuien tot kalme zuchtjes wind en zeldzame opvlammingen van bliksem, de delicate balans tussen meteorologische complexiteit en eenvoud te weerspiegelen, waardoor de gepassioneerde observator met een gevoel van ontzag en verwondering wordt vervuld.")
    submitted = st.form_submit_button("Submit")

error_ct = st.empty()
if submitted:
    text = text.strip()
    if not text:
        error_ct.error("Text cannot be empty!", icon="⚠️")
    else:
        error_ct.info("Generating abstract meaning representation (AMR)...", icon="💻")

        model, tokenizer, streamer = get_resources()
        error_ct.empty()

        for stream_simplification in simplify(text, model, tokenizer, streamer):
            st.write(stream_simplification)


########################
# Information, socials #
########################
st.header("Project background")

st.markdown("""""")


st.header("Contact ✒️")

st.markdown("Would you like  additional functionality in the demo, do you have questions, or just want to get in touch?"
            " Give me a shout on [Twitter](https://twitter.com/BramVanroy)"
            " or add me on [LinkedIn](https://www.linkedin.com/in/bramvanroy/)!")