Better button callbacks

app.py

@@ -311,12 +311,18 @@ with gradio.Blocks() as demo:
         with gradio.Column():
             img2 = gradio.Image(label="Endpoint 2")
 
-        t1.change(fn=generate_unit_cell, inputs=[t1, d1, th1], outputs=[img1])
-        d1.change(fn=generate_unit_cell, inputs=[t1, d1, th1], outputs=[img1])
-        th1.change(fn=generate_unit_cell, inputs=[t1, d1, th1], outputs=[img1])
-        t2.change(fn=generate_unit_cell, inputs=[t2, d2, th2], outputs=[img2])
-        d2.change(fn=generate_unit_cell, inputs=[t2, d2, th2], outputs=[img2])
-        th2.change(fn=generate_unit_cell, inputs=[t2, d2, th2], outputs=[img2])
+        lattice_inputs_1 = [t1, d1, th1]
+        [x.change(fn=generate_unit_cell, inputs=lattice_inputs_1, outputs=[img1]) for x in lattice_inputs_1]
+        lattice_inputs_2 = [t2, d2, th2]
+        [x.change(fn=generate_unit_cell, inputs=lattice_inputs_1, outputs=[img1]) for x in lattice_inputs_2]
+
+        # t1.change(fn=generate_unit_cell, inputs=[t1, d1, th1], outputs=[img1])
+        # d1.change(fn=generate_unit_cell, inputs=[t1, d1, th1], outputs=[img1])
+        # th1.change(fn=generate_unit_cell, inputs=[t1, d1, th1], outputs=[img1])
+        #
+        # t2.change(fn=generate_unit_cell, inputs=[t2, d2, th2], outputs=[img2])
+        # d2.change(fn=generate_unit_cell, inputs=[t2, d2, th2], outputs=[img2])
+        # th2.change(fn=generate_unit_cell, inputs=[t2, d2, th2], outputs=[img2])
     steps = gradio.Dropdown(interpolation_options, label="Interpolation Length", value=random.choice(interpolation_options))
     btn = gradio.Button("Run")
     img = gradio.Image(label="Transition")