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:
Mose Müller 2023-08-10 10:24:15 +02:00
parent 8d55f3b853
commit a333de9957
3 changed files with 57 additions and 18 deletions

View File

@ -160,6 +160,7 @@ export const GenericComponent = React.memo(
readOnly={attribute.readonly} readOnly={attribute.readonly}
docString={attribute.doc} docString={attribute.doc}
// Add any other specific props for the ImageComponent here // Add any other specific props for the ImageComponent here
format={attribute.value['format']['value'] as string}
/> />
); );
} else { } else {

View File

@ -9,33 +9,36 @@ interface ImageComponentProps {
value: string; value: string;
readOnly: boolean; readOnly: boolean;
docString: string; docString: string;
// Define your component specific props here format: string;
} }
export const ImageComponent = React.memo((props: ImageComponentProps) => { export const ImageComponent = React.memo((props: ImageComponentProps) => {
const renderCount = useRef(0); const renderCount = useRef(0);
const { name, parentPath, value, docString } = props; const { name, parentPath, value, docString, format } = props;
// Your component logic here
useEffect(() => { useEffect(() => {
renderCount.current++; renderCount.current++;
console.log(value);
}); });
return ( return (
<div className={'imageComponent'} id={parentPath.concat('.' + name)}> <div className={'imageComponent'} id={parentPath.concat('.' + name)}>
{process.env.NODE_ENV === 'development' && (
<p>Render count: {renderCount.current}</p>
)}
<DocStringComponent docString={docString} />
{/* Your component JSX here */}
<Card> <Card>
<Card.Header <Card.Header
style={{ cursor: 'pointer' }} // Change cursor style on hover style={{ cursor: 'pointer' }} // Change cursor style on hover
> >
{name} {name}
</Card.Header> </Card.Header>
<Image src={`data:image/jpeg;base64,${value}`}></Image>
{process.env.NODE_ENV === 'development' && (
<p>Render count: {renderCount.current}</p>
)}
<DocStringComponent docString={docString} />
{/* Your component JSX here */}
{format === '' && value === '' ? (
<p>No image set in the backend.</p>
) : (
<Image src={`data:image/${format.toLowerCase()};base64,${value}`}></Image>
)}
</Card> </Card>
</div> </div>
); );

View File

@ -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 from pydase.data_service.data_service import DataService
@ -6,13 +11,43 @@ from pydase.data_service.data_service import DataService
class Image(DataService): class Image(DataService):
def __init__( def __init__(
self, self,
value: bytes | str = "",
) -> None: ) -> None:
self.value = value self._value: str = ""
self._format: str = ""
super().__init__() super().__init__()
def __setattr__(self, __name: str, __value: Any) -> None: @property
if __name == "value": def value(self) -> str:
if isinstance(__value, bytes): return self._value
__value = __value.decode()
return super().__setattr__(__name, __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...")