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 :: ', '')


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


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