43 lines
1.7 KiB
Python
43 lines
1.7 KiB
Python
from __future__ import annotations
|
|
|
|
from fastapi import APIRouter, Depends
|
|
from sqlalchemy.orm import Session
|
|
|
|
from app.core.errors import AppError
|
|
from app.db.session import get_db
|
|
from app.models import Dataset
|
|
from app.schemas import Envelope, JobRead
|
|
from app.schemas.analysis import ChangeDetectionRequest
|
|
from app.services.change_detection_service import ChangeDetectionService
|
|
from app.services.job_service import JobService
|
|
from app.utils.response import envelope
|
|
|
|
router = APIRouter(prefix="/analysis", tags=["analysis"])
|
|
|
|
|
|
@router.post("/change-detection", response_model=Envelope[JobRead])
|
|
def run_change_detection(
|
|
payload: ChangeDetectionRequest,
|
|
db: Session = Depends(get_db),
|
|
) -> dict:
|
|
source_dataset = db.get(Dataset, payload.source_dataset_id)
|
|
if not source_dataset:
|
|
raise AppError(code="DATASET_NOT_FOUND", message="Source dataset not found", status_code=404)
|
|
ChangeDetectionService._get_project_vector_dataset(db, payload.source_dataset_id, source_dataset.project_id, "Source")
|
|
job = JobService.run_sync_job(
|
|
db=db,
|
|
project_id=source_dataset.project_id,
|
|
job_type="analysis.change-detection",
|
|
parameters=payload.model_dump(mode="json"),
|
|
input_dataset_id=payload.source_dataset_id,
|
|
operation=lambda: ChangeDetectionService.compare_vector_datasets(
|
|
db=db,
|
|
project_id=source_dataset.project_id,
|
|
source_dataset_id=payload.source_dataset_id,
|
|
target_dataset_id=payload.target_dataset_id,
|
|
iou_threshold=payload.iou_threshold,
|
|
include_unchanged=payload.include_unchanged,
|
|
).model_dump(mode="json"),
|
|
)
|
|
return envelope(job)
|