improve

app.py

@@ -169,6 +169,7 @@ def process(dataframe, row_number=0):
         for i in range(len(SUBMISSIONS))
     ]
     images = (SUBMISSIONS[c][int(image_id)]["images"] for c in choices)
+    print({i: c for i, c in enumerate(choices)})
 
     prompt = SUBMISSIONS[choices[0]][int(image_id)]["Prompt"]
     prompt = f'# "{prompt}"'
@@ -182,7 +183,9 @@ def write_result(user_choice, row_number, dataframe):
         return row_number, dataframe
 
     user_choice = int(user_choice)
+    print(user_choice)
     chosen_model = submission_names[dataframe.iloc[row_number][f"choice_{user_choice}"]]
+    print(chosen_model)
 
     dataframe.loc[row_number, "result"] = chosen_model
     return row_number + 1, dataframe
@@ -320,32 +323,108 @@ with gr.Blocks() as demo:
         fn=process, inputs=[dataframe], outputs=[image_1, image_2, image_3, image_4, prompt, counter]
     )
 
-    images = [image_1_button, image_2_button, image_3_button, image_4_button]
-    for i, button in enumerate(images):
-        button.click(
-            fn=lambda: i,
-            inputs=[],
-            outputs=[selected_image],
-        ).then(
-            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, start_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],
-        )
+    image_1_button.click(
+        fn=lambda: 0,
+        inputs=[],
+        outputs=[selected_image],
+    ).then(
+        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, start_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],
+    )
+
+    image_2_button.click(
+        fn=lambda: 1,
+        inputs=[],
+        outputs=[selected_image],
+    ).then(
+        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, start_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],
+    )
+
+    image_3_button.click(
+        fn=lambda: 2,
+        inputs=[],
+        outputs=[selected_image],
+    ).then(
+        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, start_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],
+    )
+
+    image_4_button.click(
+        fn=lambda: 3,
+        inputs=[],
+        outputs=[selected_image],
+    ).then(
+        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, start_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()