diff --git a/modules_forge/forge_canvas/canvas.py b/modules_forge/forge_canvas/canvas.py index 9c315c70..b7e354af 100644 --- a/modules_forge/forge_canvas/canvas.py +++ b/modules_forge/forge_canvas/canvas.py @@ -77,6 +77,7 @@ class LogicalImage(gr.Textbox): @wraps(gr.Textbox.__init__) def __init__(self, *args, numpy=True, **kwargs): self.numpy = numpy + self.infotext = dict() if 'value' in kwargs: initial_value = kwargs['value'] @@ -94,11 +95,16 @@ class LogicalImage(gr.Textbox): if not payload.startswith("data:image/png;base64,"): return None - return base64_to_image(payload, numpy=self.numpy) + image = base64_to_image(payload, numpy=self.numpy) + image.info = self.infotext + + return image def postprocess(self, value): if value is None: return None + + self.infotext = value.info return image_to_base64(value, numpy=self.numpy)