import streamlit as st 
import spacy
from spacy import displacy
from spacy_streamlit import visualize_ner
import en_core_web_sm

NER = spacy.load("en_core_web_sm")
st.title('Entity Extraction')

def prediction(raw_text):
  text1= NER(raw_text)
  st.write("List wise NERs:")
  st.write("------------------")
  st.write(f"{'Text' : <10}{'NER' : >10}")
  for word in text1.ents:
    st.write(word.text,"\t\t",word.label_)
  print()
  st.write("------------------")
  st.write("NERs in the sentence:")
  st.markdown(displacy.render(text1,style="ent"))
  visualize_ner(tex1, labels=nlp.get_pipe("ner").labels)

raw_text = """Ai-Khanoum (/aɪ ˈhɑːnjuːm/, meaning Lady Moon; Uzbek: Oyxonim) is the archaeological site of a Hellenistic city in Takhar Province, Afghanistan. 
The city, whose original name is unknown,[a] was probably founded by an early ruler of the Seleucid Empire and served as a military and economic centre for the rulers of the Greco-Bactrian Kingdom until its destruction c. 145 BC. 
Rediscovered in 1961, the ruins of the city were excavated by a French team of archaeologists until the outbreak of conflict in Afghanistan in the late 1970s. """
prediction(raw_text)