Fixes

app.py

@@ -80,8 +80,8 @@ def span(text, span1, span2, label1, label2, model):
         idx2_1 = 0
         idx2_2 = 0
 
-        span1 = span1.split(" ")
-        span2 = span2.split(" ")
+        span1 = [split for split in span1.split(" ") if split]
+        span2 = [split for split in span2.split(" ") if split]
 
         for i in range(len(list(doc))):
             tok = list(doc)[i]
@@ -133,23 +133,26 @@ with demo:
         with gr.TabItem("Dependency"):
             col_punct = gr.Checkbox(label="Collapse Punctuation", value=True)
             col_phrase = gr.Checkbox(label="Collapse Phrases", value=True)
-            compact = gr.Checkbox(label="Compact", value=True)
+            compact = gr.Checkbox(label="Compact", value=False)
             depen_output = gr.HTML()
-
+            dep_button = gr.Button("Generate this tab")
         with gr.TabItem("Entity"):
             entity_input = gr.CheckboxGroup(DEFAULT_ENTS, value=DEFAULT_ENTS)
             entity_output = gr.HTML()
+            ent_button = gr.Button("Generate this tab")
         with gr.TabItem("Tokens"):
             with gr.Column():
                 tok_input = gr.CheckboxGroup(
                     DEFAULT_TOK_ATTR, value=DEFAULT_TOK_ATTR)
                 tok_output = gr.Dataframe(
                     headers=DEFAULT_TOK_ATTR, overflow_row_behaviour="paginate")
+            tok_button = gr.Button("Generate this tab")
         with gr.TabItem("Similarity"):
             with gr.Row():
                 sim_text1 = gr.Textbox(value="David Bowie", label="Chosen")
                 sim_text2 = gr.Textbox(value="the US", label="Chosen")
             sim_output = gr.Textbox(value="0.09", label="Similarity Score")
+            sim_button = gr.Button("Generate this tab")
         with gr.TabItem("Spans"):
             with gr.Column():
                 with gr.Row():
@@ -160,8 +163,10 @@ with demo:
                     span2 = gr.Textbox(label="Span 2")
                     label2 = gr.Textbox(value="Label 2",
                                         label="Label for Span 2")
-                with gr.Row():
-                    span_output = gr.HTML()
+                span_output = gr.HTML()
+                gr.Markdown(value="\n\n\n\n")
+                gr.Markdown(value="\n\n\n\n")
+                span_button = gr.Button("Generate this tab")
     text_button.click(get_text, inputs=[model_input], outputs=text_input)
     button.click(dependency, inputs=[
         text_input, col_punct, col_phrase, compact, model_input], outputs=depen_output)
@@ -173,5 +178,15 @@ with demo:
         sim_output, sim_text1, sim_text2])
     button.click(
         span, inputs=[text_input, span1, span2, label1, label2, model_input], outputs=span_output)
+    dep_button.click(dependency, inputs=[
+        text_input, col_punct, col_phrase, compact, model_input], outputs=depen_output)
+    ent_button.click(
+        entity, inputs=[text_input, entity_input, model_input], outputs=entity_output)
+    tok_button.click(
+        token, inputs=[text_input, tok_input, model_input], outputs=tok_output)
+    sim_button.click(vectors, inputs=[text_input, model_input], outputs=[
+        sim_output, sim_text1, sim_text2])
+    span_button.click(
+        span, inputs=[text_input, span1, span2, label1, label2, model_input], outputs=span_output)
 
 demo.launch()