| 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", | |
| ) | |