changes to make it work on huggingface.co in addition to mturk

app.py

@@ -63,11 +63,13 @@ with demo:
         toggle_example_submit = gr.update(visible=not done)
         new_state_md = f"State: {state['cnt']}/{total_cnt} ({state['fooled']} fooled)"
 
-        # We need to store the assignmentId in the state before submit_hit_button
-        # is clicked. We can do this here in _predict, which is called before
-        # submit_hit_button is clicked
-        query = parse_qs(dummy[1:])
-        state["assignmentId"] = query["assignmentId"][0]
+        if "assignmentId" in query:
+            # 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.
+            query = parse_qs(dummy[1:])
+            state["assignmentId"] = query["assignmentId"][0]
 
         return pred_confidences, ret, state, toggle_example_submit, toggle_final_submit, new_state_md, dummy
 
@@ -87,11 +89,22 @@ with demo:
     # The HIT is also stored and logged on mturk when post_hit_js is run below.
     # This _store_in_huggingface_dataset function just demonstrates how easy it is
     # to automatically create a Hugging Face dataset from mturk.
-    def _store_in_huggingface_dataset(state):
+    def _store_in_huggingface_dataset(state, toggle_final_submit, toggle_example_submit, new_state_md):
         with open(DATA_FILE, "a") as jsonlfile:
-            jsonlfile.write(json.dumps(state))
+            jsonlfile.write(json.dumps(state) + "\n")
         repo.push_to_hub()
 
+        if state["assignmentId"] == "":
+            # If assignmentId is not set, then someone is using this app on
+            # huggingface.co, and we can reset the app to it's initial state
+            # after they submit their fake "HIT".
+            state = {"assignmentId": "", "cnt": 0, "fooled": 0, "data": [], "metadata": {}}
+            toggle_final_submit = gr.update(visible=False)
+            toggle_example_submit = gr.update(visible=True)
+            new_state_md = gr.Markdown(f"State: 0/{total_cnt} (0 fooled)")
+
+        return state, toggle_final_submit, toggle_example_submit, new_state_md
+
     # Button event handlers
     get_window_location_search_js = """
         function(text_input, label_input, state, dummy) {
@@ -107,27 +120,30 @@ with demo:
     )
 
     post_hit_js = """
-        function(state) {
-            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];
+        function(state, toggle_final_submit, toggle_example_submit, new_state_md) {
+            if (state["assignmentId"] !== "") {
+                //It seems that someone is using this app on mturk, so submit the 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, toggle_final_submit, toggle_example_submit, new_state_md];
         }
         """
 
     submit_hit_button.click(
         _store_in_huggingface_dataset,
-        inputs=[state],
-        outputs=None,
+        inputs=[state, toggle_final_submit, toggle_example_submit, new_state_md],
+        outputs=[state, toggle_final_submit, toggle_example_submit, new_state_md],
         _js=post_hit_js,
     )