from typing import Any

from pydantic import BaseModel


class AiNormalizerResponse(BaseModel):
    is_real_estate: bool
    reason: str | None = None
    normalized: dict[str, Any] | None = None


class AiImageAnalysisResponse(BaseModel):
    overall_condition: str | None = None
    rooms_observed: int | None = None
    issues_found: list[str] = []
    positive_highlights: list[str] = []
    view_from_window: str | None = None
    furniture_included: bool | None = None
    appliances_included: list[str] = []


class AiEnrichmentResponse(BaseModel):
    extracted_features: dict[str, Any] = {}
    price_assessment: dict[str, Any] = {}
    listing_quality_score: int | None = None
    reliability_rating: int | None = None
    sentiment_score: float | None = None
    classification: str | None = None
    image_analysis_results: dict[str, Any] = {}
    generated_description: str | None = None
    summary: str | None = None
    model_version: str | None = None
    processing_time_ms: int | None = None