add tooltip to output html

app.py

@@ -11,9 +11,9 @@ description = f"""This is a demo of QANom's nominalization detection algorithm,
 links = """<p style='text-align: center'>
 <a href='https://github.com/kleinay/QANom' target='_blank'>QANom repo</a>  |  
 <a href='https://huggingface.co/kleinay/nominalization-candidate-classifier' target='_blank'>Model Repo at Huggingface Hub</a>  |  
-<a href='https://www.aclweb.org/anthology/2020.coling-main.274/' target='_blank'>QANom Paper</a>  |  
+<a href='https://www.aclweb.org/anthology/2020.coling-main.274/' target='_blank'>QANom Paper</a>   
 </p>"""
-examples = [["The doctor was interested in Luke 's treatment .", True, 0.6], 
+examples = [["The doctor was interested in Luke 's treatment .", True, 0.5], 
             ["the construction of the officer 's building finished right after the beginning of the destruction of the previous construction .", True, 0.7]]
 
 def call(sentence: str, return_all_candidates: bool, threshold: float):
@@ -27,9 +27,11 @@ def call(sentence: str, return_all_candidates: bool, threshold: float):
     def color(idx):
         if idx in positives: return "lightgreen"
         if idx in negatives: return "pink"
-    highlighted = [(word, color(idx)) for idx, word in enumerate(sentence.split(" "))]
+    idx2verb = {d["predicate_idx"] : d["verb_form"] for d in ret}
+    idx2prob = {d["predicate_idx"] : d["predicate_detector_probability"] for d in ret}
     def word_span(word, idx):
-        return f'<span style="background-color: {color(idx)}">{word}</span>'
+        tooltip = f'title=" probability={idx2prob[idx]:.2}&#010;verb={idx2verb[idx]}"' if idx in idx2verb else ''
+        return f'<span {tooltip} style="background-color: {color(idx)}">{word}</span>'
     html = '<span>' + ' '.join(word_span(word, idx) for idx, word in enumerate(sentence.split(" "))) + '</span>'
     return html, ret