From 72508a8d19121e2814ea4dfbce8a5311f37dcd61 Mon Sep 17 00:00:00 2001
From: comfyanonymous <comfyanonymous@protonmail.com>
Date: Sat, 4 May 2024 03:39:37 -0400
Subject: [PATCH] Only set LOAD_TRUNCATED_IMAGES when if the Image open fails.

Document which PIL issues this works around.
---
 node_helpers.py | 15 ++++++++-------
 1 file changed, 8 insertions(+), 7 deletions(-)

diff --git a/node_helpers.py b/node_helpers.py
index 264bd4d5..60f8fa41 100644
--- a/node_helpers.py
+++ b/node_helpers.py
@@ -1,4 +1,4 @@
-from PIL import Image, ImageFile
+from PIL import Image, ImageFile, UnidentifiedImageError
 
 def conditioning_set_values(conditioning, values={}):
     c = []
@@ -11,14 +11,15 @@ def conditioning_set_values(conditioning, values={}):
     return c
 
 def open_image(path):
-    try :
-        ImageFile.LOAD_TRUNCATED_IMAGES = False
+    prev_value = None
+
+    try:
         img = Image.open(path)
-    
-    except:
+    except (UnidentifiedImageError, ValueError): #PIL issues #4472 and #2445
+        prev_value = ImageFile.LOAD_TRUNCATED_IMAGES
         ImageFile.LOAD_TRUNCATED_IMAGES = True
         img = Image.open(path)
-        
     finally:
-        ImageFile.LOAD_TRUNCATED_IMAGES = False
+        if prev_value is not None:
+            ImageFile.LOAD_TRUNCATED_IMAGES = prev_value
         return img