bugfix

app.py

@@ -60,21 +60,25 @@ with demo:
         state["cnt"] += 1
 
         done = state["cnt"] == total_cnt
-        toggle_final_submit = gr.update(visible=done)
         toggle_example_submit = gr.update(visible=not done)
         new_state_md = f"State: {state['cnt']}/{total_cnt} ({state['cnt_fooled']} fooled)"
 
         state["data"].append({"cnt": state["cnt"], "text": txt, "target": tgt, "model_pred": pred["label"], "fooled": fooled})
 
         query = parse_qs(dummy[1:])
-        if "assignmentId" in query:
+        if "assignmentId" in query and query["assignmentId"][0] != "ASSIGNMENT_ID_NOT_AVAILABLE":
             # It seems that someone is using this app on mturk. We need to
             # store the assignmentId in the state before submit_hit_button
             # is clicked. We can do this here in _predict. We need to save the
             # assignmentId so that the turker can get credit for their HIT.
             state["assignmentId"] = query["assignmentId"][0]
+            toggle_final_submit = gr.update(visible=done)
+            toggle_final_submit_preview = gr.update(visible=False)
+        else:
+            toggle_final_submit_preview = gr.update(visible=done)
+            toggle_final_submit = gr.update(visible=False)
 
-        return pred_confidences, ret, state, toggle_example_submit, toggle_final_submit, new_state_md, dummy
+        return pred_confidences, ret, state, toggle_example_submit, toggle_final_submit, toggle_final_submit_preview, new_state_md, dummy
 
     # Input fields
     text_input = gr.Textbox(placeholder="Enter model-fooling statement", show_label=False)
@@ -87,6 +91,8 @@ with demo:
         submit_ex_button = gr.Button("Submit")
     with gr.Column(visible=False) as final_submit:
         submit_hit_button = gr.Button("Submit HIT")
+    with gr.Column(visible=False) as final_submit_preview:
+        submit_hit_button_preview = gr.Button("Submit Work (preview mode; no mturk HIT credit)")
 
     # Store the HIT data into a Hugging Face dataset.
     # The HIT is also stored and logged on mturk when post_hit_js is run below.
@@ -115,38 +121,27 @@ with demo:
     submit_ex_button.click(
         _predict,
         inputs=[text_input, label_input, state, dummy],
-        outputs=[label_output, text_output, state, example_submit, final_submit, state_display, dummy],
+        outputs=[label_output, text_output, state, example_submit, final_submit, final_submit_preview, state_display, dummy],
         _js=get_window_location_search_js,
     )
 
     post_hit_js = """
         function(state) {
-            if (state["assignmentId"] !== "" && state["assignmentId"] !== "ASSIGNMENT_ID_NOT_AVAILABLE"){
-                // If there is an assignmentId, then the submitter is on mturk
-                // and has accepted the HIT. So, we need to submit their HIT.
-                const form = document.createElement('form');
-                form.action = 'https://workersandbox.mturk.com/mturk/externalSubmit';
-                form.method = 'post';
-                for (const key in state) {
-                    const hiddenField = document.createElement('input');
-                    hiddenField.type = 'hidden';
-                    hiddenField.name = key;
-                    hiddenField.value = state[key];
-                    form.appendChild(hiddenField);
-                };
-                document.body.appendChild(form);
-                form.submit();
-                return state;
-            } else {
-                // If there is no assignmentId, then we assume that the submitter is
-                // on huggingface.co or there is an mturker doing the preview.
-                // This means that we can't log a HIT in mturk, but
-                // _store_in_huggingface_dataset will still store the example in
-                // our dataset without an assignmentId. The following line here
-                // loads the app again so the user can enter in another "fake" HIT.
-                window.location.href = window.location.href;
-                return state;
-            }
+            // If there is an assignmentId, then the submitter is on mturk
+            // and has accepted the HIT. So, we need to submit their HIT.
+            const form = document.createElement('form');
+            form.action = 'https://workersandbox.mturk.com/mturk/externalSubmit';
+            form.method = 'post';
+            for (const key in state) {
+                const hiddenField = document.createElement('input');
+                hiddenField.type = 'hidden';
+                hiddenField.name = key;
+                hiddenField.value = state[key];
+                form.appendChild(hiddenField);
+            };
+            document.body.appendChild(form);
+            form.submit();
+            return state;
         }
         """
 
@@ -157,4 +152,20 @@ with demo:
         _js=post_hit_js,
     )
 
+    refresh_app_js = """
+        function(state) {
+            // The following line here loads the app again so the user can
+            // enter in another preview-mode "HIT".
+            window.location.href = window.location.href;
+            return state;
+        }
+        """
+
+    submit_hit_button_preview.click(
+        _store_in_huggingface_dataset,
+        inputs=[state],
+        outputs=[state],
+        _js=refresh_app_js,
+    )
+
 demo.launch()