From 089c85a0175dc2c369ad701652cee41924fb55d9 Mon Sep 17 00:00:00 2001
From: psychedelicious <4822129+psychedelicious@users.noreply.github.com>
Date: Mon, 21 Nov 2022 14:56:06 +1100
Subject: [PATCH] Fixes bug when postprocessing image with no metadata

---
 backend/invoke_ai_web_server.py | 18 +++++++++++-------
 1 file changed, 11 insertions(+), 7 deletions(-)

diff --git a/backend/invoke_ai_web_server.py b/backend/invoke_ai_web_server.py
index 81e2ee6013..66fda884ea 100644
--- a/backend/invoke_ai_web_server.py
+++ b/backend/invoke_ai_web_server.py
@@ -324,8 +324,9 @@ class InvokeAIWebServer:
                         )
                         os.remove(thumbnail_path)
                     except Exception as e:
-                        traceback.print_exc()
-                        socketio.emit("error", {"message": f"Unable to delete {f}"})
+                        socketio.emit("error", {"message": f"Unable to delete {f}: {str(e)}"})
+                        pass
+
                 socketio.emit("tempFolderEmptied")
             except Exception as e:
                 self.socketio.emit("error", {"message": (str(e))})
@@ -430,8 +431,9 @@ class InvokeAIWebServer:
                                 "category": category,
                             }
                         )
-                    except:
-                        socketio.emit("error", {"message": f"Unable to load {path}"})
+                    except Exception as e:
+                        socketio.emit("error", {"message": f"Unable to load {path}: {str(e)}"})
+                        pass
 
                 socketio.emit(
                     "galleryImages",
@@ -500,9 +502,10 @@ class InvokeAIWebServer:
                                 "category": category,
                             }
                         )
-                    except:
+                    except Exception as e:
                         print(f">> Unable to load {path}")
-                        socketio.emit("error", {"message": f"Unable to load {path}"})
+                        socketio.emit("error", {"message": f"Unable to load {path}: {str(e)}"})
+                        pass
 
                 socketio.emit(
                     "galleryImages",
@@ -574,8 +577,9 @@ class InvokeAIWebServer:
 
                 try:
                     seed = original_image["metadata"]["image"]["seed"]
-                except (NameError, AttributeError) as e:
+                except (KeyError) as e:
                     seed = "unknown_seed"
+                    pass
 
                 if postprocessing_parameters["type"] == "esrgan":
                     progress.set_current_status("Upscaling (ESRGAN)")