Update utils.py

utils.py

@@ -40,19 +40,32 @@ def preprocess(image_or_path):
     else:  # Assume the input is a PIL Image object
         img = image_or_path
     
-    # Resize the image
-    resized_img = img.resize((256, 256), Image.LANCZOS)
+    # Print image dimensions for debugging
+    print(f"Original image dimensions: {img.size}")
 
-    # Convert the image to a numpy array
-    img_array = np.array(resized_img)
+    # Check image mode
+    if img.mode != "RGB":
+        img = img.convert("RGB")
+        print("Converted image mode to RGB.")
 
-    # Normalize if necessary
-    img_array = img_array / 255.0
+    try:
+        # Resize the image
+        resized_img = img.resize((256, 256), Image.LANCZOS)
 
-    # Add a batch dimension
-    img_array = np.expand_dims(img_array, axis=0)
+        # Convert the image to a numpy array
+        img_array = np.array(resized_img)
 
-    return img_array
+        # Normalize if necessary
+        img_array = img_array / 255.0
+
+        # Add a batch dimension
+        img_array = np.expand_dims(img_array, axis=0)
+
+        return img_array
+
+    except Exception as e:
+        print(f"Error encountered: {e}")
+        return None
 
 def model_arc():
     model = tf.keras.Sequential([