allow pil image as input

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
+from thumbhash.encode import image_to_thumbhash as image_to_thumbhash, pil_image_to_thumbhash
 
-__all__ = ["image_to_thumbhash", "thumbhash_to_image"]
+__all__ = ["image_to_thumbhash", "thumbhash_to_image", "pil_image_to_thumbhash"]

thumbhash/encode.py

@@ -12,7 +12,11 @@
 def image_to_thumbhash(
     image: Union[str, bytes, Path, BinaryIO],
 ) -> str:
-    m_image = exif_transpose(Image.open(image)).convert("RGBA")
+    return pil_image_to_thumbhash(Image.open(image))
+
+
+def pil_image_to_thumbhash(image: Image.Image) -> str:
+    m_image = exif_transpose(image).convert("RGBA")
 
     m_image.thumbnail((100, 100))