from pydantic import BaseModel
from typing import Optional, List, Any
from datetime import datetime

class ScanRequest(BaseModel):
    image: str  # base64

class DocumentResponse(BaseModel):
    id: int
    title: Optional[str]
    category: Optional[str]
    extracted_text: Optional[str]
    summary: Optional[str]
    extracted_data: Optional[Any]
    risk_alerts: Optional[Any]
    tags: Optional[Any]
    file_size: Optional[int]
    created_at: datetime

class DocumentListResponse(BaseModel):
    documents: List[DocumentResponse]
    total: int