local_segment/api/models.py

from pydantic import BaseModel
from typing import List, Optional

class PointCoords(BaseModel):
    x: int
    y: int

class BoxCoords(BaseModel):
    x1: int
    y1: int
    x2: int
    y2: int

class SegmentationRequest(BaseModel):
    image_url: Optional[str] = None
    points: List[list[int,int]] = []
    labels: List[int] = []
    # type:int = 0 #0-sam1 1-sam2 2-sam3
    

class SegmentationWithBoxRequest(BaseModel):
    image_url: Optional[str] = None
    box: BoxCoords
    multimask_output: bool = True

class SegmentationResponse(BaseModel):
    success: bool
    results: List[dict]
    processing_time: float
    total_masks: int