util decode_base64_to_image

This commit is contained in:
w-e-w 2024-05-01 01:04:20 +09:00
parent 9d39380705
commit d36b732f55

View File

@ -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