File size: 441 Bytes
691f45a 3025bb3 691f45a | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 | 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",
)
|