Surn commited on
Commit
93e5537
·
1 Parent(s): 6143ad5

Image Utils error handling update

Browse files
Files changed (1) hide show
  1. utils/image_utils.py +13 -7
utils/image_utils.py CHANGED
@@ -214,19 +214,22 @@ def change_color(image, color, opacity=0.75):
214
  Returns:
215
  PIL.Image.Image: The image with the color changed.
216
  """
217
- if type(image) is str:
218
  image = open_image(image)
 
 
 
219
  try:
220
  # Convert the color to RGBA format
221
  rgba_color = detect_color_format(color)
222
  rgba_color = update_color_opacity(rgba_color, opacity)
223
-
224
- # Convert the image to RGBA mode
225
- image = image.convert("RGBA")
226
-
227
  # Create a new image with the same size and mode
228
  new_image = Image.new("RGBA", image.size, rgba_color)
229
-
 
 
 
230
  # Composite the new image with the original image
231
  result = Image.alpha_composite(image, new_image)
232
  except Exception as e:
@@ -235,6 +238,9 @@ def change_color(image, color, opacity=0.75):
235
  return result
236
 
237
  def blur_image(input_path, radius=5, blur_type="gaussian"):
 
 
 
238
  input_path, _ = get_image_from_dict(input_path)
239
  directory, _, name, _, new_ext = get_file_parts(input_path)
240
  # If the extension changes, rename the file
@@ -925,7 +931,7 @@ def apply_lut_to_image_path(lut_filename: str, image_path: str, intensity: int =
925
 
926
  img_lut = None
927
  if image_path is None:
928
- raise UserWarning("No image provided.")
929
  return None, None
930
 
931
  # Split the path into directory and filename
 
214
  Returns:
215
  PIL.Image.Image: The image with the color changed.
216
  """
217
+ if isinstance(image, (dict, str)):
218
  image = open_image(image)
219
+ if image is None:
220
+ print("Image to color is None")
221
+ return None
222
  try:
223
  # Convert the color to RGBA format
224
  rgba_color = detect_color_format(color)
225
  rgba_color = update_color_opacity(rgba_color, opacity)
226
+
 
 
 
227
  # Create a new image with the same size and mode
228
  new_image = Image.new("RGBA", image.size, rgba_color)
229
+
230
+ # Convert the image to RGBA mode
231
+ image = image.convert("RGBA")
232
+
233
  # Composite the new image with the original image
234
  result = Image.alpha_composite(image, new_image)
235
  except Exception as e:
 
238
  return result
239
 
240
  def blur_image(input_path, radius=5, blur_type="gaussian"):
241
+ if input_path is None:
242
+ print("There is no Image to Blur")
243
+ return None
244
  input_path, _ = get_image_from_dict(input_path)
245
  directory, _, name, _, new_ext = get_file_parts(input_path)
246
  # If the extension changes, rename the file
 
931
 
932
  img_lut = None
933
  if image_path is None:
934
+ raise UserWarning("No image provided for LUT.")
935
  return None, None
936
 
937
  # Split the path into directory and filename