from typing import List, Optional, Dict, Any
from pydantic import BaseModel


class ObjectResult(BaseModel):
    bbox_xyxy: List[float]
    det_confidence: float
    det_class_id: int
    det_class_name: str
    cls_class_id: int
    cls_class_name: str
    cls_confidence: float


class PredictResponse(BaseModel):
    success: bool
    image_width: int
    image_height: int
    num_objects: int
    results: List[ObjectResult]
    annotated_image_base64: Optional[str] = None
    saved_outputs: Optional[Dict[str, Any]] = None
    message: str = "ok"