fix error message
This commit is contained in:
parent
a10d6568f1
commit
e5b91cda9b
@ -69,18 +69,21 @@ def render_text_on_image(input_image_path, text_to_draw, output_image_path=None,
|
|||||||
WIDTH = 29
|
WIDTH = 29
|
||||||
HEIGHT = 10
|
HEIGHT = 10
|
||||||
|
|
||||||
|
EMPTY_DOT = "◌"
|
||||||
|
FILLED_DOT = "●"
|
||||||
|
|
||||||
|
|
||||||
def image_to_string(image_file):
|
def image_to_string(image_file):
|
||||||
img = Image.open(image_file)
|
img = Image.open(image_file)
|
||||||
|
|
||||||
img = img.resize((WIDTH, HEIGHT), Image.Resampling.NEAREST)
|
img = img.resize((WIDTH, HEIGHT), Image.Resampling.LANCZOS)
|
||||||
|
|
||||||
img = img.convert("1")
|
img = img.convert("1")
|
||||||
|
|
||||||
# Get image dimensions
|
# Get image dimensions
|
||||||
width, height = img.size
|
width, height = img.size
|
||||||
if width != WIDTH or height != HEIGHT:
|
if width != WIDTH or height != HEIGHT:
|
||||||
raise ValueError("The image must be exactly 22x9 pixels.")
|
raise ValueError(f"The image must be exactly {WIDTH}x{HEIGHT} pixels.")
|
||||||
|
|
||||||
# Create the string representation
|
# Create the string representation
|
||||||
pixel_string = ""
|
pixel_string = ""
|
||||||
@ -89,9 +92,9 @@ def image_to_string(image_file):
|
|||||||
# Get the pixel value (0 for black, 255 for white)
|
# Get the pixel value (0 for black, 255 for white)
|
||||||
pixel = img.getpixel((x, y))
|
pixel = img.getpixel((x, y))
|
||||||
if pixel == 0: # Black pixel
|
if pixel == 0: # Black pixel
|
||||||
pixel_string += "◌"
|
pixel_string += EMPTY_DOT
|
||||||
else: # White pixel
|
else: # White pixel
|
||||||
pixel_string += "●"
|
pixel_string += FILLED_DOT
|
||||||
pixel_string += " " # New line for each row
|
pixel_string += " " # New line for each row
|
||||||
|
|
||||||
return pixel_string
|
return pixel_string
|
||||||
|
Loading…
x
Reference in New Issue
Block a user