Update app.py

app.py

@@ -121,12 +121,28 @@ with col4:
         #nav_page('Social Notes')
         inputNote = "Input Social Note"
 
-
-#runtext =st.text_area('Input note here:', str(original_text2), height=300)
 runtext =st.text_area(inputNote, str(original_text2), height=300)
 
-    
-def run_model(input_text):
+def visualize (run_text,output)
+    text =''
+    import spacy
+    from spacy.lang.en import English # updated
+    nlp=spacy.load('en_core_web_sm')
+    sentences=run_text.split('.')
+    summary=output.split('.')
+
+    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>")
+        else:
+            text += ' ' + sentence
+        display(HTML(f""" {text} """))
+    best_sentences = []
+    for sentence in summary:
+       best_sentences.append(str(sentence))
 
     if model == "BertSummarizer":
         output = original_text['BertSummarizer'].values
@@ -165,7 +181,7 @@ with col1:
     st.button('Summarize')
     run_model(runtext)
     sentences=runtext.split('.')
-    #st.text_area('Reference text', str(reference_text))#,label_visibility="hidden")
+    st.text_area('Reference text', str(reference_text))#,label_visibility="hidden")
 with col2:
     st.button('NER')
     doc = nlp(str(original_text2))
@@ -177,7 +193,8 @@ with col2:
     
 col3, col4 = st.columns(2)
 with col3: 
-    st.text_area('Reference text', str(reference_text))#,label_visibility="hidden")
+    st.text_area(visualize (run_text,output))
+    
 with col4: 
     st.text_area('testing', str(reference_text))#,label_visibility="hidden")