from pydantic import BaseModel
from typing import List, Optional


class Detection(BaseModel):
    """Object detection result"""
    label: str
    confidence: float
    x: int
    y: int
    width: int
    height: int
    color: int = 0x00FF00  # Default green color


class DetectionRequest(BaseModel):
    """Request for model inference on image data"""
    model_name: str
    image_data: str  # Base64 encoded image


class DetectionResponse(BaseModel):
    """Response with detection results"""
    model_name: str
    detections: List[Detection]
    inference_time: float  # Time in milliseconds


class ModelInfo(BaseModel):
    """Model information"""
    name: str
    path: str
    size: List[int]  # [width, height]
    backend: str = "ONNX"
    loaded: bool = False


class ModelsResponse(BaseModel):
    """Response with available models"""
    models: List[ModelInfo]