import streamlit as st
from predicciones import etiquetar

example = "PARACETAMOL 500 MG COMPRIMIDO 1 COMPRIMIDO ORAL cada 6 horas durante 3 dias"

st.title("Reconocimiento de entidades en prescripciones médicas")

text = st.text_input('Insertar prescripción', example)
# st.write('Texto ingresado', title)

entidades = etiquetar(text,pandas=False)
for i, ent in enumerate(entidades):
    # print("{}\t{}".format(text.split()[i],ent))
    st.write(text.split()[i],ent)