import streamlit as st
from predicciones import etiquetar


st.title("Reconocimiento de entidades en prescripciones médicas")
abstract = """
Demostracion de un modelo de reconocimento de entidades en prescripciones médicas.\n
Proyecto para MDS Uchile en conjunto con el Ministerio de Salud de Chile.\n
Integrantes: Daniel Carmona, Camilo Carvajal, Martín Sepúlveda y Monserrat Prado\n
Para más información visitar nuestro repositorio: 
https://github.com/camilocarvajalreyes/entidades-minsal"""
st.subheader(abstract)

example = "PARACETAMOL 500 MG COMPRIMIDO 1 COMPRIMIDO ORAL cada 6 horas durante 3 dias"
text = st.text_input('Insertar prescripción', example)

# 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)

disclaimer = """Importante: las respuestas mostradas sólo son una demostracion.
 No deben usarse como información médica."""
st.text(disclaimer)