michael/stable-diffusion-webui
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Emoji are not being sorted correctly #192

Browse source
This commit is contained in:
AUTOMATIC 2022-09-10 08:45:16 +03:00
parent ee8a6fa89f
commit 6f678ec79c
1 changed files with 4 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
modules/images.py
View file

@ -6,6 +6,7 @@ import re
import numpy as np import numpy as np
from PIL import Image, ImageFont, ImageDraw, PngImagePlugin from PIL import Image, ImageFont, ImageDraw, PngImagePlugin
from fonts.ttf import Roboto from fonts.ttf import Roboto
import string
import modules.shared import modules.shared
from modules.shared import opts from modules.shared import opts
@ -235,6 +236,7 @@ def resize_image(resize_mode, im, width, height):
invalid_filename_chars = '<>:"/\\|?*\n' invalid_filename_chars = '<>:"/\\|?*\n'
re_nonletters = re.compile(r'[\s'+string.punctuation+']+')
def sanitize_filename_part(text): def sanitize_filename_part(text):
@ -256,11 +258,11 @@ def save_image(image, path, basename, seed=None, prompt=None, extension='png', i
pnginfo = None pnginfo = None
if opts.save_to_dirs and not no_prompt: if opts.save_to_dirs and not no_prompt:
words = re.findall(r'\w+', prompt or "") words = re_nonletters.split(prompt or "")
if len(words) == 0: if len(words) == 0:
words = ["empty"] words = ["empty"]
dirname = " ".join(words[0:opts.save_to_dirs_prompt_len]) dirname = " ".join(words[0:opts.save_to_dirs_prompt_len]).strip()
path = os.path.join(path, dirname) path = os.path.join(path, dirname)
os.makedirs(path, exist_ok=True) os.makedirs(path, exist_ok=True)