|
from pydantic import BaseModel, Field |
|
from typing import Optional |
|
import numpy as np |
|
from PIL import Image |
|
import io |
|
import base64 |
|
import uuid |
|
|
|
from behavior.class_behavior import Behaviors |
|
from circumstances.class_circumstance import Circumstances |
|
from physical.class_physical import PhysicalAnomalies |
|
from follow_up.class_follow_up import FollowUpEvents |
|
from geolocalisation.class_geolocalisation import Geolocalisation |
|
|
|
class Wounded(BaseModel): |
|
circumstances: Circumstances |
|
behaviors: Behaviors |
|
physical_anomalies: PhysicalAnomalies |
|
follow_up_events: FollowUpEvents |
|
|
|
class Dead(BaseModel): |
|
circumstances: Circumstances |
|
physical_anomalies: PhysicalAnomalies |
|
follow_up_events: FollowUpEvents |
|
|
|
class ImageBase64(BaseModel): |
|
image: str |
|
@classmethod |
|
def to_base64(cls, pixel_data: list): |
|
img_array = np.array(pixel_data, dtype=np.uint8) |
|
img = Image.fromarray(img_array) |
|
|
|
buffer = io.BytesIO() |
|
img.save(buffer, format="PNG") |
|
buffer.seek(0) |
|
base64_str = base64.b64encode(buffer.read()).decode('utf-8') |
|
return cls(image=base64_str) |
|
|
|
class Report(BaseModel): |
|
identifier: str |
|
image: ImageBase64 |
|
image_md5: str |
|
geolocalisation: Geolocalisation |
|
wounded_state: str |
|
wounded: Optional[Wounded] = None |
|
dead_state: str |
|
dead: Optional[Dead] = None |
|
|
