from pydantic import BaseModel
from .frame_result import FrameResult


class FlightStatistics(BaseModel):
    total_frames: int
    processed_frames: int
    refined_frames: int
    mean_confidence: float
    processing_time: float


class FlightResults(BaseModel):
    flight_id: str
    frames: list[FrameResult]
    statistics: FlightStatistics