fix html highlight

candidates.py

@@ -74,14 +74,11 @@ def launch_app(df: pd.DataFrame, share_demo: bool = False):
 
         # Function to highlight words from the candidates list
         def highlight_words(text, candidates, color="yellow"):
-            if color == "yellow":
-                pattern = r'<mark>\1</mark>'
-            else:
-                pattern = r'<mark style="background-color:{green};">\1</mark>'
-
             for word in candidates:
                 # Use word boundaries to ensure we only match whole words
-                text = re.sub(rf'\b({re.escape(word)})\b', r'<mark style="background-color:{color};">\1</mark>'.format(color=color), text, flags=re.IGNORECASE)
+                text = re.sub(rf'\b({re.escape(word)})\b', r'<@>\1</@>', text, flags=re.IGNORECASE)
+            text = re.sub("<@>",f'<mark style="background-color:{color};">', text)
+            text = re.sub("</@>",'</mark>'.format(color=color), text)
             return text
 
         # Highlight words in the 'completion' column