Update app.py

app.py

@@ -308,18 +308,21 @@ with gradio.Blocks() as demo:
             t1 = gradio.Dropdown(endpoint_types, label="Type 1", value=random.choice(endpoint_types))
             d1 = gradio.Dropdown(density_options, label="Density 1", value=random.choice(density_options))
             th1 = gradio.Dropdown(thickness_options, label="Thickness 1", value=random.choice(thickness_options))
-            img1 = gradio.Image(label="Endpoint 1")
-            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])
+        with gradio.Column():
+            img1 = gradio.Image(label="Endpoint 2")
         with gradio.Column():
             t2 = gradio.Dropdown(endpoint_types, label="Type 2", value=random.choice(endpoint_types))
             d2 = gradio.Dropdown(density_options, label="Density 2", value=random.choice(density_options))
             th2 = gradio.Dropdown(thickness_options, label="Thickness 2", value=random.choice(thickness_options))
+        with gradio.Column():
             img2 = gradio.Image(label="Endpoint 2")
-            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])
+
+        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")