Merge pull request #8 from yupbank/allow-pil-images

Astropilot · Dec 14, 2023 · 3466632 · 3466632
2 parents 34d3920 + 2460745
commit 3466632
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/thumbhash/encode.py b/thumbhash/encode.py
@@ -10,9 +10,12 @@
 
 
 def image_to_thumbhash(
-    image: Union[str, bytes, Path, BinaryIO],
+    image: Union[str, bytes, Path, BinaryIO, Image.Image],
 ) -> str:
-    m_image = exif_transpose(Image.open(image)).convert("RGBA")
+    if not isinstance(image, Image.Image):
+        image = Image.open(image)
+
+    m_image = exif_transpose(image).convert("RGBA")
 
     m_image.thumbnail((100, 100))