alltogether

app.py

@@ -215,16 +215,31 @@ def step4_track(state):
 
     return "✅ Pipeline complete!", image, state, gr.update(interactive=True)
 
+# New: run all steps sequentially
+@spaces.GPU()
+def run_pipeline(image_array, state):
+    # Step 1: Preprocess
+    status1, crop_img, state, _, _ = preprocess_image(image_array, state)
+    if "❌" in status1:
+        return status1, None, None, None, None, {}
+    # Step 2: Normals
+    status2, normals_img, state, _, _ = step2_normals(state)
+    # Step 3: UV Map
+    status3, uv_img, state, _, _ = step3_uv_map(state)
+    # Step 4: Tracking
+    status4, track_img, state, _ = step4_track(state)
+    final_status = "\n".join([status1, status2, status3, status4])
+    return final_status, crop_img, normals_img, uv_img, track_img, state
+
 # Build Gradio UI
 demo = gr.Blocks()
-
 with demo:
-    gr.Markdown("## Image Processing Pipeline")
-    gr.Markdown("Upload an image, then click the buttons in order. Uploading a new image will reset the process.")
+    gr.Markdown("## Image Processing Pipeline (Single Button)")
+    gr.Markdown("Upload an image and click 'Run Pipeline' to execute all steps.")
     with gr.Row():
         with gr.Column():
             image_in = gr.Image(label="Upload Image", type="numpy", height=512)
-            status = gr.Textbox(label="Status", lines=2, interactive=True, value="Upload an image to start.")
+            status = gr.Textbox(label="Status", lines=4, interactive=True, value="Upload an image to start.")
             state = gr.State({})
         with gr.Column():
             with gr.Row():
@@ -233,43 +248,16 @@ with demo:
             with gr.Row():
                 uv_img = gr.Image(label="UV Map", height=256)
                 track_img = gr.Image(label="Tracking", height=256)
+    run_btn = gr.Button("Run Pipeline")
 
-    with gr.Row():
-        preprocess_btn = gr.Button("Step 1: Preprocess", interactive=True)
-        normals_btn = gr.Button("Step 2: Normals", interactive=True)
-        uv_map_btn = gr.Button("Step 3: UV Map", interactive=True)
-        track_btn = gr.Button("Step 4: Track", interactive=True)
-
-    # Define component list for reset
-    outputs_for_reset = [crop_img, normals_img, uv_img, track_img, status, state, preprocess_btn, normals_btn, uv_map_btn, track_btn]
-
-    # Pipeline execution logic
-    preprocess_btn.click(
-        fn=preprocess_image,
+    # Single button click
+    run_btn.click(
+        fn=run_pipeline,
         inputs=[image_in, state],
-        outputs=[status, crop_img, state, preprocess_btn, normals_btn]
-    )
-    normals_btn.click(
-        fn=step2_normals,
-        inputs=[state],
-        outputs=[status, normals_img, state, normals_btn, uv_map_btn]
-    )
-    uv_map_btn.click(
-        fn=step3_uv_map,
-        inputs=[state],
-        outputs=[status, uv_img, state, uv_map_btn, track_btn]
-    )
-    track_btn.click(
-        fn=step4_track,
-        inputs=[state],
-        outputs=[status, track_img, state, track_btn]
+        outputs=[status, crop_img, normals_img, uv_img, track_img, state]
     )
 
-    # Event to reset everything when a new image is uploaded
-    image_in.upload(fn=reset_all, inputs=None, outputs=outputs_for_reset)
+    image_in.upload(fn=reset_all, inputs=None, outputs=[crop_img, normals_img, uv_img, track_img, status, state])
 
-# ------------------------------------------------------------------
-# START THE GRADIO SERVER
-# ------------------------------------------------------------------
 demo.queue()
 demo.launch(share=True, ssr_mode=False)