fix bug

app.py

@@ -1,6 +1,7 @@
 import os
 import shutil
 import tempfile
+import traceback
 import uuid
 from pathlib import Path
 from typing import Optional
@@ -213,7 +214,99 @@ def _export_glb_from_points(xyz: np.ndarray, rgb: np.ndarray, output_path: Path)
     cloud.export(output_path.as_posix())
 
 
+def _run_demo_impl(
+    image: np.ndarray,
+    depth_file,
+    model_type: str,
+    input_size: str,
+    output_resolution_mode: str,
+    upsample_ratio: int,
+    max_points_preview: int,
+    fx_org: Optional[float],
+    fy_org: Optional[float],
+    cx_org: Optional[float],
+    cy_org: Optional[float],
+):
+    if image is None:
+        raise ValueError("Input RGB image is required")
+
+    depth_path = None
+    if depth_file is not None:
+        depth_path = depth_file if isinstance(depth_file, str) else depth_file.name
+
+    result = run_single_image_demo(
+        image_np=image,
+        depth_path=depth_path,
+        model_type=model_type,
+        input_size_text=input_size,
+        output_resolution_mode=output_resolution_mode,
+        upsample_ratio=int(upsample_ratio),
+        max_points_preview=int(max_points_preview),
+        fx_org=_none_if_invalid(fx_org),
+        fy_org=_none_if_invalid(fy_org),
+        cx_org=_none_if_invalid(cx_org),
+        cy_org=_none_if_invalid(cy_org),
+        model_cache=MODEL_CACHE,
+    )
+    output_dir = _prepare_output_dir()
+    glb_path = output_dir / "pointcloud.glb"
+    ply_path = output_dir / "pointcloud.ply"
+    depth_vis_path = output_dir / "depth_colorized.png"
+
+    _export_glb_from_points(result.xyz, result.rgb, glb_path)
+    if os.path.exists(result.ply_path):
+        shutil.copy2(result.ply_path, ply_path)
+    depth_vis_uint8 = result.depth_vis if result.depth_vis.dtype == np.uint8 else result.depth_vis.astype(np.uint8)
+    Image.fromarray(depth_vis_uint8).save(depth_vis_path)
+
+    depth_npy_path = output_dir / "depth.npy"
+    if os.path.exists(result.depth_npy_path):
+        shutil.copy2(result.depth_npy_path, depth_npy_path)
+
+    download_files = [glb_path.as_posix(), depth_vis_path.as_posix()]
+    if depth_npy_path.exists():
+        download_files.append(depth_npy_path.as_posix())
+    if ply_path.exists():
+        download_files.append(ply_path.as_posix())
+
+    status = (
+        f"Done. Output depth: {result.output_size[0]}x{result.output_size[1]}. "
+        f"Depth source: {result.depth_source_label}. "
+        f"Preview points: {result.xyz.shape[0]}. "
+        f"Camera intrinsics: {'default' if result.used_default_intrinsics else 'custom'}."
+    )
+    return result.depth_vis, glb_path.as_posix(), download_files, status
+
+
 @spaces.GPU(duration=180)
