Update app.py

app.py

@@ -252,12 +252,20 @@ def highlight_entities(article_content,summary_output):
     markdown_end = "</mark>"
 
     matched_entities, unmatched_entities = get_and_compare_entities(article_content,summary_output)
+    
+    print(matched_entities)
+    print(unmatched_entities)
+    print("")
+    print(summary_output)
 
     for entity in matched_entities:
         summary_output = summary_output.replace(entity, markdown_start_green + entity + markdown_end)
 
     for entity in unmatched_entities:
         summary_output = summary_output.replace(entity, markdown_start_red + entity + markdown_end)
+    
+    print("")
+    print(summary_output)
     soup = BeautifulSoup(summary_output, features="html.parser")
 
     return HTML_WRAPPER.format(soup)
@@ -454,8 +462,6 @@ if summarize:
             summarized_text = ' '.join([summ['summary_text'] for summ in summarized_text])
     
     with st.spinner("Calculating and matching entities, this takes a few seconds..."):
-        print(text_to_summarize)
-        print(summarized_text)
         entity_match_html = highlight_entities(text_to_summarize,summarized_text)
         st.subheader("Summarized text with matched entities in Green and mismatched entities in Red relative to the Original Text")
         st.markdown("####")