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


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


# --- Specific Behavior classes ---
class AbnormalBreathing(Behavior):
    type: Literal["abnormal breathing"]
    description: Optional[Literal["Problems breathing, breathing sounds"]] = None


class CrashFalling(Behavior):
    type: Literal["crash, falling from the sky"]
    description: Optional[Literal["Suddenly falling from the sky"]] = None


class Diarrhea(Behavior):
    type: Literal["diarrhea"]
    description: Optional[Literal["Observed diarrhea"]] = None


class Lameness(Behavior):
    type: Literal["lameness"]
    description: Optional[
        Literal["Apparent limping or not able to walk properly"]
    ] = None


class Neurological(Behavior):
    type: Literal["neurological"]
    description: Optional[
        Literal["Circling, incoordination, tremors, convulsions, fast eye movements"]
    ] = None


class OtherAbnormalBehavior(Behavior):
    type: Literal["other abnormal behavior"]
    description: Optional[Literal["Other than weakness, other than neurologic"]] = None


class UnableToFly(Behavior):
    type: Literal["unable to fly"]
    description: Optional[
        Literal["Animal alert and tries to fly but can not take off"]
    ] = None


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


class Weakness(Behavior):
    type: Literal["weakness"]
    description: Optional[
        Literal["Non responsive, does not fly away when approached, lethargy"]
    ] = None


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


# Union of all possible behaviors
BehaviorType = Union[
    AbnormalBreathing,
    CrashFalling,
    Diarrhea,
    Lameness,
    Neurological,
    OtherAbnormalBehavior,
    UnableToFly,
    Vomiting,
    Weakness,
    NoChanges,
]


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