Update app.py

app.py

@@ -113,39 +113,45 @@ def stitch_all(vid12, vid23):
         return None
     return concat_videos(vid12, vid23)
 
-# -------- UI --------
+# ---------- UI ----------
 CSS = """
 .gradio-container { padding: 24px; }
+.pill button { border-radius: 999px !important; padding: 10px 18px; }
+.rounded textarea { border-radius: 16px !important; }
 """
 
-with gr.Blocks(css=CSS, title="Stitch — 3 uploads, 2 stitches") as demo:
-    gr.Markdown("## Stitch — Upload 3 images → Generate 1→2, 2→3, then combine.")
+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.")
 
-# --- Uploads row (side-by-side) ---
-with gr.Row():
-    with gr.Column(scale=1, min_width=280):
-        img1 = gr.Image(label="Image 1 upload", type="pil")
-    with gr.Column(scale=1, min_width=280):
-        img2 = gr.Image(label="Image 2 upload", type="pil")
-    with gr.Column(scale=1, min_width=280):
-        img3 = gr.Image(label="Image 3 upload", type="pil")
+    # Top row: 1 - 2 - 3 (three columns)
+    with gr.Row():
+        with gr.Column(scale=1, min_width=280):
+            img1 = gr.Image(label="Image 1 upload", type="pil")
+        with gr.Column(scale=1, min_width=280):
+            img2 = gr.Image(label="Image 2 upload", type="pil")
+        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 = gr.Number(value=0, precision=0, label="Seed (0 = random)")
+
+    # Stitch 1→2 (prompt+button row, then video below)
+    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.Row():
-        # Prompts + buttons
-        with gr.Column():
-            seed = gr.Number(value=0, precision=0, label="Seed (0 = random)")
-            prompt12 = gr.Textbox(placeholder="Prompt for stitching 1→2", lines=2, label="Prompt (1→2)")
-            btn12 = gr.Button("Stitch 1&2")
-            prompt23 = gr.Textbox(placeholder="Prompt for stitching 2→3", lines=2, label="Prompt (2→3)")
-            btn23 = gr.Button("Stitch 2&3")
-            btn_all = gr.Button("Stitch All (combine 1→2 and 2→3)")
-
-        with gr.Column():
-            vid12 = gr.Video(label="Video (1→2)")
-            vid23 = gr.Video(label="Video (2→3)")
-            vid_all = gr.Video(label="Final Combined Video")
-
-    # Wire
+        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)
+
+    # Wire buttons
     btn12.click(stitch_12, inputs=[prompt12, seed, img1, img2], outputs=[vid12])
     btn23.click(stitch_23, inputs=[prompt23, seed, img2, img3], outputs=[vid23])
     btn_all.click(stitch_all, inputs=[vid12, vid23], outputs=[vid_all])