+def run_demo_gpu(
+    image: np.ndarray,
+    depth_file,
+    model_type: str,
+    input_size: str,
+    output_resolution_mode: str,
+    upsample_ratio: int,
+    max_points_preview: int,
+    fx_org: Optional[float],
+    fy_org: Optional[float],
+    cx_org: Optional[float],
+    cy_org: Optional[float],
+):
+    return _run_demo_impl(
+        image=image,
+        depth_file=depth_file,
+        model_type=model_type,
+        input_size=input_size,
+        output_resolution_mode=output_resolution_mode,
+        upsample_ratio=upsample_ratio,
+        max_points_preview=max_points_preview,
+        fx_org=fx_org,
+        fy_org=fy_org,
+        cx_org=cx_org,
+        cy_org=cy_org,
+    )
+
+
 def run_demo(
     image: np.ndarray,
     depth_file,
@@ -228,71 +321,40 @@ def run_demo(
     cy_org: Optional[float],
 ):
     request_id = uuid.uuid4().hex[:8]
+    depth_path = None
+    if depth_file is not None:
+        depth_path = depth_file if isinstance(depth_file, str) else depth_file.name
+    image_shape = tuple(int(dim) for dim in image.shape) if image is not None else None
+    Log.info(
+        f"[{request_id}] run_demo start: model_type={model_type}, "
+        f"input_size={input_size}, output_resolution_mode={output_resolution_mode}, "
+        f"upsample_ratio={upsample_ratio}, max_points_preview={max_points_preview}, "
+        f"depth_path={depth_path}, image_shape={image_shape}"
+    )
     try:
-        if image is None:
-            raise ValueError("Input RGB image is required")
-
-        depth_path = None
-        if depth_file is not None:
-            depth_path = depth_file if isinstance(depth_file, str) else depth_file.name
-
-        image_shape = tuple(int(dim) for dim in image.shape) if image is not None else None
-        Log.info(
-            f"[{request_id}] run_demo start: model_type={model_type}, "
-            f"input_size={input_size}, output_resolution_mode={output_resolution_mode}, "
-            f"upsample_ratio={upsample_ratio}, max_points_preview={max_points_preview}, "
-            f"depth_path={depth_path}, image_shape={image_shape}"
-        )
-
-        result = run_single_image_demo(
-            image_np=image,
-            depth_path=depth_path,
+        result = run_demo_gpu(
+            image=image,
+            depth_file=depth_file,
             model_type=model_type,
-            input_size_text=input_size,
+            input_size=input_size,
             output_resolution_mode=output_resolution_mode,
-            upsample_ratio=int(upsample_ratio),
-            max_points_preview=int(max_points_preview),
-            fx_org=_none_if_invalid(fx_org),
-            fy_org=_none_if_invalid(fy_org),
-            cx_org=_none_if_invalid(cx_org),
-            cy_org=_none_if_invalid(cy_org),
-            model_cache=MODEL_CACHE,
-        )
-        output_dir = _prepare_output_dir()
-        glb_path = output_dir / "pointcloud.glb"
-        ply_path = output_dir / "pointcloud.ply"
-        depth_vis_path = output_dir / "depth_colorized.png"
-
-        _export_glb_from_points(result.xyz, result.rgb, glb_path)
-        if os.path.exists(result.ply_path):
-            shutil.copy2(result.ply_path, ply_path)
-        depth_vis_uint8 = result.depth_vis if result.depth_vis.dtype == np.uint8 else result.depth_vis.astype(np.uint8)
-        Image.fromarray(depth_vis_uint8).save(depth_vis_path)
-
-        depth_npy_path = output_dir / "depth.npy"
-        if os.path.exists(result.depth_npy_path):
-            shutil.copy2(result.depth_npy_path, depth_npy_path)
-
-        download_files = [glb_path.as_posix(), depth_vis_path.as_posix()]
-        if depth_npy_path.exists():
-            download_files.append(depth_npy_path.as_posix())
-        if ply_path.exists():
-            download_files.append(ply_path.as_posix())
-
-        status = (
-            f"Done. Output depth: {result.output_size[0]}x{result.output_size[1]}. "
-            f"Depth source: {result.depth_source_label}. "
-            f"Preview points: {result.xyz.shape[0]}. "
-            f"Camera intrinsics: {'default' if result.used_default_intrinsics else 'custom'}."
-        )
-        Log.info(
-            f"[{request_id}] run_demo success: output_dir={output_dir}, output_size={result.output_size}, "
-            f"depth_source={result.depth_source_label}, preview_points={result.xyz.shape[0]}"
+            upsample_ratio=upsample_ratio,
+            max_points_preview=max_points_preview,
+            fx_org=fx_org,
+            fy_org=fy_org,
+            cx_org=cx_org,
+            cy_org=cy_org,
         )
-        return result.depth_vis, glb_path.as_posix(), download_files, status
+        status_text = result[3] if isinstance(result, tuple) and len(result) == 4 else "Done."
+        Log.info(f"[{request_id}] run_demo success: {status_text}")
+        return result
     except Exception as exc:
+        error_trace = traceback.format_exc()
         Log.exception(f"[{request_id}] run_demo failed")
-        return None, None, [], f"Error [{request_id}]: {exc}"
+        error_message = f"Error [{request_id}]: {exc}"
+        if os.getenv("INFINIDEPTH_SHOW_TRACEBACK", "0") == "1":
+            error_message = f"{error_message}\n\n{error_trace}"
+        return None, None, [], error_message
 
 
 DESCRIPTION_MD = """