Update app.py

app.py

@@ -9,6 +9,7 @@ Eventually add filters based on headers
 """
 HEADERS = ["__index_level_0__", "problem", "username", "entrypoint", "tests_passed", "total_tests"]
 DATATYPES = ["number", "str", "str", "str", "number", "number"]
+
 SUCCESS_HEADERS = ["is_success", "first_attempt","is_first_success", "last_attempt", "is_last_success"]
 SUCCESS_DATATYPES = ["bool"]*5
 
@@ -90,7 +91,22 @@ def filter_by(
     
     dataset = gr.State(ds)
     return [dataset, *update_components(ds, 0, *components_to_update)]
-        
+
+def next_example(ds, *components):
+    slider_value = components[0]
+    print(slider_value)
+    # Adjust slider value within bounds (min 0, max len(ds)-1)
+    new_slider_value = int(slider_value)+1 if slider_value < len(ds)-1 else len(ds)-1
+    lesscomponents = components[1:]
+    return update_components(ds, new_slider_value, *lesscomponents)
+
+def prev_example(ds, *components):
+    slider_value = components[0]
+    # Adjust slider value within bounds (min 0, max len(ds)-1)
+    new_slider_value = int(slider_value)-1 if slider_value > 0 else 0
+    lesscomponents = components[1:]
+    return update_components(ds, new_slider_value, *lesscomponents)
+
 def main(args):
     ds = load_dataset(args.dataset, split=args.split)
     ds = ds.to_pandas()    
@@ -122,10 +138,14 @@ def main(args):
             interactive=False,
         )
         
-        prompt = gr.Code("__prompt__", language="python", label="Prompt")            
+        prompt = gr.Code("__prompt__", language="python", label="Prompt")     
+        with gr.Row():
+            prev_btn = gr.Button("Previous")
+            next_btn = gr.Button("Next")       
         submitted_text = gr.Textbox("__submitted_text__", type="text", label="Submitted Text")
         completion = gr.Code("__completion__", language="python", label="Completion")
                 
+
         with gr.Row():
             assertions = gr.Code("__assertions__", language="python", label="Assertions")
             prints = gr.Code("__prints__", language="python", label="Prints")
@@ -149,6 +169,10 @@ def main(args):
         # change example on slider change
         components = [slider, header_data, success_data, prompt, submitted_text, completion, assertions, prints, code_err, code_output]
         slider.input(fn=update_components, inputs=[dataset, *components], outputs=components)
+                
+        prev_btn.click(fn=prev_example, inputs=[dataset, *components], outputs=components)
+        next_btn.click(fn=next_example, inputs=[dataset, *components], outputs=components)
+
         # log
         callback.setup(components, "flagged_data_points")
         flagbtn.click(lambda *args: callback.flag(list(args)), components, flagout,
@@ -175,7 +199,7 @@ def main(args):
         filter_btn.click(fn=partial(filter_by, args.dataset, args.split), inputs=[*success_boxes, problem_box, student_box, *components], 
                          outputs=[dataset, *components])
 
-    demo.launch(share=True)
+    demo.launch(share=args.share)
     
 if __name__ == "__main__":
     parser = argparse.ArgumentParser()
@@ -184,3 +208,4 @@ if __name__ == "__main__":
     parser.add_argument("--share", action="store_true")
     args = parser.parse_args()
     main(args)
+