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

Gracefully handles corrupted images; fixes #1486

Browse Source 
- App does not crash if corrupted image loaded
- Error is displayed in the UI console and CLI output if an image cannot be loaded
This commit is contained in:
psychedelicious 2022-11-20 11:16:41 +11:00 committed by blessedcoolant
parent b908f2b4bc
commit 9d34213b4c
1 changed files with 51 additions and 46 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

97
backend/invoke_ai_web_server.py
View File

@ -334,32 +334,34 @@ class InvokeAIWebServer:
image_array = []
for path in image_paths:
if os.path.splitext(path)[1] == ".png":
metadata = retrieve_metadata(path)
# sd_metadata = metadata["sd-metadata"]
else:
# sd_metadata = {}
metadata = {}
try:
if os.path.splitext(path)[1] == ".png":
metadata = retrieve_metadata(path)
else:
metadata = {}
pil_image = Image.open(path)
(width, height) = pil_image.size
pil_image = Image.open(path)
(width, height) = pil_image.size
thumbnail_path = save_thumbnail(
pil_image, os.path.basename(path), self.thumbnail_image_path
)
thumbnail_path = save_thumbnail(
pil_image, os.path.basename(path), self.thumbnail_image_path
)
image_array.append(
{
"url": self.get_url_from_image_path(path),
"thumbnail": self.get_url_from_image_path(thumbnail_path),
"mtime": os.path.getmtime(path),
"metadata": metadata,
# "metadata": sd_metadata,
"width": width,
"height": height,
"category": category,
}
)
image_array.append(
{
"url": self.get_url_from_image_path(path),
"thumbnail": self.get_url_from_image_path(
thumbnail_path
),
"mtime": os.path.getmtime(path),
"metadata": metadata,
"width": width,
"height": height,
"category": category,
}
)
except:
socketio.emit("error", {"message": f"Unable to load {path}"})
socketio.emit(
"galleryImages",
@ -402,32 +404,35 @@ class InvokeAIWebServer:
image_array = []
for path in image_paths:
if os.path.splitext(path)[1] == ".png":
metadata = retrieve_metadata(path)
# sd_metadata = metadata["sd-metadata"]
else:
# sd_metadata = {}
metadata = {}
try:
if os.path.splitext(path)[1] == ".png":
metadata = retrieve_metadata(path)
else:
metadata = {}
pil_image = Image.open(path)
(width, height) = pil_image.size
pil_image = Image.open(path)
(width, height) = pil_image.size
thumbnail_path = save_thumbnail(
pil_image, os.path.basename(path), self.thumbnail_image_path
)
thumbnail_path = save_thumbnail(
pil_image, os.path.basename(path), self.thumbnail_image_path
)
image_array.append(
{
"url": self.get_url_from_image_path(path),
"thumbnail": self.get_url_from_image_path(thumbnail_path),
"mtime": os.path.getmtime(path),
# "metadata": sd_metadata,
"metadata": metadata,
"width": width,
"height": height,
"category": category,
}
)
image_array.append(
{
"url": self.get_url_from_image_path(path),
"thumbnail": self.get_url_from_image_path(
thumbnail_path
),
"mtime": os.path.getmtime(path),
"metadata": metadata,
"width": width,
"height": height,
"category": category,
}
)
except:
print(f'>> Unable to load {path}')
socketio.emit("error", {"message": f"Unable to load {path}"})
socketio.emit(
"galleryImages",