From 48f91b74e2c7ef6d31c094eafa5332784a275a8b Mon Sep 17 00:00:00 2001 From: Mel Massadian Date: Wed, 13 Mar 2024 01:20:14 +0100 Subject: [PATCH] =?UTF-8?q?fix:=20=F0=9F=90=9B=20support=20pillow=20<=2010?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit It seems I did this after VLM nodes installed it. This fix handle both cases. closes #155 --- nodes/generate.py | 8 +++++++- requirements.txt | 2 +- 2 files changed, 8 insertions(+), 2 deletions(-) diff --git a/nodes/generate.py b/nodes/generate.py index 449f22e..ecc7734 100644 --- a/nodes/generate.py +++ b/nodes/generate.py @@ -333,9 +333,15 @@ class TextToImage: else: # bottom y_text = (height - (line_height_px * len(lines))) - v_offset + def get_width(line): + if hasattr(font, "getsize"): + return font.getsize(line)[0] + else: + return font.getlength(line) + # Draw each line of text for line in lines: - line_width = font.getlength(line) + line_width = get_width(line) # Horizontal alignment if h_align == "left": x_text = h_offset diff --git a/requirements.txt b/requirements.txt index a5a35e8..e3079d8 100644 --- a/requirements.txt +++ b/requirements.txt @@ -7,4 +7,4 @@ imageio_ffmpeg rich rich_argparse matplotlib -pillow>=10 +pillow