Update app.py

app.py

@@ -56,6 +56,19 @@ def read_instruction(idx):
     assert len(item) == 1
     return item[0]['instruction']
 
+def on_prev(state):
+    for i, item in enumerate(data):
+        if item['index'] == state:
+            break
+    return data[i - 1]['index'], data[i - 1]['index']
+
+def on_next(state):
+    for i, item in enumerate(data):
+        if item['index'] == state:
+            break
+    return data[i + 1]['index'], data[i + 1]['index']
+    
+
 with gr.Blocks() as demo:
     gr.Markdown("# Gallery of Generation Results on RISEBench")
 
@@ -69,6 +82,8 @@ with gr.Blocks() as demo:
                 def update_state(problem_index):
                     return problem_index
                 problem_index.submit(fn=update_state, inputs=[problem_index], outputs=[state])
+                prev_button.click(fn=on_prev, inputs=[state], outputs=[state, problem_index])
+                next_button.click(fn=on_next, inputs=[state], outputs=[state, problem_index])
 
             model_checkboxes = gr.CheckboxGroup(label="Select Models", choices=model_names, value=model_names)