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):
#     st.write(text.split()[i]+'\t',ent)

entidades = etiquetar(text,pandas=True)
st.dataframe(data=entidades)