feat: add governed nationwide AOI orchestration and CUDA enforcement

backend/app/main.py

@@ -1,6 +1,7 @@
 from __future__ import annotations
 
 import logging
+import asyncio
 import re
 import time
 import uuid
@@ -11,7 +12,7 @@ from fastapi.exceptions import RequestValidationError
 from fastapi.middleware.cors import CORSMiddleware
 from fastapi.responses import JSONResponse
 
-from app.api.routes import analysis, areas, assistant, auth, datasets, demo, detection, exports, external, health, jobs, projects, qa, quality_checks, segmentation, selection_partitions, temporal
+from app.api.routes import analysis, aoi_operations, areas, assistant, auth, datasets, demo, detection, exports, external, health, jobs, projects, qa, quality_checks, segmentation, selection_partitions, temporal
 from app.core.config import get_settings
 from app.core.errors import AppError
 from app.core.logging import configure_logging
@@ -19,6 +20,7 @@ from app.core.request_context import reset_request_id, set_request_id
 from app.db.session import SessionLocal
 from app.services.runtime_reconciliation_service import RuntimeReconciliationService
 from app.services.auth_service import AuthService
+from app.services.aoi_operation_worker import AoiOperationWorker
 
 
 logger = logging.getLogger("geointel")
@@ -46,14 +48,18 @@ def create_app() -> FastAPI:
 
     @asynccontextmanager
     async def lifespan(_: FastAPI):
+        worker_stop = asyncio.Event()
+        worker_task = None
         if settings.reconcile_interrupted_runs_on_startup:
             db = SessionLocal()
             try:
                 result = RuntimeReconciliationService.reconcile(db)
                 logger.info(
-                    "Runtime reconciliation completed: jobs=%s analysis_runs=%s",
+                    "Runtime reconciliation completed: jobs=%s analysis_runs=%s resumed_aoi_partitions=%s exhausted_aoi_partitions=%s",
                     result.interrupted_jobs,
                     result.interrupted_analysis_runs,
+                    result.resumed_aoi_partitions,
+                    result.exhausted_aoi_partitions,
                 )
             except Exception:
                 db.rollback()
@@ -61,7 +67,14 @@ def create_app() -> FastAPI:
                 raise
             finally:
                 db.close()
-        yield
+        if settings.aoi_worker_enabled:
+            worker_task = asyncio.create_task(AoiOperationWorker.run(worker_stop, settings.aoi_worker_poll_seconds))
+        try:
+            yield
+        finally:
+            worker_stop.set()
+            if worker_task is not None:
+                await worker_task
 
     app = FastAPI(
         title="GeoIntel",
@@ -82,6 +95,7 @@ def create_app() -> FastAPI:
     app.include_router(health.router)
     app.include_router(auth.router, prefix=settings.api_prefix)
     app.include_router(analysis.router, prefix=settings.api_prefix)
+    app.include_router(aoi_operations.router, prefix=settings.api_prefix)
     app.include_router(projects.router, prefix=settings.api_prefix)
     app.include_router(areas.router, prefix=settings.api_prefix)
     app.include_router(datasets.router, prefix=settings.api_prefix)