Update app.py

app.py

@@ -70,7 +70,6 @@ def get_video_info(video_path):
     return seconds_cfg, fps_cfg
 
 # Step 1: Trim video based on user-defined duration and fps based on user-defined duration and fps
-@spaces.GPU()
 def step1_trim(video_path, seconds, fps, state):
     session_id = str(uuid.uuid4())
     base_dir = os.path.join(os.environ["PIXEL3DMM_PREPROCESSED_DATA"], session_id)
@@ -78,12 +77,22 @@ def step1_trim(video_path, seconds, fps, state):
 
     tmp = tempfile.mkdtemp()
     trimmed = os.path.join(tmp, f"{session_id}.mp4")
-    subprocess.run([
+    
+    try:
+        # capture both stdout & stderr
+        p = subprocess.run([
         "ffmpeg", "-y", "-i", video_path,
         "-t", str(seconds),  # user-specified duration
         "-r", str(fps),      # user-specified fps
         trimmed
-    ], check=True, stdout=subprocess.PIPE, stderr=subprocess.STDOUT)
+        ], check=True, stdout=subprocess.PIPE, stderr=subprocess.STDOUT)
+        
+    except subprocess.CalledProcessError as e:
+        # e.stdout contains everything
+        err = f"❌ Preprocess failed (exit {e.returncode}).\n\n{e.stdout}"
+        return err, None, state
+
+
     state["trimmed_path"] = trimmed
     return f"✅ Step 1: Trimmed to {seconds}s @{fps}fps", state
 
@@ -94,8 +103,6 @@ def step2_preprocess(state):
     base_dir = state["base_dir"]
     trimmed = state["trimmed_path"]
 
-
-
     try:
         # capture both stdout & stderr
         p =     subprocess.run([
@@ -105,7 +112,7 @@ def step2_preprocess(state):
     except subprocess.CalledProcessError as e:
         # e.stdout contains everything
         err = f"❌ Preprocess failed (exit {e.returncode}).\n\n{e.stdout}"
-        raise RuntimeError(err)
+        return err, None, state
 
     crop_dir = os.path.join(base_dir, "cropped")
     out = os.path.join(os.path.dirname(trimmed), f"crop_{session_id}.mp4")
@@ -127,7 +134,7 @@ def step3_normals(state):
     except subprocess.CalledProcessError as e:
         # e.stdout contains everything
         err = f"❌ Normal map failed (exit {e.returncode}).\n\n{e.stdout}"
-        raise RuntimeError(err)
+        return err, None, state
         
     normals_dir = os.path.join(base_dir, "p3dmm", "normals")
     out = os.path.join(os.path.dirname(state["trimmed_path"]), f"normals_{session_id}.mp4")
@@ -149,7 +156,7 @@ def step4_uv_map(state):
     except subprocess.CalledProcessError as e:
         # e.stdout contains everything
         err = f"❌ UV map failed (exit {e.returncode}).\n\n{e.stdout}"
-        raise RuntimeError(err)
+        return err, None, state
 
     uv_dir = os.path.join(base_dir, "p3dmm", "uv_map")
     out = os.path.join(os.path.dirname(state["trimmed_path"]), f"uv_map_{session_id}.mp4")
@@ -171,7 +178,7 @@ def step5_track(state):
     except subprocess.CalledProcessError as e:
         # e.stdout contains everything
         err = f"❌ Tracking failed (exit {e.returncode}).\n\n{e.stdout}"
-        raise RuntimeError(err)
+        return err, None, state
 
     # if we get here, it succeeded:
     tracking_dir = os.path.join(os.environ["PIXEL3DMM_TRACKING_OUTPUT"], session_id, "frames")
@@ -199,17 +206,26 @@ with demo:
             with gr.Row():
                 uv_vid      = gr.Video(label="UV Map", height=256)
                 track_vid   = gr.Video(label="Tracking", height=256)
-    run_btn = gr.Button("Run Pipeline")
+    run_btn_1 = gr.Button("Run Pipeline 1")
+    run_btn_2 = gr.Button("Run Pipeline 2")
+    run_btn_3 = gr.Button("Run Pipeline 3")
+    run_btn_4 = gr.Button("Run Pipeline 4")
+    run_btn_5 = gr.Button("Run Pipeline 5")
 
     # Update sliders after video upload
     video_in.change(fn=get_video_info, inputs=video_in, outputs=[seconds_slider, fps_slider])
 
     # Pipeline execution
-    (run_btn.click(fn=step1_trim, inputs=[video_in, seconds_slider, fps_slider, state], outputs=[status, state])
-         .then(fn=step2_preprocess, inputs=[state], outputs=[status, crop_vid, state])
-         .then(fn=step3_normals,    inputs=[state], outputs=[status, normals_vid, state])
-         .then(fn=step4_uv_map,     inputs=[state], outputs=[status, uv_vid, state])
-         .then(fn=step5_track,      inputs=[state], outputs=[status, track_vid, state])
+    run_btn_1.click(fn=step1_trim, inputs=[video_in, seconds_slider, fps_slider, state], outputs=[status, state])
+    run_btn_2.click(fn=step2_preprocess, inputs=[state], outputs=[status, crop_vid, state])
+    run_btn_3.click(fn=step3_normals, inputs=[state], outputs=[status, normals_vid, state])
+    run_btn_4.click(fn=step4_uv_map, inputs=[state], outputs=[status, uv_vid, state])
+    run_btn_5.click(fn=step5_track, inputs=[state], outputs=[status, track_vid, state])
+    
+         # .then(fn=step2_preprocess, inputs=[state], outputs=[status, crop_vid, state])
+         # .then(fn=step3_normals,    inputs=[state], outputs=[status, normals_vid, state])
+         # .then(fn=step4_uv_map,     inputs=[state], outputs=[status, uv_vid, state])
+         # .then(fn=step5_track,      inputs=[state], outputs=[status, track_vid, state])
     )
 
 # ------------------------------------------------------------------