more fixes for gradio update
This commit is contained in:
parent
76f256fe8f
commit
b46b97fa29
@ -38,7 +38,7 @@ def quote(text):
|
|||||||
def image_from_url_text(filedata):
|
def image_from_url_text(filedata):
|
||||||
if type(filedata) == dict and filedata["is_file"]:
|
if type(filedata) == dict and filedata["is_file"]:
|
||||||
filename = filedata["name"]
|
filename = filedata["name"]
|
||||||
is_in_right_dir = any(Path(temp_dir).resolve() in Path(filename).resolve().parents for temp_dir in shared.demo.temp_dirs)
|
is_in_right_dir = any([filename in fileset for fileset in shared.demo.temp_file_sets])
|
||||||
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'
|
||||||
|
|
||||||
return Image.open(filename)
|
return Image.open(filename)
|
||||||
|
@ -45,7 +45,7 @@ def on_tmpdir_changed():
|
|||||||
|
|
||||||
os.makedirs(shared.opts.temp_dir, exist_ok=True)
|
os.makedirs(shared.opts.temp_dir, exist_ok=True)
|
||||||
|
|
||||||
shared.demo.temp_dirs = shared.demo.temp_dirs | {os.path.abspath(shared.opts.temp_dir)}
|
shared.demo.temp_file_sets[0] = shared.demo.temp_file_sets[0] | {os.path.abspath(shared.opts.temp_dir)}
|
||||||
|
|
||||||
|
|
||||||
def cleanup_tmpdr():
|
def cleanup_tmpdr():
|
||||||
|
Loading…
Reference in New Issue
Block a user