diff --git a/modules/util.py b/modules/util.py index 0db13736c..f1fd5efba 100644 --- a/modules/util.py +++ b/modules/util.py @@ -211,3 +211,13 @@ Requested path was: {path} subprocess.Popen(["wsl-open", path]) else: subprocess.Popen(["xdg-open", path]) + + +def decode_base64_to_image(base64_str: str): + from modules import images + from io import BytesIO + import base64 + if base64_str.startswith("data:image/"): + base64_str = base64_str.partition(';')[2].partition(',')[2] + image = images.read(BytesIO(base64.b64decode(base64_str))) + return image