import spacy
import streamlit as st
import spacy_streamlit
from config import COLORS_NER

nlp = spacy.load("ner_model")

st.title("SpaCy NER Model")
st.header("Entidades en el texto de un curriculum-cv")

cv_text=st.text_input("Ingrese texto de cv aquí","Ivan Ducales, hablo ingles B2 y tengo conocimientos en Java y Python, web development, postgres. he trabajado como web developer y back-end developer")

if cv_text is not None:
   doc= nlp(cv_text)
   spacy_streamlit.visualize_ner(doc,labels=nlp.get_pipe('ner').labels,colors = COLORS_NER)