"""Data models for pose classification API."""

from pydantic import BaseModel, Field


class PoseClassificationRequest(BaseModel):
    """Request body for pose classification endpoint."""
    url: str = Field(
        description="Image URL for classification"
    )


class PosePrediction(BaseModel):
    """Single pose prediction result."""
    label: str
    score: float


class PoseClassificationResponse(BaseModel):
    """Response body for pose classification endpoint."""
    model_name: str = "vit-human-pose-classification"
    prediction: PosePrediction
    prediction_time_ms: int = Field(
        description="Time taken for inference in milliseconds"
    )