integrate html from nominalization detector demo

app.py

@@ -27,14 +27,26 @@ def call(model_name, sentence, detection_threshold):
     def pretty_pred_output(pred_info) -> str:
         return "\n".join([f"{qa['question']} --- {';'.join(qa['answers'])}"
                           for qa in pred_info['QAs']])
-    pretty_output = "\n".join(pretty_pred_output(pred_info) for pred_info in pipe_out_pred_infos)
-    return pretty_output, pipe_out_pred_infos
+    pretty_qa_output = "\n".join(pretty_pred_output(pred_info) for pred_info in pipe_out_pred_infos)
+    # also present highlighted predicates
+    positives = [pred_info['predicate_idx'] for pred_info in pred_infos] 
+    def color(idx):
+        if idx in positives: return "lightgreen"
+    idx2verb = {d["predicate_idx"] : d["verb_form"] for d in pred_infos}
+    idx2prob = {d["predicate_idx"] : d["predicate_detector_probability"] for d in pred_infos}
+    def word_span(word, idx):
+        tooltip = f'title=" probability={idx2prob[idx]:.2}
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, pretty_qa_output , pipe_out_pred_infos
 
 iface = gr.Interface(fn=call, 
                      inputs=[gr.inputs.Radio(choices=models, default=models[0], label="Model"), 
                              gr.inputs.Textbox(placeholder=input_sent_box_label, label="Sentence", lines=4), 
                              gr.inputs.Slider(minimum=0., maximum=1., step=0.01, default=0.5, label="Nominalization Detection Threshold")], 
-                     outputs=[gr.outputs.Textbox(label="Model Output"), gr.outputs.JSON(label="Model Output - JSON")],
+                     outputs=[gr.outputs.HTML(label="Detected Nominalizations"),
+                              gr.outputs.Textbox(label="Generated QAs"), 
+                              gr.outputs.JSON(label="Raw Model Output")],
                      title=title,
                      description=description,
                      article=links,