Update app.py

app.py

@@ -65,6 +65,10 @@ def parse_envs(folder='./videos'):
     return envs
 
 def build_interface(iter=3, data_folder='./videos'):
+    HF_TOKEN = os.getenv('HF_TOKEN')
+    hf_writer = gr.HuggingFaceDatasetSaver(HF_TOKEN, "crowdsourced-robotinder-demo")
+    callback = hf_writer
+    
     # build gradio interface
     with gr.Blocks() as demo:
         gr.Markdown("Here is RoboTinder!")
@@ -88,12 +92,17 @@ def build_interface(iter=3, data_folder='./videos'):
                 right = gr.Image(right_video_path)
 
         btn1 = gr.Button("Replay")
+        user_choice = gr.Radio(["Left", "Right", "Not Sure"], label="Which one is your favorite?")
+        btn2 = gr.Button("Next")
+
+        # This needs to be called at some point prior to the first call to callback.flag()
+        callback.setup([user_choice, left, right], "flagged_data_points")
+        
         btn1.click(fn=replay, inputs=[left, right], outputs=[left, right])
+        btn2.click(fn=update, inputs=None, outputs=[left, right])
 
-        gr.Radio(["Left", "Right", "Not Sure"], label="Which one is your favorite?")
-        btn2 = gr.Button("Next")
-        # choose env
-        btn2.click(fn=update, inputs=[left, right], outputs=[left, right])
+        # We can choose which components to flag -- in this case, we'll flag all of them
+        btn2.click(lambda *args: callback.flag(args), [user_choice, left, right], None, preprocess=False)
 
     return demo