Merge pull request #1284 from neptune-ai/aw/fix_jpeg_reco

fix image recognition

src/neptune/attributes/series/file_series.py

@@ -15,7 +15,7 @@
 #
 __all__ = ["FileSeries"]
 
-import imghdr
+import io
 import os
 import pathlib
 from typing import (
@@ -24,6 +24,11 @@
     Optional,
 )
 
+from PIL import (
+    Image,
+    UnidentifiedImageError,
+)
+
 from neptune.attributes.series.series import Series
 from neptune.exceptions import (
     FileNotFound,
@@ -77,8 +82,10 @@ def _get_base64_image_content(file: File) -> str:
         else:
             file_content = file.content
 
-        ext = imghdr.what("", h=file_content)
-        if not ext:
+        try:
+            with Image.open(io.BytesIO(file_content)):
+                ...
+        except UnidentifiedImageError:
             raise OperationNotSupported(
                 "FileSeries supports only image files for now. Other file types will be implemented in future."
             )