Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fixes cascading bugs

Browse Source
This commit is contained in:
psychedelicious 2022-09-21 00:44:57 +10:00
parent 3e8a861fc0
commit 1ee3023cdd
1 changed files with 8 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
backend/server.py
View File

@ -158,9 +158,8 @@ def handle_request_all_images():
paths.sort(key=lambda x: os.path.getmtime(x)) paths.sort(key=lambda x: os.path.getmtime(x))
image_array = [] image_array = []
for path in paths: for path in paths:
# image = Image.open(path)
metadata = retrieve_metadata(path) metadata = retrieve_metadata(path)
image_array.append({"url": path, "metadata": metadata}) image_array.append({"url": path, "metadata": metadata["sd-metadata"]})
socketio.emit("galleryImages", {"images": image_array}) socketio.emit("galleryImages", {"images": image_array})
eventlet.sleep(0) eventlet.sleep(0)
@ -508,7 +507,13 @@ def save_image(
pngwriter = PngWriter(output_dir) pngwriter = PngWriter(output_dir)
prefix = pngwriter.unique_prefix() prefix = pngwriter.unique_prefix()
filename = f"{prefix}.{metadata['image']['seed']}" seed = "unknown_seed"
if "image" in metadata:
if "seed" in metadata["image"]:
seed = metadata["image"]["seed"]
filename = f"{prefix}.{seed}"
if step_index: if step_index:
filename += f".{step_index}" filename += f".{step_index}"