address comment

Author: yupbank

thumbhash/__init__.py

@@ -1,4 +1,4 @@
 from thumbhash.decode import thumbhash_to_image as thumbhash_to_image
-from thumbhash.encode import image_to_thumbhash as image_to_thumbhash, pil_image_to_thumbhash
+from thumbhash.encode import image_to_thumbhash as image_to_thumbhash
 
-__all__ = ["image_to_thumbhash", "thumbhash_to_image", "pil_image_to_thumbhash"]
+__all__ = ["image_to_thumbhash", "thumbhash_to_image"]

thumbhash/encode.py

@@ -9,13 +9,10 @@
 from thumbhash.hash import Hash
 
 
-def image_to_thumbhash(
-    image: Union[str, bytes, Path, BinaryIO],
-) -> str:
-    return pil_image_to_thumbhash(Image.open(image))
+def image_to_thumbhash(image: Union[str, bytes, Path, BinaryIO, Image.Image]) -> str:
+    if not isinstance(image, Image.Image):
+        image = Image.open(image)
 
-
-def pil_image_to_thumbhash(image: Image.Image) -> str:
     m_image = exif_transpose(image).convert("RGBA")
 
     m_image.thumbnail((100, 100))