from pydantic import BaseModel, Field
from typing import Literal, List, Union, Optional


class BehaviorSimple(BaseModel):
    type: str
    description: Optional[str] = None  # Making the description field optional


# --- Specific BehaviorSimple classes ---
class GeneralWeakness(BehaviorSimple):
    type: Literal["general weakness"]
    description: Optional[
        Literal[
            "Abnormal breathing (dyspnoea), sudden crash, apathy, lethargy, unable to fly but responsive"
        ]
    ] = None


class Vomiting(BehaviorSimple):
    type: Literal["vomiting"]
    description: Optional[Literal["Throwing up undigested food, regurgitating"]] = None


class AtypicalBehavior(BehaviorSimple):
    type: Literal["atypical behavior"]
    description: Optional[
        Literal["Circling, incoordination, tremors, convulsions"]
    ] = None


class NoChanges(BehaviorSimple):
    type: Literal["no changes"]
    description: Optional[Literal["Animal is acting normally"]] = None


# Union of all possible behaviors
BehaviorSimpleType = Union[GeneralWeakness, Vomiting, AtypicalBehavior, NoChanges]


# Main class that logs multiple behaviors
class BehaviorsSimple(BaseModel):
    behaviors_radio: str  # e.g., "Yes"
    behaviors_type: Optional[List[BehaviorSimpleType]] = None