Update app.py

app.py

@@ -123,7 +123,7 @@ CSS = """
 with gr.Blocks(css=CSS, title="Stitch — vertical flow") as demo:
     gr.Markdown("## Stitch — Upload 3 images (top), then stitch 1→2 and 2→3, then merge.")
 
-    # Top row: 1 - 2 - 3 (three columns)
+    # Top row: 1 - 2 - 3 (side-by-side)
     with gr.Row():
         with gr.Column(scale=1, min_width=280):
             img1 = gr.Image(label="Image 1 upload", type="pil")
@@ -132,24 +132,37 @@ with gr.Blocks(css=CSS, title="Stitch — vertical flow") as demo:
         with gr.Column(scale=1, min_width=280):
             img3 = gr.Image(label="Image 3 upload", type="pil")
 
-    # Seed (standalone, not inside the Row above)
+    # Seed under the uploads
     seed = gr.Number(value=0, precision=0, label="Seed (0 = random)")
 
-    # Stitch 1→2 (prompt+button row, then video below)
+    # Stitch 1→2: LEFT = prompt+button, RIGHT = video
     with gr.Row():
-        prompt12 = gr.Textbox(placeholder="Prompt for stitching 1→2", lines=2, label="Prompt (1→2)", elem_classes=["rounded"])
-        btn12 = gr.Button("Stitch 1→2", elem_classes=["pill"])
-    vid12 = gr.Video(label="Video (1→2)", interactive=False)
-
-    # Stitch 2→3
+        with gr.Column(scale=1, min_width=420):
+            prompt12 = gr.Textbox(
+                placeholder="Prompt for stitching 1→2",
+                lines=2, label="Prompt (1→2)", elem_classes=["rounded"]
+            )
+            btn12 = gr.Button("Stitch 1→2", elem_classes=["pill"], full_width=True)
+        with gr.Column(scale=1, min_width=420):
+            vid12 = gr.Video(label="Video (1→2)", interactive=False)
+
+    # Stitch 2→3: LEFT = prompt+button, RIGHT = video
     with gr.Row():
-        prompt23 = gr.Textbox(placeholder="Prompt for stitching 2→3", lines=2, label="Prompt (2→3)", elem_classes=["rounded"])
-        btn23 = gr.Button("Stitch 2→3", elem_classes=["pill"])
-    vid23 = gr.Video(label="Video (2→3)", interactive=False)
-
-    # Final merge
-    btn_all = gr.Button("Stitch All (concatenate 1→2 then 2→3)", elem_classes=["pill"])
-    vid_all = gr.Video(label="Final Combined Video", interactive=False)
+        with gr.Column(scale=1, min_width=420):
+            prompt23 = gr.Textbox(
+                placeholder="Prompt for stitching 2→3",
+                lines=2, label="Prompt (2→3)", elem_classes=["rounded"]
+            )
+            btn23 = gr.Button("Stitch 2→3", elem_classes=["pill"], full_width=True)
+        with gr.Column(scale=1, min_width=420):
+            vid23 = gr.Video(label="Video (2→3)", interactive=False)
+
+    # Final merge: LEFT = button, RIGHT = final video
+    with gr.Row():
+        with gr.Column(scale=1, min_width=420):
+            btn_all = gr.Button("Stitch All (concatenate 1→2 then 2→3)", elem_classes=["pill"], full_width=True)
+        with gr.Column(scale=1, min_width=420):
+            vid_all = gr.Video(label="Final Combined Video", interactive=False)
 
     # Wire buttons
     btn12.click(stitch_12, inputs=[prompt12, seed, img1, img2], outputs=[vid12])