import streamlit as st

from pii import PII_IDENTIFIER


pii_id_instance = PII_IDENTIFIER()

st.title("Personal data ID")

text_to_predict = st.text_input('Enter text :: ', 'Mein Name ist MARK DEF1100 Noah Schmidt noahschmidt@debank.com und ich lebe in Irbid with number swift WWBADE3AXXX 6282368558 DE89370400440532013000 +491635551584. Allianz Versicherungs-Aktiengesellschaft Allianz ® Nachtrag vom 30.11.2014 zur')


if st.button("Mask Data"):
    print(text_to_predict)
    out = pii_id_instance.identify(text_to_predict)
    st.write(out['anonymized_text'])


### TO RUN :: streamlit run ui_app.py