Update app.py

app.py

@@ -200,15 +200,12 @@ def get_all_entities(text):
     
 def get_and_compare_entities(article_content,summary_output):
     
-    print(article_content)
-    print(summary_output)
     all_entities_per_sentence = get_all_entities_per_sentence(article_content)
     entities_article = list(itertools.chain.from_iterable(all_entities_per_sentence))
-    print(entities_article)
-
+   
     all_entities_per_sentence = get_all_entities_per_sentence(summary_output)
     entities_summary = list(itertools.chain.from_iterable(all_entities_per_sentence))
-    print(entities_summary)
+   
     matched_entities = []
     unmatched_entities = []
     for entity in entities_summary:
@@ -262,23 +259,10 @@ def highlight_entities(article_content,summary_output):
     for entity in unmatched_entities:
         summary_output = summary_output.replace(entity, markdown_start_red + entity + markdown_end)
     soup = BeautifulSoup(summary_output, features="html.parser")
-    print(soup)
-    print(matched_entities)
-    print(unmatched_entities)
-    return HTML_WRAPPER.format(soup)
 
+    return HTML_WRAPPER.format(soup)
 
 
-def render_svg(svg_file):
-    with open(svg_file, "r") as f:
-        lines = f.readlines()
-        svg = "".join(lines)
-
-        # """Renders the given svg string."""
-        b64 = base64.b64encode(svg.encode("utf-8")).decode("utf-8")
-        html = r'<img src="data:image/svg+xml;base64,%s"/>' % b64
-        return html
-
 def clean_text(text,doc=False,plain_text=False,url=False):
     """Return clean text from the various input sources"""
 
@@ -470,7 +454,7 @@ if summarize:
     
     with st.spinner("Calculating and matching entities, this takes a few seconds..."):
     
-        entity_match_html = highlight_entities(' '.join(text_to_summarize),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("####")