from pydantic import BaseModel
from enum import IntEnum, Enum

class WhitePiece(IntEnum):
    WhiteRook1 = 0
    WhiteKnight1 = 1
    WhiteBishop1 = 2
    WhiteQueen = 3
    WhiteKing = 4
    WhiteBishop2 = 5
    WhiteKnight2 = 6
    WhiteRook2 = 7
    WhitePawn1 = 8
    WhitePawn2 = 9
    WhitePawn3 = 10
    WhitePawn4 = 11
    WhitePawn5 = 12
    WhitePawn6 = 13
    WhitePawn7 = 14
    WhitePawn8 = 15

class SpeechConfig(BaseModel):
    voice_id: str

class SpeakVelocity(str, Enum):
    SLOW = "slow"
    MEDIUM = "medium"
    HIGH = "fast"

class ChessPiece(BaseModel):
    id: WhitePiece
    name: str
    personality: str
    speech_config: SpeechConfig
    speak_velocity: SpeakVelocity