Update app.py

app.py

@@ -94,10 +94,18 @@ def step2_preprocess(state):
     base_dir = state["base_dir"]
     trimmed = state["trimmed_path"]
 
-    subprocess.run([
+
+
+    try:
+        # capture both stdout & stderr
+        p =     subprocess.run([
         "python", "scripts/run_preprocessing.py",
         "--video_or_images_path", trimmed
     ], 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}"
+        raise RuntimeError(err)
 
     crop_dir = os.path.join(base_dir, "cropped")
     out = os.path.join(os.path.dirname(trimmed), f"crop_{session_id}.mp4")
@@ -110,11 +118,17 @@ def step3_normals(state):
     session_id = state["session_id"]
     base_dir = state["base_dir"]
 
-    subprocess.run([
+    try:
+        # capture both stdout & stderr
+        p = subprocess.run([
         "python", "scripts/network_inference.py",
         "model.prediction_type=normals", f"video_name={session_id}"
     ], check=True, stdout=subprocess.PIPE, stderr=subprocess.STDOUT)
-
+    except subprocess.CalledProcessError as e:
+        # e.stdout contains everything
+        err = f"❌ Normal map failed (exit {e.returncode}).\n\n{e.stdout}"
+        raise RuntimeError(err)
+        
     normals_dir = os.path.join(base_dir, "p3dmm", "normals")
     out = os.path.join(os.path.dirname(state["trimmed_path"]), f"normals_{session_id}.mp4")
     video = make_video_from_frames(normals_dir, out)
@@ -126,10 +140,16 @@ def step4_uv_map(state):
     session_id = state["session_id"]
     base_dir = state["base_dir"]
 
-    subprocess.run([
+    try:
+        # capture both stdout & stderr
+        p = subprocess.run([
         "python", "scripts/network_inference.py",
         "model.prediction_type=uv_map", f"video_name={session_id}"
     ], check=True, stdout=subprocess.PIPE, stderr=subprocess.STDOUT)
+    except subprocess.CalledProcessError as e:
+        # e.stdout contains everything
+        err = f"❌ UV map failed (exit {e.returncode}).\n\n{e.stdout}"
+        raise RuntimeError(err)
 
     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")
@@ -151,7 +171,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}"
-        return err, None, state
+        raise RuntimeError(err)
 
     # if we get here, it succeeded:
     tracking_dir = os.path.join(os.environ["PIXEL3DMM_TRACKING_OUTPUT"], session_id, "frames")