from __future__ import annotations

from PIL import Image

from dataflow.enums import DataTypeId
from dataflow.types import DataType
from nodes import utils

TextType = DataType[str](
    id=DataTypeId.TEXT,
    name="Text",
    py_type=str,
    color="#eab308",
)

ImageType = DataType(
    id=DataTypeId.IMAGE,
    name="Image",
    py_type=Image.Image,
    encode=utils.encode_image,
    decode=utils.decode_image,
    color="#a855f7",
)