import io
# Rust import
from .modern_colorthief import *
__doc__ = modern_colorthief.__doc__
__version__ = modern_colorthief.__version__
[docs]
def get_palette(
image: str | bytes | io.BytesIO,
color_count: int | None = 10,
quality: int | None = 10,
) -> list[tuple[int, int, int]]:
if isinstance(image, str):
return _get_palette_given_location(image, color_count, quality)
if isinstance(image, bytes):
return _get_palette_given_bytes(image, color_count, quality)
if isinstance(image, io.BytesIO):
return _get_palette_given_bytes(image.getvalue(), color_count, quality)
[docs]
def get_color(
image: str | bytes | io.BytesIO,
quality: int | None = 10,
) -> tuple[int, int, int]:
if isinstance(image, str):
return _get_color_given_location(image, quality)
if isinstance(image, bytes):
return _get_color_given_bytes(image, quality)
if isinstance(image, io.BytesIO):
return _get_color_given_bytes(image.getvalue(), quality)