From ee2173405ab8ca820219b4fbcfc0e17c1dad4b50 Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Sun, 5 Nov 2023 16:38:43 -0600 Subject: [PATCH] parse region coordinates --- api/onnx_web/diffusers/utils.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/api/onnx_web/diffusers/utils.py b/api/onnx_web/diffusers/utils.py index 5a5cfcd4..b4a9f031 100644 --- a/api/onnx_web/diffusers/utils.py +++ b/api/onnx_web/diffusers/utils.py @@ -458,5 +458,10 @@ def slice_prompt(prompt: str, slice: int) -> str: Region = Tuple[int, int, int, int, Literal["add", "replace"], str] +def parse_region_group(group) -> Region: + top, left, bottom, right, mode, prompt = group + return (int(top), int(left), int(bottom), int(right), mode, prompt) + + def parse_regions(prompt: str) -> Tuple[str, List[Region]]: - return get_tokens_from_prompt(prompt, REGION_TOKEN, lambda it: it) + return get_tokens_from_prompt(prompt, REGION_TOKEN, parser=parse_region_group)