useknockout 0.0.4__tar.gz → 0.2.0__tar.gz

{useknockout-0.0.4 → useknockout-0.2.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: useknockout
-Version: 0.0.4
+Version: 0.2.0
 Summary: Python SDK for useknockout — state-of-the-art background removal API.
 Project-URL: Homepage, https://useknockout.com
 Project-URL: Documentation, https://github.com/useknockout/python

{useknockout-0.0.4 → useknockout-0.2.0}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "useknockout"
-version = "0.0.4"
+version = "0.2.0"
 description = "Python SDK for useknockout — state-of-the-art background removal API."
 readme = "README.md"
 license = { text = "MIT" }

useknockout-0.2.0/src/useknockout/_version.py

@@ -0,0 +1 @@
+__version__ = "0.2.0"

{useknockout-0.0.4 → useknockout-0.2.0}/src/useknockout/async_client.py

@@ -2,7 +2,7 @@
 
 from __future__ import annotations
 
-from typing import Any, Dict, List, Optional
+from typing import Any, Dict, List, Optional, Tuple
 
 import httpx
 
@@ -14,6 +14,7 @@ from useknockout._helpers import (
     _multipart_batch,
     _multipart_files,
     _resolve_token,
+    _to_bytes_and_name,
 )
 from useknockout._version import __version__
 from useknockout.errors import KnockoutError, raise_for_status
@@ -362,3 +363,62 @@ class AsyncKnockout:
             files=_multipart_files(file),
             data=_form({"only_center_face": only_center_face, "format": format}),
         )
+
+    async def colorize(
+        self,
+        file: FileInput,
+        *,
+        format: str = "png",
+    ) -> bytes:
+        """POST /colorize — DDColor (Apache-2.0) image colorization. Async."""
+        return await self._request_bytes(
+            "POST",
+            "/colorize",
+            files=_multipart_files(file),
+            data=_form({"format": format}),
+        )
+
+    async def silhouette(
+        self,
+        file: FileInput,
+        *,
+        subject_color: str = "#7C3AED",
+        bg_color: str = "#FFFFFF",
+        format: str = "png",
+    ) -> bytes:
+        """POST /silhouette — two-tone silhouette portrait. Async."""
+        return await self._request_bytes(
+            "POST",
+            "/silhouette",
+            files=_multipart_files(file),
+            data=_form({
+                "subject_color": subject_color,
+                "bg_color": bg_color,
+                "format": format,
+            }),
+        )
+
+    async def inpaint(
+        self,
+        file: FileInput,
+        *,
+        mask: Optional[FileInput] = None,
+        bbox: Optional[Tuple[int, int, int, int]] = None,
+        dilation: int = 8,
+        format: str = "png",
+    ) -> bytes:
+        """POST /inpaint — LaMa-based image inpainting. Async."""
+        files = _multipart_files(file)
+        if mask is not None:
+            mdata, mname = _to_bytes_and_name(mask, default_name="mask.png")
+            files.append(("mask", (mname, mdata, "application/octet-stream")))
+        data = {"dilation": str(int(dilation)), "format": format}
+        if bbox is not None:
+            x, y, w, h = bbox
+            data.update({"x": str(int(x)), "y": str(int(y)), "w": str(int(w)), "h": str(int(h))})
+        return await self._request_bytes(
+            "POST",
+            "/inpaint",
+            files=files,
+            data=_form(data),
+        )

{useknockout-0.0.4 → useknockout-0.2.0}/src/useknockout/client.py

@@ -2,7 +2,7 @@
 
 from __future__ import annotations
 
-from typing import Any, Dict, List, Optional
+from typing import Any, Dict, List, Optional, Tuple
 
 import httpx
 
@@ -14,6 +14,7 @@ from useknockout._helpers import (
     _multipart_batch,
     _multipart_files,
     _resolve_token,
+    _to_bytes_and_name,
 )
 from useknockout._version import __version__
 from useknockout.errors import KnockoutError, raise_for_status
@@ -403,3 +404,88 @@ class Knockout:
             files=_multipart_files(file),
             data=_form({"only_center_face": only_center_face, "format": format}),
         )
+
+    def colorize(
+        self,
+        file: FileInput,
+        *,
+        format: str = "png",
+    ) -> bytes:
+        """POST /colorize — DDColor (Apache-2.0) image colorization.
+
+        Predicts plausible color from grayscale luminance via a ConvNeXt-Large
+        backbone (single feed-forward, ~500ms warm). Works on B&W or color
+        input — color images are converted to grayscale internally before
+        prediction, which makes round-trip recoloring straightforward.
+
+        Added in v0.1.0; requires API ≥ v0.7.0.
+        """
+        return self._request_bytes(
+            "POST",
+            "/colorize",
+            files=_multipart_files(file),
+            data=_form({"format": format}),
+        )
+
+    def silhouette(
+        self,
+        file: FileInput,
+        *,
+        subject_color: str = "#7C3AED",
+        bg_color: str = "#FFFFFF",
+        format: str = "png",
+    ) -> bytes:
+        """POST /silhouette — two-tone silhouette portrait.
+
+        Subject filled with one solid color, background filled with another.
+        Apple Music / Spotify avatar style. Reuses BiRefNet's mask path; no
+        new model load.
+
+        Added in v0.1.1; requires API ≥ v0.7.1.
+        """
+        return self._request_bytes(
+            "POST",
+            "/silhouette",
+            files=_multipart_files(file),
+            data=_form({
+                "subject_color": subject_color,
+                "bg_color": bg_color,
+                "format": format,
+            }),
+        )
+
+    def inpaint(
+        self,
+        file: FileInput,
+        *,
+        mask: Optional[FileInput] = None,
+        bbox: Optional[Tuple[int, int, int, int]] = None,
+        dilation: int = 8,
+        format: str = "png",
+    ) -> bytes:
+        """POST /inpaint — LaMa-based image inpainting.
+
+        Three auto-detected modes:
+          - ``mask`` provided           → user-supplied mask
+          - ``bbox=(x, y, w, h)``       → rectangular region
+          - neither                     → auto-subject (BiRefNet, inverted)
+
+        ``dilation`` (0..32, default 8) expands the mask before LaMa runs
+        to reduce ghost outlines.
+
+        Added in v0.2.0; requires API ≥ v0.8.0.
+        """
+        files = _multipart_files(file)
+        if mask is not None:
+            mdata, mname = _to_bytes_and_name(mask, default_name="mask.png")
+            files.append(("mask", (mname, mdata, "application/octet-stream")))
+        data = {"dilation": str(int(dilation)), "format": format}
+        if bbox is not None:
+            x, y, w, h = bbox
+            data.update({"x": str(int(x)), "y": str(int(y)), "w": str(int(w)), "h": str(int(h))})
+        return self._request_bytes(
+            "POST",
+            "/inpaint",
+            files=files,
+            data=_form(data),
+        )

useknockout-0.0.4/src/useknockout/_version.py

@@ -1 +0,0 @@
-__version__ = "0.0.4"