Update Summarization_Simple_25Nov.py

Summarization_Simple_25Nov.py

@@ -5,6 +5,11 @@ from math import ceil
 from collections import Counter
 from string import punctuation
 
+import spacy
+from spacy import displacy
+import en_ner_bc5cdr_md
+
+nlp = spacy.load("en_ner_bc5cdr_md")
 #nlp = en_core_web_lg.load()
 
 st.set_page_config(layout='wide')
@@ -85,31 +90,17 @@ def run_model(input_text):
         st.write('Summary')
         st.success(output)
         
-if st.button('Submit'):
+if st.button('Summarize'):
     run_model(runtext)
     
     sentences=runtext.split('.')
-def visualize(title, sentence_list, best_sentences):
-  text = ''
-
-  #display(HTML(f'<h1>Summary - {title}</h1>'))
-  for sentence in sentence_list:
-    if sentence in best_sentences:
-      #text += ' ' + str(sentence).replace(sentence, f"<mark>{sentence}</mark>")
-      text += ' ' + str(sentence).replace(sentence, f"<span class='highlight yellow'>{sentence}</span>")
-    else:
-      text += ' ' + sentence
-      display(HTML(f""" {text} """))
-
-    output = ''   
-    best_sentences = []
-    for sentence in output:
-       #print(sentence)
-       best_sentences.append(str(sentence))
-    return text     
-
-    t = "<div>Hello there my <span class='highlight blue'>name <span class='bold'>yo</span> </span> is <span class='highlight red'>Fanilo <span class='bold'>Name</span></span></div>"
-
-    st.write("<div>Hello there my <span class='highlight blue'>name <span class='bold'>yo</span> </span> is <span class='highlight red'>Fanilo <span class='bold'>Name</span></span></div>")  
+    
   
-    st.text_area('Reference text', str(reference_text))
+    st.text_area('Reference text', str(reference_text))
+
+if st.button('NER'):
+   doc = nlp(str(original_text2))
+   colors = { "DISEASE": "pink","CHEMICAL": "orange"}
+   options = {"ents": [ "DISEASE", "CHEMICAL"],"colors": colors}
+   ent_html = displacy.render(doc, style="ent", options=options)
+   st.markdown(ent_html, unsafe_allow_html=True)