novelai-python 0.6.1__tar.gz → 0.7.0__tar.gz

{novelai_python-0.6.1 → novelai_python-0.7.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: novelai-python
-Version: 0.6.1
+Version: 0.7.0
 Summary: NovelAI Python Binding With Pydantic
 Author-Email: sudoskys <coldlando@hotmail.com>
 License: Apache License 2.0
@@ -45,7 +45,8 @@ Description-Content-Type: text/markdown
 
 ✨ NovelAI api python sdk with Pydantic, modern and user-friendly.
 
-The goal of this repository is to use Pydantic to build legitimate requests to access the NovelAI API service.
+The goal of this repository is to use Pydantic to build legitimate requests to access
+the [NovelAI API service](https://api.novelai.net/docs).
 
 > Python >= 3.9 is required.
 
@@ -60,6 +61,8 @@ The goal of this repository is to use Pydantic to build legitimate requests to a
 pip -U install novelai-python
 ```
 
+All API users must adhere to the NovelAI Terms of Service: https://novelai.net/terms.
+
 **More examples can be found in the [playground](https://github.com/LlmKira/novelai-python/tree/main/playground)
 directory, read code as documentation.**
 
@@ -76,6 +79,7 @@ from novelai_python.sdk.ai.generate_image.schema import PositionMap
 
 load_dotenv()
 session = ApiCredential(api_token=SecretStr(os.getenv("NOVELAI_JWT")))  # pst-***
+# For security reasons, storing user credentials in plaintext is strongly discouraged.
 
 prompt = "1girl, year 2023,dynamic angle,  best quality, amazing quality, very aesthetic, absurdres"
 
@@ -175,8 +179,12 @@ loop.run_until_complete(chat("Hello"))
 ```python
 from novelai_python.tool.random_prompt import RandomPromptGenerator
 
-prompt = RandomPromptGenerator(nsfw_enabled=False).random_prompt()
-print(prompt)
+generator = RandomPromptGenerator()
+for i in range(10):
+    print(generator.generate_common_tags(nsfw=False))
+    print(generator.generate_scene_tags())
+    print(generator.generate_scene_composition())
+    print(generator.get_holiday_themed_tags())
 ```
 
 #### 📦 Run A Server
@@ -189,10 +197,15 @@ python3 -m novelai_python.server -h '127.0.0.1' -p 7888
 #### 📦 Tokenizer
 
 ```python
-from novelai_python._enum import get_tokenizer_model, TextLLMModel
+from novelai_python._enum import get_tokenizer_model, TextLLMModel, TextTokenizerGroup
 from novelai_python.tokenizer import NaiTokenizer
 
+# Through llm model name to get the tokenizer
 tokenizer_package = NaiTokenizer(get_tokenizer_model(TextLLMModel.ERATO))
+# Directly use the tokenizer
+clip_tokenizer = NaiTokenizer(TextTokenizerGroup.CLIP)
+
+# Tokenize a text
 t_text = "a fox jumped over the lazy dog"
 encode_tokens = tokenizer_package.encode(t_text)
 print(tokenizer_package.tokenize_text(t_text))
@@ -236,8 +249,6 @@ https://dghs-imgutils.deepghs.org/main/api_doc/operate/censor.html
 
 ## 🙏 Acknowledgements
 
-[BackEnd](https://api.novelai.net/docs)
-
 [novelai-api](https://github.com/Aedial/novelai-api)
 
 [NovelAI-API](https://github.com/HanaokaYuzu/NovelAI-API)

{novelai_python-0.6.1 → novelai_python-0.7.0}/README.md

@@ -7,7 +7,8 @@
 
 ✨ NovelAI api python sdk with Pydantic, modern and user-friendly.
 
-The goal of this repository is to use Pydantic to build legitimate requests to access the NovelAI API service.
+The goal of this repository is to use Pydantic to build legitimate requests to access
+the [NovelAI API service](https://api.novelai.net/docs).
 
 > Python >= 3.9 is required.
 
@@ -22,6 +23,8 @@ The goal of this repository is to use Pydantic to build legitimate requests to a
 pip -U install novelai-python
 ```
 
+All API users must adhere to the NovelAI Terms of Service: https://novelai.net/terms.
+
 **More examples can be found in the [playground](https://github.com/LlmKira/novelai-python/tree/main/playground)
 directory, read code as documentation.**
 
@@ -38,6 +41,7 @@ from novelai_python.sdk.ai.generate_image.schema import PositionMap
 
 load_dotenv()
 session = ApiCredential(api_token=SecretStr(os.getenv("NOVELAI_JWT")))  # pst-***
+# For security reasons, storing user credentials in plaintext is strongly discouraged.
 
 prompt = "1girl, year 2023,dynamic angle,  best quality, amazing quality, very aesthetic, absurdres"
 
@@ -137,8 +141,12 @@ loop.run_until_complete(chat("Hello"))
 ```python
 from novelai_python.tool.random_prompt import RandomPromptGenerator
 
-prompt = RandomPromptGenerator(nsfw_enabled=False).random_prompt()
-print(prompt)
+generator = RandomPromptGenerator()
+for i in range(10):
+    print(generator.generate_common_tags(nsfw=False))
+    print(generator.generate_scene_tags())
+    print(generator.generate_scene_composition())
+    print(generator.get_holiday_themed_tags())
 ```
 
 #### 📦 Run A Server
@@ -151,10 +159,15 @@ python3 -m novelai_python.server -h '127.0.0.1' -p 7888
 #### 📦 Tokenizer
 
 ```python
-from novelai_python._enum import get_tokenizer_model, TextLLMModel
+from novelai_python._enum import get_tokenizer_model, TextLLMModel, TextTokenizerGroup
 from novelai_python.tokenizer import NaiTokenizer
 
+# Through llm model name to get the tokenizer
 tokenizer_package = NaiTokenizer(get_tokenizer_model(TextLLMModel.ERATO))
+# Directly use the tokenizer
+clip_tokenizer = NaiTokenizer(TextTokenizerGroup.CLIP)
+
+# Tokenize a text
 t_text = "a fox jumped over the lazy dog"
 encode_tokens = tokenizer_package.encode(t_text)
 print(tokenizer_package.tokenize_text(t_text))
@@ -198,8 +211,6 @@ https://dghs-imgutils.deepghs.org/main/api_doc/operate/censor.html
 
 ## 🙏 Acknowledgements
 
-[BackEnd](https://api.novelai.net/docs)
-
 [novelai-api](https://github.com/Aedial/novelai-api)
 
 [NovelAI-API](https://github.com/HanaokaYuzu/NovelAI-API)

{novelai_python-0.6.1 → novelai_python-0.7.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "novelai-python"
-version = "0.6.1"
+version = "0.7.0"
 description = "NovelAI Python Binding With Pydantic"
 authors = [
     { name = "sudoskys", email = "coldlando@hotmail.com" },
@@ -59,4 +59,5 @@ distribution = true
 [tool.pdm.dev-dependencies]
 dev = [
     "pre-commit>=3.5.0",
+    "wasmtime>=28.0.0",
 ]

{novelai_python-0.6.1 → novelai_python-0.7.0}/src/novelai_python/_enum.py

@@ -40,6 +40,7 @@ class TextTokenizerGroup(object):
     LLAMA3 = "llama3_tokenizer.def"
     GPT2 = "gpt2_tokenizer.def"
     CLIP = "clip_tokenizer.def"
+    T5 = "t5_tokenizer.def"
 
 
 TextLLMModelTypeAlias = Union[TextLLMModel, str]

{novelai_python-0.6.1 → novelai_python-0.7.0}/src/novelai_python/sdk/ai/generate/__init__.py

@@ -341,7 +341,8 @@ class LLM(ApiBaseModel):
                       override_headers: Optional[dict] = None
                       ) -> LLMResp:
         """
-        生成图片
+        Generate text using NovelAI's large language models. According to our Terms of Service, all generation requests must be initiated by a human action. Automating text or image generation to create excessive load on our systems is not allowed.
+
         :param override_headers:
         :param session:  session
         :return: LLMStreamResp

{novelai_python-0.6.1 → novelai_python-0.7.0}/src/novelai_python/sdk/ai/generate/_schema.py

@@ -115,7 +115,7 @@ class LLMGenerationParams(BaseModel):
     repetition_penalty_slope: Optional[Union[float, int]] = Field(default=None, ge=0, allow_inf_nan=False)
 
     eos_token_id: int = None
-    bad_words_ids: List[List[int]] = None
+    bad_words_ids: Optional[List[List[int]]] = None
     logit_bias_groups: Optional[List[LogitBiasGroup]] = Field(default_factory=list)
 
     repetition_penalty_frequency: Optional[Union[float, int]] = Field(default=None, allow_inf_nan=False)

{novelai_python-0.6.1 → novelai_python-0.7.0}/src/novelai_python/sdk/ai/generate_image/__init__.py

@@ -891,7 +891,8 @@ class GenerateImageInfer(ApiBaseModel):
                       override_headers: Optional[dict] = None,
                       ) -> ImageGenerateResp:
         """
-        生成图片
+        Generate images using NovelAI's diffusion models. According to our Terms of Service, all generation requests must be initiated by a human action. Automating text or image generation to create excessive load on our systems is not allowed.
+
         :param override_headers: the headers to override
         :param session:  session
         :return:

{novelai_python-0.6.1 → novelai_python-0.7.0}/src/novelai_python/tokenizer/__init__.py

@@ -1,5 +1,6 @@
 import gzip
 import io
+import json
 import pathlib
 import zlib
 from typing import Dict, List, Optional
@@ -56,6 +57,14 @@ class LLMTokenizerSetting(TokenizerSetting):
         return self
 
 
+def is_clip_model(model_name: str) -> bool:
+    return model_name.lower().startswith("clip_tokenizer")
+
+
+def is_t5_model(model_name: str) -> bool:
+    return model_name.lower().startswith("t5_tokenizer")
+
+
 class NaiTokenizer:
     def __init__(self,
                  model: str,
@@ -129,8 +138,10 @@ class NaiTokenizer:
         decoded_str = self._read_compressed_def(model_bytes)
         repaired_data = repair_json(decoded_str, return_objects=True)
         try:
-            if "clip" in self.model_full_name.lower():
+            if is_clip_model(self.model_full_name):
                 self.tokenizer_setting = ClipTokenizerSetting.model_validate(repaired_data)
+            elif is_t5_model(self.model_full_name):
+                self.tokenizer_setting = json.dumps(repaired_data)
             else:
                 self.tokenizer_setting = LLMTokenizerSetting.model_validate(repaired_data)
         except Exception as e:
@@ -140,8 +151,10 @@ class NaiTokenizer:
             )
 
     def _create_tokenizer(self):
-        if "clip" in self.model_full_name.lower():
+        if is_clip_model(self.model_full_name):
             self._create_clip_tokenizer()
+        elif is_t5_model(self.model_full_name):
+            self.tokenizer = Tokenizer.from_str(self.tokenizer_setting)
         else:
             self._create_bpe_tokenizer()
 

{novelai_python-0.6.1 → novelai_python-0.7.0}/src/novelai_python/tool/image_metadata/__init__.py

@@ -13,94 +13,109 @@ from nacl.signing import VerifyKey
 from pydantic import BaseModel, ConfigDict
 
 from novelai_python.sdk.ai._enum import PROMOTION, Model
-from ._type import IMAGE_INPUT_TYPE, get_image_bytes
 from .bch_utils import fec_decode
 from .lsb_extractor import ImageLsbDataExtractor
 from .lsb_injector import inject_data
 
 
+class BrokenMetaDataError(Exception):
+    pass
+
+
+class BrokenCommentError(Exception):
+    pass
+
+
+class CommentModel(BaseModel):
+    prompt: Optional[str] = None
+    steps: Optional[int] = None
+    height: Optional[int] = None
+    width: Optional[int] = None
+    scale: Optional[float] = None
+    uncond_scale: Optional[float] = None
+    cfg_rescale: Optional[float] = None
+    seed: Optional[int] = None
+    n_samples: Optional[int] = None
+    hide_debug_overlay: Optional[bool] = None
+    noise_schedule: Optional[str] = None
+    legacy_v3_extend: Optional[bool] = None
+    reference_information_extracted: Optional[float] = None
+    reference_strength: Optional[float] = None
+    reference_strength_multiple: Optional[List[float]] = None
+    reference_information_extracted_multiple: Optional[List[float]] = None
+    sampler: Optional[str] = None
+    controlnet_strength: Optional[float] = None
+    controlnet_model: Optional[Union[None, str]] = None
+    dynamic_thresholding: Optional[bool] = None
+    dynamic_thresholding_percentile: Optional[float] = None
+    dynamic_thresholding_mimic_scale: Optional[float] = None
+    sm: Optional[bool] = None
+    sm_dyn: Optional[bool] = None
+    skip_cfg_below_sigma: Optional[float] = None
+    lora_unet_weights: Optional[Union[None, str]] = None
+    lora_clip_weights: Optional[Union[None, str]] = None
+    uc: Optional[str] = None
+    request_type: Optional[str] = None
+    signed_hash: Optional[str] = None
+    skip_cfg_above_sigma: Optional[float] = None
+    deliberate_euler_ancestral_bug: Optional[bool] = None
+    prefer_brownian: Optional[bool] = None
+    cfg_sched_eligibility: Optional[str] = None
+    explike_fine_detail: Optional[bool] = None
+    minimize_sigma_inf: Optional[bool] = None
+    uncond_per_vibe: Optional[bool] = None
+    wonky_vibe_correlation: Optional[bool] = None
+    version: Optional[int] = None
+
+    extra_passthrough_testing: Optional[dict] = None
+    """More test infomation"""
+    v4_prompt: Optional[dict] = None
+    v4_negative_prompt: Optional[dict] = None
+
+    model_config = ConfigDict(extra="allow")
+
+    @property
+    def generate_method(self):
+        return self.request_type or "Unknown"
+
+    @property
+    def negative_prompt(self):
+        return self.uc
+
+    @property
+    def vibe_transfer_strength(self) -> List[float]:
+        """
+        Get the vibe transfer strength totally
+        """
+        if self.reference_strength_multiple:
+            return self.reference_strength_multiple
+        reference_strength = [] if self.reference_strength is None else [self.reference_strength]
+        return reference_strength
+
+    @property
+    def vibe_transfer_information(self) -> List[float]:
+        """
+        Get the vibe transfer information totally
+        """
+        if self.reference_information_extracted_multiple:
+            return self.reference_information_extracted_multiple
+        reference_information = [] if self.reference_information_extracted is None else [
+            self.reference_information_extracted]
+        return reference_information
+
+
 class ImageMetadata(BaseModel):
     """
     ImageMetadata is a class that represents the metadata of a NovelAI generated image.
     """
 
-    class CommentModel(BaseModel):
-        prompt: Optional[str] = None
-        steps: Optional[int] = None
-        height: Optional[int] = None
-        width: Optional[int] = None
-        scale: Optional[float] = None
-        uncond_scale: Optional[float] = None
-        cfg_rescale: Optional[float] = None
-        seed: Optional[int] = None
-        n_samples: Optional[int] = None
-        hide_debug_overlay: Optional[bool] = None
-        noise_schedule: Optional[str] = None
-        legacy_v3_extend: Optional[bool] = None
-        reference_information_extracted: Optional[float] = None
-        reference_strength: Optional[float] = None
-        reference_strength_multiple: Optional[List[float]] = None
-        reference_information_extracted_multiple: Optional[List[float]] = None
-        sampler: Optional[str] = None
-        controlnet_strength: Optional[float] = None
-        controlnet_model: Optional[Union[None, str]] = None
-        dynamic_thresholding: Optional[bool] = None
-        dynamic_thresholding_percentile: Optional[float] = None
-        dynamic_thresholding_mimic_scale: Optional[float] = None
-        sm: Optional[bool] = None
-        sm_dyn: Optional[bool] = None
-        skip_cfg_below_sigma: Optional[float] = None
-        lora_unet_weights: Optional[Union[None, str]] = None
-        lora_clip_weights: Optional[Union[None, str]] = None
-        uc: Optional[str] = None
-        request_type: Optional[str] = None
-        signed_hash: Optional[str] = None
-        skip_cfg_above_sigma: Optional[float] = None
-        deliberate_euler_ancestral_bug: Optional[bool] = None
-        prefer_brownian: Optional[bool] = None
-        cfg_sched_eligibility: Optional[str] = None
-        explike_fine_detail: Optional[bool] = None
-        minimize_sigma_inf: Optional[bool] = None
-        uncond_per_vibe: Optional[bool] = None
-        wonky_vibe_correlation: Optional[bool] = None
-        version: Optional[int] = None
-
-        model_config = ConfigDict(extra="allow")
-
-        @property
-        def generate_method(self):
-            return self.request_type or "Unknown"
-
-        @property
-        def negative_prompt(self):
-            return self.uc
-
-        @property
-        def vibe_transfer_strength(self) -> List[float]:
-            """
-            Get the vibe transfer strength totally
-            """
-            if self.reference_strength_multiple:
-                return self.reference_strength_multiple
-            reference_strength = [] if self.reference_strength is None else [self.reference_strength]
-            return reference_strength
-
-        @property
-        def vibe_transfer_information(self) -> List[float]:
-            """
-            Get the vibe transfer information totally
-            """
-            if self.reference_information_extracted_multiple:
-                return self.reference_information_extracted_multiple
-            reference_information = [] if self.reference_information_extracted is None else [
-                self.reference_information_extracted]
-            return reference_information
-
-    Title: str = "AI generated image"
-    Software: str = "NovelAI"
-    Source: str = None
-    Description: str
-    Comment: CommentModel
+    Generation_time: Optional[float] = None
+    Comment: Optional[CommentModel] = None
+    Title: Optional[str] = None
+    Description: Optional[str] = None
+    Software: Optional[str] = None
+    Source: Optional[str] = None
+    dpi: Optional[Tuple[float, float]] = None
 
     model_config = ConfigDict(extra="allow")
 
@@ -113,14 +128,13 @@ class ImageMetadata(BaseModel):
         return PROMOTION.get(self.Source, None)
 
     @staticmethod
-    def reset_alpha(image: IMAGE_INPUT_TYPE) -> BytesIO:
+    def reset_alpha(image: Image.Image) -> BytesIO:
         """
         Remove LSB from the image, set the alpha channel to 254
         :param image: Type: Union[str, bytes, Path, BytesIO]
         :return: BytesIO
         """
-        image_data = get_image_bytes(image)
-        image = Image.open(image_data).convert('RGBA')
+        image = image.convert('RGBA')
         data = np.array(image)
         data[..., 3] = 254
         new_image = Image.fromarray(data, 'RGBA')
@@ -128,7 +142,7 @@ class ImageMetadata(BaseModel):
         new_image.save(_new_img_io, format="PNG")
         return _new_img_io
 
-    def apply_to_image(self, image: IMAGE_INPUT_TYPE, *, inject_lsb: bool = True) -> BytesIO:
+    def apply_to_image(self, image: Image.Image, *, inject_lsb: bool = True) -> BytesIO:
         """
         Write metadata to origin_image
         If you set inject_lsb to True, the image will be injected with metadata using LSB.
@@ -138,10 +152,9 @@ class ImageMetadata(BaseModel):
         :param inject_lsb:  Inject metadata using LSB
         :return:  BytesIO
         """
-        image_data = get_image_bytes(image)
-        origin_image = Image.open(image_data)
+        origin_image = image
         metadata = PngInfo()
-        for k, v in self.model_dump(mode="json").items():
+        for k, v in self.model_dump(mode="json", exclude_none=True).items():
             if isinstance(v, dict):
                 v = json.dumps(v)
             metadata.add_text(k, v)
@@ -155,7 +168,7 @@ class ImageMetadata(BaseModel):
         return be_copy_image
 
     @classmethod
-    def _extract_metadata_from_lsb(cls, image: IMAGE_INPUT_TYPE):
+    def _extract_metadata_from_lsb(cls, image: Image.Image):
         """
         Extract metadata using LSB extraction method.
         :param image: IMAGE_INPUT_TYPE
@@ -168,38 +181,50 @@ class ImageMetadata(BaseModel):
             return None
 
     @classmethod
-    def _extract_metadata_from_comments(cls, image: IMAGE_INPUT_TYPE):
+    def _extract_metadata_from_comments(cls, image: Image.Image):
         """
         Extract metadata from image comments and other info.
         :param image: Union[str, bytes, Path, BytesIO]
         """
-        with Image.open(image) as img:
-            title = img.info.get("Title", None)
-            description = img.info.get("Description", None)
-            comment = img.info.get("Comment", "{}")
-            try:
-                comment_dict = json.loads(comment)
-            except Exception as e:
-                logger.trace(f"Error loading comment: {e}")
-                comment_dict = {}
+        generate_map = image.info
+        comment = generate_map.get("Comment", "{}")
         try:
-            comment_dict.setdefault("prompt", description)
-            comment_model = cls.CommentModel.model_validate(comment_dict)
-            return cls(Title=title, Description=description, Comment=comment_model)
+            comment_dict = json.loads(comment)
         except Exception as e:
-            logger.debug(f"Error loading comment: {e}")
-            return None
+            raise BrokenMetaDataError(
+                f"ImageMetadata: non-standard comment json cant be loaded, given image maybe be modified!"
+            ) from e
+        comment_dict["prompt"] = comment_dict.get("prompt", "")
+        try:
+            comment_model = CommentModel.model_validate(comment_dict)
+        except Exception as e:
+            raise BrokenCommentError(
+                f"ImageMetadata: comment model validate failed, given image maybe be modified."
+                "If you are sure that the image is from novelai.net, please issue this problem to "
+                "https://github.com/LlmKira/novelai-python/issues."
+            ) from e
+        try:
+            return cls.model_validate(
+                {
+                    **generate_map,
+                    "Comment": comment_model.model_dump(mode="python"),
+                }
+            )
+        except Exception as e:
+            raise BrokenMetaDataError(
+                f"ImageMetadata: some error occurred when validate the model, are the image from novelai.net?"
+            ) from e
 
     @classmethod
-    def load_image(cls, image: IMAGE_INPUT_TYPE):
+    def load_image(cls, image: Image.Image):
         """
         Load image and extract metadata using LSB/Metadata
         :param image: str, bytes, Path, BytesIO
         :return: ImageMetadata
         :raises ValueError: Data extraction failed
         """
-        image_data = get_image_bytes(image)
-        metadata = cls._extract_metadata_from_lsb(image_data)
+
+        metadata = cls._extract_metadata_from_lsb(image)
         if metadata:
             return metadata
         metadata = cls._extract_metadata_from_comments(image)
@@ -207,12 +232,37 @@ class ImageMetadata(BaseModel):
             return metadata
         raise ValueError("No metadata found")
 
+    @classmethod
+    def load_from_watermark(cls, image: Image.Image):
+        """
+        Load image and extract metadata using LSB/Metadata
+        :param image: str, bytes, Path, BytesIO
+        :return: ImageMetadata
+        :raises ValueError: Data extraction failed
+        """
+        metadata = cls._extract_metadata_from_lsb(image)
+        if metadata:
+            return metadata
+        raise ValueError("No metadata found")
+
+    @classmethod
+    def load_from_pnginfo(cls, image: Image.Image):
+        """
+        Load image and extract metadata using LSB/Metadata
+        :param image: str, bytes, Path, BytesIO
+        :return: ImageMetadata
+        :raises ValueError: Data extraction failed
+        """
+        metadata = cls._extract_metadata_from_comments(image)
+        if metadata:
+            return metadata
+        raise ValueError("No metadata found")
+
 
 class ImageVerifier:
 
     @staticmethod
     def verify_latents(image: Image.Image, signed_hash: bytes, verify_key: VerifyKey):
-        image.load()
         sig = None
         latents = None
         try:
@@ -248,20 +298,19 @@ class ImageVerifier:
             return False, False, None
 
     def verify(self,
-               image,
+               image: Image.Image,
                verify_key_hex: str = "Y2JcQAOhLwzwSDUJPNgL04nS0Tbqm7cSRc4xk0vRMic="
                ) -> Tuple[bool, bool]:
         """
         Verify if the image is a NovelAI generated image
 
-        :param image: Union[str, bytes, Path, BytesIO] - The input image to verify.
+        :param image: PIL.Image.Image - The image to verify.
         :param verify_key_hex: str - The verification key in base64 format.
         :return: Tuple[bool, bool] - The first bool indicates if the image is verified, the second bool indicates if the image has latents.
         :raises ValueError: If the required metadata or signed hash is missing.
         """
-        image_obj = Image.open(get_image_bytes(image))
-        w, h = image_obj.size
-        image_array = np.array(image_obj)
+        w, h = image.size
+        image_array = np.array(image)
         try:
             metadata, fec_data = ImageLsbDataExtractor().extract_data(image, get_fec=True)
         except Exception as e:
@@ -277,7 +326,7 @@ class ImageVerifier:
         # Build verify key
         verify_key = VerifyKey(verify_key_hex.encode("utf-8"), encoder=Base64Encoder)
         # Verify latents
-        good_latents, have_latents, latents = self.verify_latents(image_obj, signed_hash, verify_key)
+        good_latents, have_latents, latents = self.verify_latents(image, signed_hash, verify_key)
         if not good_latents:
             return False, False
         rgb = image_array[:, :, :3].tobytes()