useknockout 0.0.1__tar.gz → 0.0.2__tar.gz

{useknockout-0.0.1 → useknockout-0.0.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: useknockout
-Version: 0.0.1
+Version: 0.0.2
 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.1 → useknockout-0.0.2}/pyproject.toml

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

useknockout-0.0.2/src/useknockout/_version.py

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

{useknockout-0.0.1 → useknockout-0.0.2}/src/useknockout/async_client.py

@@ -323,3 +323,34 @@ class AsyncKnockout:
             "/estimate",
             json={"endpoint": endpoint, "width": width, "height": height},
         )
+
+    async def upscale(
+        self,
+        file: FileInput,
+        *,
+        scale: int = 4,
+        face_enhance: bool = False,
+        format: str = "png",
+    ) -> bytes:
+        if scale not in (2, 4):
+            raise ValueError("scale must be 2 or 4")
+        return await self._request_bytes(
+            "POST",
+            "/upscale",
+            files=_multipart_files(file),
+            data=_form({"scale": scale, "face_enhance": face_enhance, "format": format}),
+        )
+
+    async def face_restore(
+        self,
+        file: FileInput,
+        *,
+        only_center_face: bool = False,
+        format: str = "png",
+    ) -> bytes:
+        return await self._request_bytes(
+            "POST",
+            "/face-restore",
+            files=_multipart_files(file),
+            data=_form({"only_center_face": only_center_face, "format": format}),
+        )

{useknockout-0.0.1 → useknockout-0.0.2}/src/useknockout/client.py

@@ -352,3 +352,45 @@ class Knockout:
             "/estimate",
             json={"endpoint": endpoint, "width": width, "height": height},
         )
+
+    def upscale(
+        self,
+        file: FileInput,
+        *,
+        scale: int = 4,
+        face_enhance: bool = False,
+        format: str = "png",
+    ) -> bytes:
+        """POST /upscale — Real-ESRGAN x2/x4 super-resolution.
+
+        ``face_enhance=True`` routes through GFPGAN for sharper facial detail
+        (slower; use for portraits).
+        """
+        if scale not in (2, 4):
+            raise ValueError("scale must be 2 or 4")
+        return self._request_bytes(
+            "POST",
+            "/upscale",
+            files=_multipart_files(file),
+            data=_form({"scale": scale, "face_enhance": face_enhance, "format": format}),
+        )
+
+    def face_restore(
+        self,
+        file: FileInput,
+        *,
+        only_center_face: bool = False,
+        format: str = "png",
+    ) -> bytes:
+        """POST /face-restore — GFPGAN v1.4 portrait restoration.
+
+        Fixes blurry / damaged / low-res faces; background is 2x upscaled by
+        Real-ESRGAN. ``only_center_face=True`` restores just the most prominent
+        face (faster).
+        """
+        return self._request_bytes(
+            "POST",
+            "/face-restore",
+            files=_multipart_files(file),
+            data=_form({"only_center_face": only_center_face, "format": format}),
+        )

useknockout-0.0.1/src/useknockout/_version.py

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