StableDiffusion/WebUI
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Merge pull request #9760 from Sakura-Luna/refresh

Browse Source 
Fix gallery not being refreshed correctly
This commit is contained in:
AUTOMATIC1111 2023-05-08 15:06:02 +03:00 committed by GitHub
commit 879ed5422c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
modules/generation_parameters_copypaste.py
View File

@ -59,6 +59,7 @@ def image_from_url_text(filedata):
is_in_right_dir = ui_tempdir.check_tmp_file(shared.demo, filename) is_in_right_dir = ui_tempdir.check_tmp_file(shared.demo, filename)
assert is_in_right_dir, 'trying to open image file outside of allowed directories' assert is_in_right_dir, 'trying to open image file outside of allowed directories'
filename = filename.rsplit('?', 1)[0]
return Image.open(filename) return Image.open(filename)
if type(filedata) == list: if type(filedata) == list:

2
modules/ui_tempdir.py
View File

@ -2,6 +2,7 @@ import os
import tempfile import tempfile
from collections import namedtuple from collections import namedtuple
from pathlib import Path from pathlib import Path
from time import time
import gradio as gr import gradio as gr
@ -34,6 +35,7 @@ def check_tmp_file(gradio, filename):
def save_pil_to_file(pil_image, dir=None): def save_pil_to_file(pil_image, dir=None):
already_saved_as = getattr(pil_image, 'already_saved_as', None) already_saved_as = getattr(pil_image, 'already_saved_as', None)
if already_saved_as and os.path.isfile(already_saved_as): if already_saved_as and os.path.isfile(already_saved_as):
already_saved_as += f'?{int(time())}'
register_tmp_file(shared.demo, already_saved_as) register_tmp_file(shared.demo, already_saved_as)
file_obj = Savedfile(already_saved_as) file_obj = Savedfile(already_saved_as)