Update app.py

app.py

@@ -233,28 +233,26 @@ with gr.Blocks() as demo:
         )
         gr.Markdown("Click on start to play again!")
 
+    with gr.Column(visible=False):
+        selected_image = gr.Number(label="Selected index", value=-1, precision=0)
+
     with gr.Column(visible=False) as gallery_view:
         counter = gr.Markdown(f" ### 1/{NUM_QUESTIONS}")
         prompt = gr.Markdown(PROMPT_FORMAT.format(""))
         with gr.Blocks():
             with gr.Row() as images:
-                with gr.Column:
+                with gr.Column():
                     image_1 = gr.Image(interactive=False)
                     image_1_button = gr.Button("Select 1").style(full_width=True)
-                with gr.Column:
+                with gr.Column():
                     image_2 = gr.Image(interactive=False)
                     image_2_button = gr.Button("Select 2").style(full_width=True)
-                with gr.Column:
+                with gr.Column():
                     image_3 = gr.Image(interactive=False)
                     image_3_button = gr.Button("Select 3").style(full_width=True)
-                with gr.Column:
+                with gr.Column():
                     image_3 = gr.Image(interactive=False)
                     image_3_button = gr.Button("Select 4").style(full_width=True)         
-                
-        next_button = gr.Button("Confirm").style(full_width=True)
-
-    with gr.Column(visible=False):
-        selected_image = gr.Number(label="Selected index", value=-1, precision=0)
 
     start_button.click(
         fn=start,
@@ -273,32 +271,27 @@ with gr.Blocks() as demo:
         fn=process, inputs=[dataframe], outputs=[image_1, image_2, image_3, image_4, prompt, counter]
     )
 
-    gallery.select(
-        fn=get_index,
-        outputs=selected_image,
-        queue=False,
-    )
-
-    next_button.click(
-        fn=write_result,
-        inputs=[selected_image, row_number, dataframe],
-        outputs=[row_number, dataframe],
-    ).then(
-        fn=change_view,
-        inputs=[row_number, dataframe],
-        outputs=[intro_view, result_view, gallery_view, result]
-    ).then(
-        fn=process,
-        inputs=[dataframe, row_number],
-        outputs=[image_1, image_2, image_3, image_4, prompt, counter]
-    ).then(
-        fn=lambda x: 0 if x == NUM_QUESTIONS else x,
-        inputs=[row_number],
-        outputs=[row_number],
-    ).then(
-        fn=refresh,
-        inputs=[row_number, dataframe],
-        outputs=[dataframe],
-    )
+    for i, button in enumerate([image_1_button, image_2_button, image_3_button, image_4_button]):
+        button.click(
+            fn=write_result,
+            inputs=[i, row_number, dataframe],
+            outputs=[row_number, dataframe],
+        ).then(
+            fn=change_view,
+            inputs=[row_number, dataframe],
+            outputs=[intro_view, result_view, gallery_view, result]
+        ).then(
+            fn=process,
+            inputs=[dataframe, row_number],
+            outputs=[image_1, image_2, image_3, image_4, prompt, counter]
+        ).then(
+            fn=lambda x: 0 if x == NUM_QUESTIONS else x,
+            inputs=[row_number],
+            outputs=[row_number],
+        ).then(
+            fn=refresh,
+            inputs=[row_number, dataframe],
+            outputs=[dataframe],
+        )
 
 demo.launch()