Update app.py

app.py

@@ -9,6 +9,13 @@ import gradio as gr
 import requests
 
 
+def mark_tokens_bold(string, tokens):
+  for token in tokens:
+    pattern = r"\b" + re.escape(token) + r"\b"
+    string = re.sub(pattern, "<b>" + token + "</b>", string)
+  return string
+
+
 def process_results(results, highlight_terms):
     if len(results) == 0:
         return """<br><p style='font-family: Arial; color:Silver; text-align: center;'>
@@ -16,14 +23,8 @@ def process_results(results, highlight_terms):
 
     results_html = ""
     for result in results:
-        tokens = result["text"].split()
-        tokens_html = []
-        for token in tokens:
-            if token in highlight_terms:
-                tokens_html.append("<b>{}</b>".format(token))
-            else:
-                tokens_html.append(token)
-        tokens_html = " ".join(tokens_html)
+        text_html = result["text"]
+        text_html = mark_tokens_bold(text_html, highlight_terms)
         meta_html = (
             """
                 <p class='underline-on-hover' style='font-size:12px; font-family: Arial; color:#585858; text-align: left;'>