update

app.py

@@ -220,12 +220,17 @@ def _append_trace(trace_path: str, stage: str) -> None:
         handle.flush()
 
 
-def _read_trace(trace_path: str, max_lines: int = 12) -> str:
+def _read_trace(trace_path: str) -> str:
     trace_file = Path(trace_path)
     if not trace_file.exists():
         return ""
-    lines = trace_file.read_text(encoding="utf-8").splitlines()
-    return "\n".join(lines[-max_lines:])
+    return trace_file.read_text(encoding="utf-8").rstrip()
+
+
+def _should_show_full_traceback() -> bool:
+    if SPACE_RUNTIME:
+        return True
+    return os.getenv("INFINIDEPTH_SHOW_TRACEBACK", "0") == "1"
 
 
 def _export_glb_from_points(xyz: np.ndarray, rgb: np.ndarray, output_path: Path) -> None:
@@ -368,7 +373,7 @@ def run_demo(
         return result.depth_vis, glb_path.as_posix(), download_files, status
     except Exception as exc:
         error_trace = traceback.format_exc()
-        trace_summary = _read_trace(trace_path)
+        trace_content = _read_trace(trace_path)
         Log.exception(f"[{request_id}] run_demo failed")
 
         # Classify the error for clearer diagnostics
@@ -387,13 +392,13 @@ def run_demo(
         else:
             error_message = f"Error [{request_id}] ({exc_type}): {exc}"
 
-        if trace_summary:
-            error_message = f"{error_message}\n\nLast worker stages:\n{trace_summary}"
-        if os.getenv("INFINIDEPTH_SHOW_TRACEBACK", "0") == "1":
+        if trace_content:
+            error_message = f"{error_message}\n\nWorker trace:\n{trace_content}"
+        if _should_show_full_traceback():
             error_message = f"{error_message}\n\n{error_trace}"
         # Always log full traceback to server logs (visible in HF Space Logs tab)
         Log.error(f"[{request_id}] trace_path={trace_path}")
-        Log.error(f"[{request_id}] Last worker stages:\n{trace_summary or '<none>'}")
+        Log.error(f"[{request_id}] Worker trace:\n{trace_content or '<none>'}")
         Log.error(f"[{request_id}] Full traceback:\n{error_trace}")
         return None, None, [], error_message