Update app.py

app.py

@@ -123,7 +123,14 @@ def set_sidebar(css):
 
 def get_el_annotations(response):
     # swap labels key with ents
-    ents = [{"start": l.start, "end": l.end, "label": l.label} for l in response.labels]
+    ents = [
+        {
+            "start": l.start,
+            "end": l.end,
+            "label": f"<a href='https://en.wikipedia.org/wiki/{l.label.replace(' ', '_')}' style='color: #222222'>{l.label}</a>",
+        }
+        for l in response.labels
+    ]
     dict_of_ents = {"text": response.text, "ents": ents}
     label_in_text = set(l["label"] for l in dict_of_ents["ents"])
     options = {"ents": label_in_text, "colors": get_random_color(label_in_text)}
@@ -228,13 +235,6 @@ def run_client():
                     dict_of_ents, manual=True, style="ent", options=options
                 )
                 display = display.replace("\n", " ")
-                # wsd_display = re.sub(
-                #     r"(wiki::\d+\w)",
-                #     r"<a href='https://babelnet.org/synset?id=\g<1>&orig=\g<1>&lang={}'>\g<1></a>".format(
-                #         language.upper()
-                #     ),
-                #     wsd_display,
-                # )
                 with st.container():
                     st.write(display, unsafe_allow_html=True)