from pydantic import BaseModel, Field


class CapitalImageRecordBase(BaseModel):
    id: str
    trade_date: str | None
    subject: str | None
    snapshot_time: str | None
    main_force_amount_yi: float | None
    institution_amount_yi: float | None
    large_household_amount_yi: float | None
    retail_amount_yi: float | None
    overall_trend: str | None
    intraday_summary: str | None
    review_status: str
    extraction_method: str
    image_name: str
    image_url: str
    created_at: str
    updated_at: str


class CapitalImageRecordDetail(CapitalImageRecordBase):
    raw_extraction: dict = Field(default_factory=dict)


class CapitalImageListResponse(BaseModel):
    items: list[CapitalImageRecordBase]
    total: int


class CapitalImageUploadResponse(BaseModel):
    item: CapitalImageRecordDetail