add predict button

app_text_classification.py

@@ -5,10 +5,11 @@ import gradio as gr
 from io_utils import (get_logs_file, read_scanners, write_scanners)
 from text_classification_ui_helpers import (check_dataset_and_get_config,
                                             check_dataset_and_get_split,
-                                            check_model_and_show_prediction,
+                                            align_columns_and_show_prediction,
                                             deselect_run_inference,
                                             select_run_mode, try_submit,
-                                            write_column_mapping_to_config)
+                                            write_column_mapping_to_config,
+                                            precheck_model_ds_enable_example_btn)
 from wordings import CONFIRM_MAPPING_DETAILS_MD, INTRODUCTION_MD
 
 MAX_LABELS = 40
@@ -40,6 +41,13 @@ def get_demo():
         dataset_config_input = gr.Dropdown(label="Dataset Config", visible=False)
         dataset_split_input = gr.Dropdown(label="Dataset Split", visible=False)
 
+    with gr.Row():
+        example_btn = gr.Button(
+            "Auto-align Columns & Get Sample Prediction", 
+            visible=True, 
+            variant="primary",
+            interactive=False)
+
     with gr.Row():
         example_input = gr.HTML(visible=False)
     with gr.Row():
@@ -90,7 +98,7 @@ def get_demo():
         run_btn = gr.Button(
             "Get Evaluation Result",
             variant="primary",
-            interactive=True,
+            interactive=False,
             size="lg",
         )
 
@@ -122,7 +130,7 @@ def get_demo():
         inputs=[run_local],
         outputs=[inference_token, run_inference],
     )
-    
+
     gr.on(
         triggers=[label.change for label in column_mappings],
         fn=write_column_mapping_to_config,
@@ -147,9 +155,21 @@ def get_demo():
             model_id_input.change,
             dataset_id_input.change,
             dataset_config_input.change,
-            dataset_split_input.change,
+            dataset_split_input.change], 
+        fn=precheck_model_ds_enable_example_btn,
+        inputs=[
+            model_id_input,
+            dataset_id_input,
+            dataset_config_input,
+            dataset_split_input,
+        ], 
+        outputs=[example_btn])
+
+    gr.on(
+        triggers=[
+            example_btn.click,
         ],
-        fn=check_model_and_show_prediction,
+        fn=align_columns_and_show_prediction,
         inputs=[
             model_id_input,
             dataset_id_input,
@@ -161,6 +181,7 @@ def get_demo():
             example_input,
             example_prediction,
             column_mapping_accordion,
+            run_btn,
             *column_mappings,
         ],
     )
@@ -188,13 +209,10 @@ def get_demo():
 
     gr.on(
         triggers=[
-            model_id_input.change,
-            dataset_config_input.change,
-            dataset_split_input.change,
-            run_inference.change,
-            run_local.change,
-            inference_token.change,
-            scanners.change,
+            run_inference.input,
+            run_local.input,
+            inference_token.input,
+            scanners.input,
         ],
         fn=enable_run_btn,
         inputs=None,
@@ -202,8 +220,8 @@ def get_demo():
     )
 
     gr.on(
-        triggers=[label.change for label in column_mappings],
+        triggers=[label.input for label in column_mappings],
         fn=enable_run_btn,
-        inputs=None,
+        inputs=column_mappings,
         outputs=[run_btn],
     )

text_classification_ui_helpers.py

@@ -141,8 +141,21 @@ def list_labels_and_features_from_dataset(ds_labels, ds_features, model_id2label
 
     return lables + features
 
+def precheck_model_ds_enable_example_btn(model_id, dataset_id, dataset_config, dataset_split):
+    ppl = check_model(model_id)
+    if ppl is None or not isinstance(ppl, TextClassificationPipeline):
+        gr.Warning("Please check your model.")
+        return gr.update(interactive=False)
+    ds_labels, ds_features = get_labels_and_features_from_dataset(
+        dataset_id, dataset_config, dataset_split
+    )
+    if not isinstance(ds_labels, list) or not isinstance(ds_features, list):
+        gr.Warning(CHECK_CONFIG_OR_SPLIT_RAW)
+        return gr.update(interactive=False)
 
-def check_model_and_show_prediction(
+    return gr.update(interactive=True)
+
+def align_columns_and_show_prediction(
     model_id, dataset_id, dataset_config, dataset_split, uid
 ):
     ppl = check_model(model_id)
@@ -151,6 +164,8 @@ def check_model_and_show_prediction(
         return (
             gr.update(visible=False),
             gr.update(visible=False),
+            gr.update(visible=False, open=False),
+            gr.update(interactive=False),
             *[gr.update(visible=False) for _ in range(MAX_LABELS + MAX_FEATURES)],
         )
 
@@ -164,6 +179,7 @@ def check_model_and_show_prediction(
             gr.update(visible=False),
             gr.update(visible=False),
             gr.update(visible=False, open=False),
+            gr.update(interactive=False),
             *dropdown_placement,
         )
     model_id2label = ppl.model.config.id2label
@@ -178,6 +194,7 @@ def check_model_and_show_prediction(
             gr.update(visible=False),
             gr.update(visible=False),
             gr.update(visible=False, open=False),
+            gr.update(interactive=False),
             *dropdown_placement,
         )
 
@@ -198,6 +215,7 @@ def check_model_and_show_prediction(
             gr.update(value=MAPPING_STYLED_ERROR_WARNING, visible=True),
             gr.update(visible=False),
             gr.update(visible=True, open=True),
+            gr.update(interactive=True),
             *column_mappings,
         )
 
@@ -208,6 +226,7 @@ def check_model_and_show_prediction(
         gr.update(value=get_styled_input(prediction_input), visible=True),
         gr.update(value=prediction_output, visible=True),
         gr.update(visible=True, open=False),
+        gr.update(interactive=True),
         *column_mappings,
     )