Initial GeoIntel V1 foundation
This commit is contained in:
@@ -0,0 +1,52 @@
|
||||
from __future__ import annotations
|
||||
|
||||
import argparse
|
||||
import json
|
||||
import sys
|
||||
from pathlib import Path
|
||||
|
||||
|
||||
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 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("--json", action="store_true", help="Print JSON output only.")
|
||||
args = parser.parse_args()
|
||||
|
||||
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,
|
||||
)
|
||||
|
||||
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
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
raise SystemExit(main())
|
||||
Reference in New Issue
Block a user