vision-agent 0.2.191__tar.gz → 0.2.193__tar.gz

{vision_agent-0.2.191 → vision_agent-0.2.193}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: vision-agent
-Version: 0.2.191
+Version: 0.2.193
 Summary: Toolset for Vision Agent
 Author: Landing AI
 Author-email: dev@landing.ai

{vision_agent-0.2.191 → vision_agent-0.2.193}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poetry]
 name = "vision-agent"
-version = "0.2.191"
+version = "0.2.193"
 description = "Toolset for Vision Agent"
 authors = ["Landing AI <dev@landing.ai>"]
 readme = "README.md"

{vision_agent-0.2.191 → vision_agent-0.2.193}/vision_agent/tools/tools.py

@@ -1798,18 +1798,33 @@ def flux_image_inpainting(
         ... )
         >>> save_image(result, "inpainted_room.png")
     """
-    if (
-        image.shape[0] < 8
-        or image.shape[1] < 8
-        or mask.shape[0] < 8
-        or mask.shape[1] < 8
-    ):
-        raise ValueError("The image or mask does not have enough size for inpainting")
+
+    min_dim = 8
+
+    if any(dim < min_dim for dim in image.shape[:2] + mask.shape[:2]):
+        raise ValueError(f"Image and mask must be at least {min_dim}x{min_dim} pixels")
+
+    max_size = (512, 512)
+
+    if image.shape[0] > max_size[0] or image.shape[1] > max_size[1]:
+        scaling_factor = min(max_size[0] / image.shape[0], max_size[1] / image.shape[1])
+        new_size = (
+            int(image.shape[1] * scaling_factor),
+            int(image.shape[0] * scaling_factor),
+        )
+        new_size = ((new_size[0] // 8) * 8, (new_size[1] // 8) * 8)
+        image = cv2.resize(image, new_size, interpolation=cv2.INTER_AREA)
+        mask = cv2.resize(mask, new_size, interpolation=cv2.INTER_NEAREST)
+
+    elif image.shape[0] % 8 != 0 or image.shape[1] % 8 != 0:
+        new_size = ((image.shape[1] // 8) * 8, (image.shape[0] // 8) * 8)
+        image = cv2.resize(image, new_size, interpolation=cv2.INTER_AREA)
+        mask = cv2.resize(mask, new_size, interpolation=cv2.INTER_NEAREST)
 
     if np.array_equal(mask, mask.astype(bool).astype(int)):
         mask = np.where(mask > 0, 255, 0).astype(np.uint8)
     else:
-        raise ValueError("The mask should be a binary mask with 0's and 1's")
+        raise ValueError("Mask should contain only binary values (0 or 1)")
 
     image_file = numpy_to_bytes(image)
     mask_file = numpy_to_bytes(mask)