Refactor and clean up codebase by removing unused functions and reorganizing imports

- Removed unused model keys and functions from properties.py.
- Deleted projection.py as its functions were no longer needed.
- Cleaned up model_catalog.py by removing unused model directory function.
- Updated imports in various test files to reflect new module structure.
- Removed deprecated test cases related to image editing tools.
- Added new tests for ONNX runtime provider selection logic.

src/anyimage/server/geometry/__init__.py

@@ -1,7 +1 @@
-from .projection import unproject_camera_depth, unproject_depth_frame
-
-
-__all__ = (
-    "unproject_camera_depth",
-    "unproject_depth_frame",
-)
+"""Depth and normal artifact processing."""

src/anyimage/server/geometry/projection.py

@@ -1,31 +0,0 @@
-def unproject_depth_frame(depth, intrinsics, extrinsics):
-    import numpy as np
-
-    height, width = depth.shape
-    x, y = np.meshgrid(
-        np.arange(width, dtype=np.float32),
-        np.arange(height, dtype=np.float32),
-    )
-    pixels = np.stack((x, y, np.ones_like(x)), axis=-1).reshape(-1, 3)
-    rays = (np.linalg.inv(intrinsics) @ pixels.T).T
-    camera_points = rays * depth.reshape(-1, 1)
-    camera_points = np.column_stack(
-        (camera_points, np.ones(camera_points.shape[0], dtype=np.float32))
-    )
-    world_to_camera = np.eye(4, dtype=np.float32)
-    world_to_camera[: extrinsics.shape[0], :4] = extrinsics
-    world_points = (np.linalg.inv(world_to_camera) @ camera_points.T).T
-    return world_points[:, :3].reshape(height, width, 3)
-
-
-def unproject_camera_depth(depth, intrinsics):
-    import numpy as np
-
-    height, width = depth.shape
-    x, y = np.meshgrid(
-        np.arange(width, dtype=np.float32),
-        np.arange(height, dtype=np.float32),
-    )
-    pixels = np.stack((x, y, np.ones_like(x)), axis=-1).reshape(-1, 3)
-    rays = (np.linalg.inv(intrinsics) @ pixels.T).T
-    return (rays * depth.reshape(-1, 1)).reshape(height, width, 3)

src/anyimage/server/model_catalog.py

@@ -178,7 +178,6 @@ MOGE2_MODELS = {
         ),),
     ),
 }
-MOGE2_MODEL = MOGE2_MODELS["MOGE2_VITS_NORMAL"]
 DOWNLOADABLE_MODELS = {
     BEN2_MODEL.key: BEN2_MODEL,
     **UPSCALE_MODELS,
@@ -193,11 +192,6 @@ def get_downloadable_model(model_key):
         raise ValueError(f"Unknown model: {model_key}") from error
 
 
-def model_directory_name(model_key):
-    """Return the model's folder name inside the shared models directory."""
-    return get_downloadable_model(model_key).directory_name
-
-
 def model_record(model, ready=False):
     """Return public model metadata exposed by the Models Resource."""
     return {

src/anyimage/server/models/onnx_moge2.py

@@ -4,7 +4,6 @@ from pathlib import Path
 
 from .onnx_runtime import create_session as create_runtime_session
 from .onnx_runtime import run_session
-from .onnx_runtime import select_providers as select_providers
 
 
 ONNX_FILENAME = "model.onnx"
@@ -15,10 +14,6 @@ COREML_PROVIDER_OPTIONS = {
 }
 
 
-def is_onnx_model(model_directory):
-    return (Path(model_directory) / ONNX_FILENAME).is_file()
-
-
 def create_session(model_directory, requested_device="auto"):
     path = Path(model_directory) / ONNX_FILENAME
     return create_runtime_session(

src/anyimage/server/models/onnx_runtime.py

@@ -95,15 +95,3 @@ def run_session(
         "gpu:0",
     )
     return session.run(output_names, input_feed, run_options=run_options)
-
-
-def provider_label(session):
-    provider = session.get_providers()[0]
-    labels = {
-        "CUDAExecutionProvider": "CUDA",
-        "CoreMLExecutionProvider": "CoreML",
-        "DmlExecutionProvider": "DirectML",
-        "OpenVINOExecutionProvider": "OpenVINO",
-        "CPUExecutionProvider": "CPU",
-    }
-    return labels.get(provider, provider)