bugfixes

app.py

@@ -63,12 +63,12 @@ with demo:
         toggle_example_submit = gr.update(visible=not done)
         new_state_md = f"State: {state['cnt']}/{total_cnt} ({state['fooled']} fooled)"
 
+        query = parse_qs(dummy[1:])
         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
@@ -89,7 +89,7 @@ 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, toggle_final_submit, toggle_example_submit, new_state_md):
+    def _store_in_huggingface_dataset(state):
         with open(DATA_FILE, "a") as jsonlfile:
             jsonlfile.write(json.dumps(state) + "\n")
         repo.push_to_hub()
@@ -136,14 +136,14 @@ with demo:
                 document.body.appendChild(form);
                 form.submit();
             }
-            return [state, toggle_final_submit, toggle_example_submit, new_state_md];
+            return [state];
         }
         """
 
     submit_hit_button.click(
         _store_in_huggingface_dataset,
-        inputs=[state, toggle_final_submit, toggle_example_submit, new_state_md],
-        outputs=[state, toggle_final_submit, toggle_example_submit, new_state_md],
+        inputs=[state],
+        outputs=[state, final_submit, example_submit, state_display],
         _js=post_hit_js,
     )