-
Notifications
You must be signed in to change notification settings - Fork 3
/
Copy pathadd_to_metadata.py
37 lines (32 loc) · 1.11 KB
/
add_to_metadata.py
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
from nodes import LoadImage
import folder_paths
from PIL import Image
class AddToMetadata:
CATEGORY = "quicknodes/metadata"
@classmethod
def INPUT_TYPES(s):
return {"required": {
"image": ("IMAGE", {}),
"label": ("STRING",{"default":"prompt"}),
"value": ("STRING",{"default":""}),
},"hidden" : {
"extra_pnginfo": "EXTRA_PNGINFO",
}}
RETURN_TYPES = ("IMAGE",)
FUNCTION = "func"
def func(self, image, label, value, extra_pnginfo):
extra_pnginfo[label] = value
return (image,)
class LoadImageWithMetadata(LoadImage):
CATEGORY = "quicknodes/metadata"
FUNCTION = "func"
RETURN_TYPES = ["IMAGE", "MASK", "JSON", "STRING", ]
RETURN_NAMES = ["image", "mask", "json", "filepath", ]
def func(self, **kwargs):
filepath = folder_paths.get_annotated_filepath(kwargs['image'])
try:
with Image.open(filepath) as img: text = img.text
except:
text = {}
return self.load_image(**kwargs) + (text,filepath,)
CLAZZES = [AddToMetadata, LoadImageWithMetadata]