debug

app.py

@@ -18,7 +18,7 @@ with demo:
 
     # We keep track of state as a Variable
     state_dict = {"assignmentId": "", "cnt": 0, "fooled": 0, "data": [], "metadata": {}}
-    state = gr.Variable(state_dict)
+    state = gr.JSON(state_dict, visible=False)
 
     gr.Markdown("# DADC in Gradio example")
     gr.Markdown("Try to fool the model and find an example where it predicts the wrong label!")
@@ -27,7 +27,7 @@ with demo:
 
     # Generate model prediction
     # Default model: distilbert-base-uncased-finetuned-sst-2-english
-    def _predict(txt, tgt, state):
+    def _predict(txt, tgt, state, dummy):
         pred = pipe(txt)[0]
         other_label = 'negative' if pred['label'].lower() == "positive" else "positive"
         pred_confidences = {pred['label'].lower(): pred['score'], other_label: 1 - pred['score']}
@@ -46,7 +46,11 @@ with demo:
         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['fooled']} fooled)"
-        return pred_confidences, ret, state, toggle_example_submit, toggle_final_submit, new_state_md
+
+        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
 
     # Input fields
     text_input = gr.Textbox(placeholder="Enter model-fooling statement", show_label=False)
@@ -73,16 +77,20 @@ with demo:
     # Button event handlers
     submit_ex_button.click(
         _predict,
-        inputs=[text_input, label_input, state],
-        outputs=[label_output, text_output, state, example_submit, final_submit, state_display],
+        inputs=[text_input, label_input, state, dummy],
+        outputs=[label_output, text_output, state, example_submit, final_submit, state_display, dummy],
+        _js="function(text_input, label_input, state, dummy) { console.log(text_input); console.log(label_input); console.log(state); console.log(dummy); return [text_input, label_input, state, window.location.search]; }",
     )
 
-    response_output = gr.Textbox()
+    def _something(state):
+        print(state)
+        return state
+
     submit_hit_button.click(
-        _submit,
-        inputs=[state, dummy],
-        outputs=[response_output, state, dummy],
-        _js="function(state, dummy) { console.log(window); return [state, window.location.search]; }",
+        _something,
+        inputs=[state],
+        outputs=[state],
+        _js="function(state) { console.log(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); console.log(state); console.log(form); form.submit(); return [state];}",
     )
 
-demo.launch()
+demo.launch(share=True)