Make YOLO preflight runnable in backend container

scripts/yolo_preflight.py

@@ -1,59 +1,22 @@
 from __future__ import annotations
 
-import argparse
-import json
-import sys
+import importlib.util
 from pathlib import Path
+from types import ModuleType
 
 
-ROOT = Path(__file__).resolve().parents[1]
-BACKEND_ROOT = ROOT / "backend"
-if str(BACKEND_ROOT) not in sys.path:
-    sys.path.insert(0, str(BACKEND_ROOT))
-
-from app.core.config import Settings  # noqa: E402
-from app.services.yolo_preflight_service import YoloPreflightService  # noqa: E402
+def load_backend_script() -> ModuleType:
+    script = Path(__file__).resolve().parents[1] / "backend" / "scripts" / "yolo_preflight.py"
+    spec = importlib.util.spec_from_file_location("backend_yolo_preflight", script)
+    if spec is None or spec.loader is None:
+        raise RuntimeError(f"Could not load YOLO preflight implementation from {script}")
+    module = importlib.util.module_from_spec(spec)
+    spec.loader.exec_module(module)
+    return module
 
 
-def main() -> int:
-    parser = argparse.ArgumentParser(description="Run local YOLO configuration preflight without loading a model.")
-    parser.add_argument("--model-path", help="Existing local YOLO model path.")
-    parser.add_argument("--tile-manifest-path", help="Existing raster tile manifest path.")
-    parser.add_argument("--enabled", action="store_true", help="Treat YOLO as enabled for this preflight.")
-    parser.add_argument("--max-tiles", type=int, default=100, help="Maximum tile count allowed by preflight.")
-    parser.add_argument(
-        "--assume-dependencies",
-        action="store_true",
-        help="Skip checking installed ultralytics/torch packages; useful for validating local paths on non-AI machines.",
-    )
-    parser.add_argument(
-        "--check-model-load",
-        action="store_true",
-        help="Explicitly load the configured local model file to verify Ultralytics compatibility; no inference is run.",
-    )
-    parser.add_argument("--json", action="store_true", help="Print JSON output only.")
-    args = parser.parse_args()
-    if args.check_model_load and args.assume_dependencies:
-        parser.error("--check-model-load cannot be combined with --assume-dependencies")
-
-    settings = Settings(
-        yolo_enabled=args.enabled or bool(args.model_path),
-        yolo_model_path=args.model_path,
-        yolo_max_tiles=args.max_tiles,
-    )
-    payload = YoloPreflightService.run(
-        settings=settings,
-        tile_manifest_path=args.tile_manifest_path,
-        assume_dependencies=args.assume_dependencies,
-        check_model_load=args.check_model_load,
-    )
-
-    if args.json:
-        print(json.dumps(payload, indent=2, sort_keys=True))
-    else:
-        print("GeoIntel YOLO preflight")
-        print(json.dumps(payload, indent=2, sort_keys=True))
-    return 0 if payload["status"] in {"ready", "not_configured", "dependency_unavailable"} else 1
+_impl = load_backend_script()
+main = _impl.main
 
 
 if __name__ == "__main__":