Update app.py

app.py

@@ -328,15 +328,30 @@ async def predict_tomato(file: UploadFile = File(...)):
             "message": "Please upload a clear image of a PLANT leaf."
         })
 
-        image_array = np.array(image)
-        white_mask = (
-            (image_array[:, :, 0] > 240) &
-            (image_array[:, :, 1] > 240) &
-            (image_array[:, :, 2] > 240)
+        # --- SMART BACKGROUND REMOVAL ---
+        image_array = np.array(image).astype(np.float32)
+
+        R = image_array[:, :, 0]
+        G = image_array[:, :, 1]
+        B = image_array[:, :, 2]
+
+        # Brightness: pixel must be very bright overall
+        brightness = (R + G + B) / 3.0
+        is_bright = brightness > 200
+
+        # Saturation: true background has no color — R, G, B are all similar
+        # A green leaf pixel will have G much higher than R and B
+        channel_range = (
+            np.maximum(np.maximum(R, G), B) -
+            np.minimum(np.minimum(R, G), B)
         )
-        image_array[white_mask] = [0, 0, 0]
-        image = Image.fromarray(image_array)
+        is_unsaturated = channel_range < 30  # low color variation = background
 
+        # Only replace pixels that are BOTH bright AND colorless
+        background_mask = is_bright & is_unsaturated
+        image_array[background_mask] = [0, 0, 0]
+
+        image = Image.fromarray(image_array.astype(np.uint8))
         
         # 1. Standard Resize
         image = image.resize((224, 224))