Error handling update

iz_helpers/helpers.py

@@ -74,20 +74,23 @@ def do_upscaleImg(curImg, upscale_do, upscaler_name, upscale_by):
             + str(rheight),
             end="\r",
         )
-
-    pp = postprocessing_upscale.scripts_postprocessing.PostprocessedImage(curImg)
-    ups = postprocessing_upscale.ScriptPostprocessingUpscale()
-    ups.process(
-        pp,
-        upscale_mode=ups_mode,
-        upscale_by=upscale_by,
-        upscale_to_width=rwidth,
-        upscale_to_height=rheight,
-        upscale_crop=False,
-        upscaler_1_name=upscaler_name,
-        upscaler_2_name=None,
-        upscaler_2_visibility=0.0,
-    )
+    try:
+        pp = postprocessing_upscale.scripts_postprocessing.PostprocessedImage(curImg)
+        ups = postprocessing_upscale.ScriptPostprocessingUpscale()
+        ups.process(
+            pp,
+            upscale_mode=ups_mode,
+            upscale_by=upscale_by,
+            upscale_to_width=rwidth,
+            upscale_to_height=rheight,
+            upscale_crop=False,
+            upscaler_1_name=upscaler_name,
+            upscaler_2_name=None,
+            upscaler_2_visibility=0.0,
+        )
+    except Exception as e:
+        print("Infinite Zoom: upscaling failed: " + str(e))
+        return curImg
     return pp.image
 
 async def showGradioErrorAsync(txt, delay=1):

iz_helpers/image.py

@@ -44,20 +44,23 @@ def open_image(image_path):
     Returns:
         Image: A PIL Image object of the opened image.
     """
-    # Strip leading and trailing double quotation marks, if present
-    image_path = image_path.strip('"')
-    if image_path.startswith('http'):
-        # If the image path is a URL, download the image using requests
-        response = requests.get(image_path)
-        img = Image.open(BytesIO(response.content))
-    elif image_path.startswith('data'):
-        # If the image path is a DataURL, decode the base64 string
-        encoded_data = image_path.split(',')[1]
-        decoded_data = base64.b64decode(encoded_data)
-        img = Image.open(BytesIO(decoded_data))
-    else:
-        # Assume that the image path is a file path
-        img = Image.open(image_path)    
+    try:
+        # Strip leading and trailing double quotation marks, if present
+        image_path = image_path.strip('"')
+        if image_path.startswith('http'):
+            # If the image path is a URL, download the image using requests
+            response = requests.get(image_path)
+            img = Image.open(BytesIO(response.content))
+        elif image_path.startswith('data'):
+            # If the image path is a DataURL, decode the base64 string
+            encoded_data = image_path.split(',')[1]
+            decoded_data = base64.b64decode(encoded_data)
+            img = Image.open(BytesIO(decoded_data))
+        else:
+            # Assume that the image path is a file path
+            img = Image.open(image_path)
+    except Exception as e:
+        raise Exception(f'Error opening image: {e}')
     return img
 
 def apply_alpha_mask(image, mask_image, invert = False):