File size: 3,714 Bytes
edb09f2 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 | from PIL import Image, ImageOps, ImageFont, ImageDraw
import folder_paths
import numpy as np
import torch
import os
MAX_RESOLUTION = 32768
node_path = os.path.dirname(os.path.realpath(__file__))
base_path = os.path.dirname(node_path)
extras_dir = os.path.join(base_path, "extras")
folder_paths.folder_names_and_paths["chibi-fonts"] = (
[os.path.join(extras_dir, "fonts")],
{".ttf"},
)
class ImageAddText:
def __init__(self):
pass
@classmethod
def INPUT_TYPES(s):
return {
"required": {
"text": (
"STRING",
{"default": "Chibi-Nodes", "multiline": True},
),
"font": [sorted(folder_paths.get_filename_list("chibi-fonts"))],
"font_size": ("INT", {"default": 24, "min": 0, "max": 200, "step": 1}),
"font_colour": (["black", "white", "red", "green", "blue"],),
"invert_mask": ([False, True],),
"position_x": (
"INT",
{"default": 0, "min": 0, "max": MAX_RESOLUTION, "step": 1},
),
"position_y": (
"INT",
{"default": 0, "min": 0, "max": MAX_RESOLUTION, "step": 1},
),
"width": (
"INT",
{"default": 512, "min": 0, "max": MAX_RESOLUTION, "step": 1},
),
"height": (
"INT",
{"default": 512, "min": 0, "max": MAX_RESOLUTION, "step": 1},
),
},
"optional": {
"image": ("IMAGE",),
},
}
RETURN_TYPES = (
"IMAGE",
"MASK",
"STRING",
)
RETURN_NAMES = (
"IMAGE",
"MASK",
"text",
)
FUNCTION = "addtext"
CATEGORY = "Chibi-Nodes/Image"
def addtext(
self,
text,
width,
height,
font,
font_size,
position_x,
position_y,
font_colour,
invert_mask,
image=None,
):
if image is not None:
width = image.shape[2]
height = image.shape[1]
image = Image.fromarray(
np.clip(255.0 * image[0].cpu().numpy(),
0, 255).astype(np.uint8)
)
image = image.convert("RGBA")
else:
image = Image.new("RGBA", (width, height), (255, 255, 255, 0))
text_image = Image.new("RGBA", (width, height), (0, 255, 255, 0))
imaget = ImageDraw.Draw(
text_image,
)
msg = text
imaget.fontmode = "L"
fnt = ImageFont.truetype(
folder_paths.get_full_path("chibi-fonts", font), font_size
)
imaget.text((position_x, position_y), msg, font=fnt, fill=font_colour)
if "A" in text_image.getbands():
mask = np.array(text_image.getchannel(
"A")).astype(np.float32) / 255.0
mask = 1.0 - torch.from_numpy(mask)
else:
mask = torch.zeros((64, 64), dtype=torch.float32, device="cpu")
image.paste(text_image, (0, 0), text_image)
image = ImageOps.exif_transpose(image)
image = image.convert("RGB")
image = np.array(image).astype(np.float32) / 255.0
image = torch.from_numpy(image)[None,]
if invert_mask:
mask = 1.0 - mask
return (
image,
mask.unsqueeze(0),
text,
)
|