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

from behavior.class_behavior import Behaviors
from circumstances.class_circumstance import Circumstances
from physical.class_physical import PhysicalAnomaly
from follow_up.class_follow_up import FollowUpEvents
from geolocalisation.class_geolocalisation import Geolocalisation

class Wounded(BaseModel):
    circumstances: List[Circumstances] 
    behaviors: List[Behaviors]
    physical_anomalies: List[PhysicalAnomaly]
    follow_up_events: List[FollowUpEvents]

class Dead(BaseModel):
    circumstances: List[Circumstances] 
    physical_anomalies: List[PhysicalAnomaly]
    follow_up_events: List[FollowUpEvents]

class Image(BaseModel):
    image: List[float]

class Report(BaseModel):
    image: Image
    geolocalisation: Geolocalisation
    wounded_state: bool
    wounded: Optional[Wounded] = None  
    dead_state: bool
    dead: Optional[Dead] = None 
    
# Example usage
# json_data = {
#     "circumstance": "COLLISION",
#     "circumstance_radio": "Yes",
#     "circumstance_type": {
#         "type": "Train",
#         "infrastructure_number": "56"
#     }
# }
# circumstance_instance = Circumstance(**json_data)
# circumstance_schema = Circumstance.schema_json(indent=2)