92 lines
2.8 KiB
Python
92 lines
2.8 KiB
Python
"""Annotation and review-state routes (REQ-040…045)."""
|
|
|
|
from typing import Any, Dict, List, Optional
|
|
|
|
from fastapi import APIRouter, HTTPException
|
|
from pydantic import BaseModel
|
|
|
|
from backend import batches as batch_store
|
|
from backend import review as review_store
|
|
|
|
router = APIRouter(tags=["review"])
|
|
|
|
|
|
class AnnotationRequest(BaseModel):
|
|
class_id: int = 0
|
|
geometry: Dict[str, Any]
|
|
|
|
|
|
class AnnotationPatch(BaseModel):
|
|
class_id: Optional[int] = None
|
|
geometry: Optional[Dict[str, Any]] = None
|
|
|
|
|
|
class StatusRequest(BaseModel):
|
|
status: str
|
|
|
|
|
|
class AssistRequest(BaseModel):
|
|
box: List[float]
|
|
class_id: int = 0
|
|
threshold: float = 0.5
|
|
|
|
|
|
@router.get("/api/frames/{frame_id}/annotations")
|
|
def list_annotations(frame_id: int) -> dict:
|
|
target = review_store.frame(frame_id)
|
|
if target is None:
|
|
raise HTTPException(404, "No such frame")
|
|
return {
|
|
"frame": {
|
|
"id": target["id"], "batch_id": target["batch_id"], "idx": target["idx"],
|
|
"width": target["width"], "height": target["height"],
|
|
"review_status": target["review_status"], "label_type": target["label_type"],
|
|
},
|
|
"annotations": review_store.listing(frame_id),
|
|
}
|
|
|
|
|
|
@router.post("/api/frames/{frame_id}/annotations")
|
|
def add_annotation(frame_id: int, request: AnnotationRequest) -> dict:
|
|
try:
|
|
return review_store.add(frame_id, request.class_id, request.geometry)
|
|
except review_store.ReviewError as exc:
|
|
raise HTTPException(400, str(exc))
|
|
|
|
|
|
@router.patch("/api/annotations/{annotation_id}")
|
|
def patch_annotation(annotation_id: int, request: AnnotationPatch) -> dict:
|
|
try:
|
|
return review_store.update(annotation_id, request.class_id, request.geometry)
|
|
except review_store.ReviewError as exc:
|
|
raise HTTPException(400, str(exc))
|
|
|
|
|
|
@router.delete("/api/annotations/{annotation_id}")
|
|
def delete_annotation(annotation_id: int) -> dict:
|
|
return {"deleted": review_store.delete(annotation_id)}
|
|
|
|
|
|
@router.post("/api/frames/{frame_id}/assist")
|
|
def assist(frame_id: int, request: AssistRequest) -> dict:
|
|
try:
|
|
return review_store.assist(frame_id, request.box, request.class_id,
|
|
request.threshold)
|
|
except review_store.ReviewError as exc:
|
|
raise HTTPException(400, str(exc))
|
|
|
|
|
|
@router.post("/api/frames/{frame_id}/status")
|
|
def set_status(frame_id: int, request: StatusRequest) -> dict:
|
|
try:
|
|
return review_store.set_status(frame_id, request.status)
|
|
except review_store.ReviewError as exc:
|
|
raise HTTPException(400, str(exc))
|
|
|
|
|
|
@router.get("/api/batches/{batch_id}/next-pending")
|
|
def next_pending(batch_id: int, after_idx: int = -1) -> dict:
|
|
if batch_store.get(batch_id) is None:
|
|
raise HTTPException(404, "No such batch")
|
|
return {"frame_id": review_store.next_pending(batch_id, after_idx)}
|