prompture 1.1.6.dev1__tar.gz → 1.1.7.dev1__tar.gz

{prompture-1.1.6.dev1 → prompture-1.1.7.dev1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: prompture
-Version: 1.1.6.dev1
+Version: 1.1.7.dev1
 Summary: Ask LLMs to return structured JSON and run cross-model tests. API-first.
 Author-email: Juan Denis <juan@vene.co>
 License-Expression: MIT

{prompture-1.1.6.dev1 → prompture-1.1.7.dev1}/prompture/_version.py

@@ -18,7 +18,7 @@ version_tuple: tuple[int | str, ...]
 commit_id: str | None
 __commit_id__: str | None
 
-__version__ = version = '1.1.6.dev1'
-__version_tuple__ = version_tuple = (1, 1, 6, 'dev1')
+__version__ = version = '1.1.7.dev1'
+__version_tuple__ = version_tuple = (1, 1, 7, 'dev1')
 
 __commit_id__ = commit_id = None

{prompture-1.1.6.dev1 → prompture-1.1.7.dev1}/prompture/drivers/img_gen_base.py

@@ -34,6 +34,11 @@ class ImageGenDriver:
     supported_sizes: list[str] = []
     max_images: int = 10
 
+    # Editing capabilities (default: not supported). Drivers that implement
+    # editing flip these so callers can feature-detect before offering UI.
+    supports_edit: bool = False
+    supports_variation: bool = False
+
     callbacks: DriverCallbacks | None = None
 
     def generate_image(self, prompt: str, options: dict[str, Any]) -> dict[str, Any]:
@@ -48,6 +53,45 @@ class ImageGenDriver:
         """
         raise NotImplementedError
 
+    def edit_image(
+        self,
+        image: bytes,
+        prompt: str,
+        options: dict[str, Any],
+        *,
+        mask: bytes | None = None,
+    ) -> dict[str, Any]:
+        """Edit an existing image with a text instruction (image-to-image).
+
+        Args:
+            image: Source image bytes (PNG/JPEG).
+            prompt: Instruction describing the desired change.
+            options: Provider-specific options (size, n, model, etc.).
+            mask: Optional PNG mask; transparent pixels mark the region to edit
+                (inpainting). When omitted the whole image is editable.
+
+        Returns:
+            Dict with ``images`` and ``meta`` keys (same contract as
+            :meth:`generate_image`).
+        """
+        raise NotImplementedError(
+            f"{self.__class__.__name__} does not support image editing"
+        )
+
+    def create_variation(self, image: bytes, options: dict[str, Any]) -> dict[str, Any]:
+        """Produce variation(s) of an existing image (no prompt).
+
+        Args:
+            image: Source image bytes (PNG).
+            options: Provider-specific options (size, n, model, etc.).
+
+        Returns:
+            Dict with ``images`` and ``meta`` keys.
+        """
+        raise NotImplementedError(
+            f"{self.__class__.__name__} does not support variations"
+        )
+
     # ------------------------------------------------------------------
     # Hook-aware wrapper
     # ------------------------------------------------------------------

{prompture-1.1.6.dev1 → prompture-1.1.7.dev1}/prompture/drivers/openai_img_gen_driver.py

@@ -2,6 +2,7 @@
 
 from __future__ import annotations
 
+import io
 import logging
 import os
 from typing import Any
@@ -23,6 +24,8 @@ class OpenAIImageGenDriver(ImageCostMixin, ImageGenDriver):
 
     supports_multiple = True
     supports_size_variants = True
+    supports_edit = True
+    supports_variation = True
     supported_sizes = [
         "256x256",
         "512x512",
@@ -148,3 +151,80 @@ class OpenAIImageGenDriver(ImageCostMixin, ImageGenDriver):
                 "raw_response": {},
             },
         }
+
+    # ------------------------------------------------------------------
+    # Editing
+    # ------------------------------------------------------------------
+
+    _VARIATION_SIZES = {"256x256", "512x512", "1024x1024"}
+
+    def _safe_cost(self, model: str, size: str, n: int, quality: str = "standard") -> float:
+        try:
+            return round(self._calculate_image_cost("openai", model, size=size, quality=quality, n=n), 6)
+        except Exception:  # pragma: no cover - pricing table gaps shouldn't break a call
+            return 0.0
+
+    def _result(self, data: Any, model: str, size: str, n: int) -> dict[str, Any]:
+        images = [image_from_base64(d.b64_json, media_type="image/png") for d in data]
+        return {
+            "images": images,
+            "meta": {
+                "image_count": len(images),
+                "size": size,
+                "revised_prompt": None,
+                "cost": self._safe_cost(model, size, n),
+                "model_name": f"openai/{model}",
+                "raw_response": {},
+            },
+        }
+
+    def edit_image(
+        self,
+        image: bytes,
+        prompt: str,
+        options: dict[str, Any],
+        *,
+        mask: bytes | None = None,
+    ) -> dict[str, Any]:
+        """Edit an image with an instruction. Uses gpt-image-1 by default."""
+        if self.client is None:
+            raise RuntimeError("openai package (>=1.0.0) is not installed")
+
+        model = options.get("model", "gpt-image-1")
+        size = options.get("size", "1024x1024")
+        n = options.get("n", 1)
+        kwargs: dict[str, Any] = {
+            "model": model,
+            "prompt": prompt,
+            "image": ("image.png", io.BytesIO(image), "image/png"),
+            "n": n,
+            "size": size,
+        }
+        if mask is not None:
+            kwargs["mask"] = ("mask.png", io.BytesIO(mask), "image/png")
+        # gpt-image-* always returns b64; DALL·E 2 edits need it requested.
+        if not model.startswith("gpt-image"):
+            kwargs["response_format"] = "b64_json"
+
+        resp = self.client.images.edit(**kwargs)
+        return self._result(resp.data, model, size, n)
+
+    def create_variation(self, image: bytes, options: dict[str, Any]) -> dict[str, Any]:
+        """Produce variation(s) of an image. Only DALL·E 2 supports this."""
+        if self.client is None:
+            raise RuntimeError("openai package (>=1.0.0) is not installed")
+
+        model = "dall-e-2"  # the only OpenAI model with a variations endpoint
+        size = options.get("size", "1024x1024")
+        if size not in self._VARIATION_SIZES:
+            size = "1024x1024"
+        n = options.get("n", 1)
+
+        resp = self.client.images.create_variation(
+            model=model,
+            image=("image.png", io.BytesIO(image), "image/png"),
+            n=n,
+            size=size,
+            response_format="b64_json",
+        )
+        return self._result(resp.data, model, size, n)

{prompture-1.1.6.dev1 → prompture-1.1.7.dev1}/prompture.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: prompture
-Version: 1.1.6.dev1
+Version: 1.1.7.dev1
 Summary: Ask LLMs to return structured JSON and run cross-model tests. API-first.
 Author-email: Juan Denis <juan@vene.co>
 License-Expression: MIT