Merge pull request #78 from LlmKira/dev

(refactor): Add ImageVerifier for tool.image_metadata / add bch_utils

playground/image_metadata/read_lsb.py

@@ -1,4 +1,3 @@
-import os
 from pathlib import Path
 
 from novelai_python.tool.image_metadata.lsb_extractor import ImageLsbDataExtractor

playground/image_metadata/read_nai_tag.py

@@ -2,14 +2,13 @@
 # @Time    : 2024/2/8 下午4:57
 # @Author  : sudoskys
 # @File    : read_nai_tag.py
-from io import BytesIO
 from pathlib import Path
 
-from novelai_python.tool.image_metadata import ImageMetadata
+from novelai_python.tool.image_metadata import ImageMetadata, ImageVerifier
 
 image = Path(__file__).parent.joinpath("sample-0316.png")
 image_clear = ImageMetadata.reset_alpha(
-    input_img=BytesIO(image.read_bytes())
+    image=image
 )
 
 try:
@@ -18,23 +17,11 @@
     raise LookupError("Cant find a MetaData")
 
 print(meta.Title)
-print(meta.Description)
-print(meta.Comment)
-
-try:
-    meta = ImageMetadata.load_image(image_clear)
-    print(meta.Title)
-    print(meta.Description)
-    print(meta.Comment)
-except ValueError:
-    print("Cant find a MetaData")
-
-image = Path(__file__).parent.joinpath("sample-0317.png")
-try:
-    meta = ImageMetadata.load_image(image)
-except ValueError:
-    raise LookupError("Cant find a MetaData")
-
-print(meta.Title)
-print(meta.Description)
-print(meta.Comment)
+print(f"Description: {meta.Description}")
+print(f"Comment: {meta.Comment}")
+print(f"Request Method: {meta.Comment.request_type}")
+print(f"Used image model: {meta.used_model}")
+# Verify if the image is from NovelAI
+is_novelai, have_latent = ImageVerifier().verify(image=image)
+print(f"Is NovelAI: {is_novelai}")
+print(f"Have Latent: {have_latent}")

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "novelai-python"
-version = "0.5.1"
+version = "0.5.2"
 description = "NovelAI Python Binding With Pydantic"
 authors = [
     { name = "sudoskys", email = "[email protected]" },
@@ -28,6 +28,7 @@ dependencies = [
     "tokenizers>=0.15.2",
     "json-repair>=0.29.4",
     "robust-downloader>=0.0.2",
+    "bchlib>=2.1.3",
 ]
 requires-python = ">=3.9"
 readme = "README.md"