mirror of
https://github.com/tiqi-group/pydase.git
synced 2026-02-14 06:18:41 +01:00
feat: updating ImageComponent
- adding format - adding method to read image from file - adding method to load image from base64 byte string
This commit is contained in:
@@ -1,4 +1,9 @@
|
||||
from typing import Any
|
||||
import base64
|
||||
import io
|
||||
from pathlib import Path
|
||||
|
||||
import PIL.Image
|
||||
from loguru import logger
|
||||
|
||||
from pydase.data_service.data_service import DataService
|
||||
|
||||
@@ -6,13 +11,43 @@ from pydase.data_service.data_service import DataService
|
||||
class Image(DataService):
|
||||
def __init__(
|
||||
self,
|
||||
value: bytes | str = "",
|
||||
) -> None:
|
||||
self.value = value
|
||||
self._value: str = ""
|
||||
self._format: str = ""
|
||||
super().__init__()
|
||||
|
||||
def __setattr__(self, __name: str, __value: Any) -> None:
|
||||
if __name == "value":
|
||||
if isinstance(__value, bytes):
|
||||
__value = __value.decode()
|
||||
return super().__setattr__(__name, __value)
|
||||
@property
|
||||
def value(self) -> str:
|
||||
return self._value
|
||||
|
||||
@property
|
||||
def format(self) -> str:
|
||||
return self._format
|
||||
|
||||
def load_from_path(self, path: Path | str) -> None:
|
||||
with PIL.Image.open(path) as image:
|
||||
self._load_from_PIL_Image(image)
|
||||
|
||||
def load_from_base64(self, value: bytes) -> None:
|
||||
if isinstance(value, bytes):
|
||||
# Decode the base64 string
|
||||
image_data = base64.b64decode(value)
|
||||
|
||||
# Create a writable memory buffer for the image
|
||||
image_buffer = io.BytesIO(image_data)
|
||||
|
||||
# Read the image from the buffer
|
||||
image = PIL.Image.open(image_buffer)
|
||||
self._load_from_PIL_Image(image)
|
||||
|
||||
def _load_from_PIL_Image(self, image: PIL.Image.Image) -> None:
|
||||
if isinstance(image, PIL.Image.Image):
|
||||
if image.format is not None:
|
||||
self._format = image.format
|
||||
buffered = io.BytesIO()
|
||||
image.save(buffered, format=self._format)
|
||||
img_base64 = base64.b64encode(buffered.getvalue())
|
||||
|
||||
self._value = img_base64.decode()
|
||||
else:
|
||||
logger.error("Image format is 'None'. Skipping...")
|
||||
|
||||
Reference in New Issue
Block a user