diff options
author | AUTOMATIC <16777216c@gmail.com> | 2022-11-27 08:52:53 +0000 |
---|---|---|
committer | AUTOMATIC <16777216c@gmail.com> | 2022-11-27 08:52:53 +0000 |
commit | b006382784a2f0887317bb60ea49d19b50a5dc7e (patch) | |
tree | ea9ea0e697e630924c86ad264de97d5399c186e3 /modules/ui.py | |
parent | b5050ad2071644f7b4c99660dc66a8a95136102f (diff) | |
download | stable-diffusion-webui-gfx803-b006382784a2f0887317bb60ea49d19b50a5dc7e.tar.gz stable-diffusion-webui-gfx803-b006382784a2f0887317bb60ea49d19b50a5dc7e.tar.bz2 stable-diffusion-webui-gfx803-b006382784a2f0887317bb60ea49d19b50a5dc7e.zip |
serve images from where they are saved instead of a temporary directory
add an option to choose a different temporary directory in the UI
add an option to cleanup the selected temporary directory at startup
Diffstat (limited to 'modules/ui.py')
-rw-r--r-- | modules/ui.py | 16 |
1 files changed, 0 insertions, 16 deletions
diff --git a/modules/ui.py b/modules/ui.py index c8b8fecd..ea925c40 100644 --- a/modules/ui.py +++ b/modules/ui.py @@ -157,22 +157,6 @@ def save_files(js_data, images, do_make_zip, index): return gr.File.update(value=fullfns, visible=True), '', '', plaintext_to_html(f"Saved: {filenames[0]}")
-def save_pil_to_file(pil_image, dir=None):
- use_metadata = False
- metadata = PngImagePlugin.PngInfo()
- for key, value in pil_image.info.items():
- if isinstance(key, str) and isinstance(value, str):
- metadata.add_text(key, value)
- use_metadata = True
-
- file_obj = tempfile.NamedTemporaryFile(delete=False, suffix=".png", dir=dir)
- pil_image.save(file_obj, pnginfo=(metadata if use_metadata else None))
- return file_obj
-
-
-# override save to file function so that it also writes PNG info
-gr.processing_utils.save_pil_to_file = save_pil_to_file
-
def wrap_gradio_call(func, extra_outputs=None, add_stats=False):
def f(*args, extra_outputs_array=extra_outputs, **kwargs):
|