Update app.py

app.py

@@ -112,6 +112,24 @@ runtext =st.text_area('Input Clinical Note here:', str(original_text2), height=3
 
 reference_text = original_text['Reference_text'].values
 
+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     
 
 
 #===== Pysummarization =====
@@ -327,4 +345,6 @@ if st.button('Submit'):
     
     st.markdown('<span style="background-color: #FFFF00">testing</span> if this **works**', unsafe_allow_html=True)
     st.text_area('Reference text', str(reference_text))
+
+    st.markdown(visualize(runtext,reference_text),unsafe_allow_html=True)