Update app.py

app.py

@@ -135,26 +135,21 @@ with gr.Blocks() as block_demo:
         # if type(choice) is not str : choice = choice.value
         print(choice, image, method1, method2, property, ip)
         send_message_to_mongodb(image, property, method1, method2, choice, ip)
+        img1, img2, prop_text, image_state, method1_state, method2_state, property_state, img_input, ip_state = on_load()
         # new_image, new_method1, new_method2, new_image1, new_image2, new_property = get_random_comparison()
         return [
-            # gr.Markdown("### Thanks for your submission!"),
-            gr.Button(interactive=False),
-            gr.Button(interactive=False),
-            gr.Button(interactive=False),
-            gr.Button(interactive=False),
-            # gr.Markdown(f'Left image: {method_dict[method1]}'),
-            # gr.Markdown(f'Right image: {method_dict[method2]}'),
-            gr.Button(visible=False, interactive=True),
-            # gr.Image(new_image1), 
-            # gr.Image(new_image2),
-            # f'### Submit your choice for **{new_property}**'
+            img1, img2, prop_text, image_state, method1_state, method2_state, property_state, img_input, ip_state
         ]
 
-    l_butt.click(fn=update_interface, inputs=[method1_state, image_state, method1_state, method2_state, property_state, ip_state], outputs=[l_butt, r_butt, both_good, both_bad, refresh_butt], js="sleep(1);window.location.reload()")
-    r_butt.click(fn=update_interface, inputs=[method2_state, image_state, method1_state, method2_state, property_state, ip_state], outputs=[l_butt, r_butt, both_good, both_bad,  refresh_butt],js="sleep(1);window.location.reload()")
-    both_good.click(fn=update_interface, inputs=[gr.State('both_good'), image_state, method1_state, method2_state, property_state, ip_state], outputs=[l_butt, r_butt, both_good, both_bad, refresh_butt],js="sleep(1);window.location.reload()")
-    both_bad.click(fn=update_interface, inputs=[gr.State('both_bad'), image_state, method1_state, method2_state, property_state, ip_state], outputs=[l_butt, r_butt, both_good, both_bad, refresh_butt], js="sleep(1);window.location.reload()")
-
+    l_butt.click(fn=update_interface, inputs=[method1_state, image_state, method1_state, method2_state, property_state, ip_state], 
+                 outputs=[img1, img2, prop_text, image_state, method1_state, method2_state, property_state, img_input, ip_state])
+    r_butt.click(fn=update_interface, inputs=[method2_state, image_state, method1_state, method2_state, property_state, ip_state], 
+                 outputs=[img1, img2, prop_text, image_state, method1_state, method2_state, property_state, img_input, ip_state])
+    both_good.click(fn=update_interface, inputs=[gr.State('both_good'), image_state, method1_state, method2_state, property_state, ip_state], 
+                    outputs=[img1, img2, prop_text, image_state, method1_state, method2_state, property_state, img_input, ip_state])
+    both_bad.click(fn=update_interface, inputs=[gr.State('both_bad'), image_state, method1_state, method2_state, property_state, ip_state], 
+                   outputs=[img1, img2, prop_text, image_state, method1_state, method2_state, property_state, img_input, ip_state])
+    
 
     # refresh_butt.click(None, js="window.location.reload()")