legend update

app.py

@@ -476,6 +476,8 @@ def sub_ex(*inputs):
         out = inputs
     return out
 
+def show_legend(show_legend):
+    return gr.update(show_legend=show_legend)
 
 def create_timeline_plot(data: List[Dict[str, Any]]):
     df = pd.DataFrame(data)
@@ -548,8 +550,10 @@ with gr.Blocks(head=shortcut_js,
                         placeholder="Enter text here...")
                 text_btn = gr.Button('Run')
             with gr.Column():
-                gr.Markdown("## Labeled Summary or Note"),
-                text_out = gr.Highlight(label="", combine_adjacent=True, show_legend=False, color_map=color_map)
+                with gr.Row():
+                    gr.Markdown("## Labeled Summary or Note")
+                    show_legend_checkbox = gr.Checkbox(label="Show Legend", value=True,)
+                text_out = gr.Highlight(label="", combine_adjacent=True, show_legend=True, color_map=color_map)
         gr.Examples(text_examples, inputs=text_input)
     with gr.Tab("Patient Visualization"):
         gr.Markdown(vis_desc)
@@ -667,8 +671,13 @@ with gr.Blocks(head=shortcut_js,
         outputs=text_input
     )
 
-    text_input.submit(process, inputs=text_input, outputs=text_out)
-    text_btn.click(process, inputs=text_input, outputs=text_out)
+    text_input.submit(process, inputs=[text_input], outputs=text_out)
+    text_btn.click(process, inputs=[text_input], outputs=text_out)
+    show_legend_checkbox.change(
+        show_legend,
+        inputs=[show_legend_checkbox],
+        outputs=text_out
+    )
     upload.change(update_inputs, inputs=upload, outputs=sum_inputs)
     ex_add.click(add_ex, inputs=sum_inputs, outputs=sum_inputs)
     ex_sub.click(sub_ex, inputs=sum_inputs, outputs=sum_inputs)