Merge pull request #15 from parishwolfe/Fix-image_util

fixes #14

Author: parishwolfe

img/test_1.png

@@ -1,6 +1,24 @@
 from PIL import Image, ImageDraw, ImageFont
-
-def create_text_image(text: str, height: int, width: int, file_name: str, color: str="#000000"):
+import os
+
+
+def get_text_width(font, text):
+    if hasattr(font, 'getlength'):
+        # For newer versions of Pillow
+        return font.getlength(text)
+    elif hasattr(font, 'getsize'):
+        # For older versions of Pillow
+        return font.getsize(text)[0]
+    elif hasattr(font, 'getbbox'):
+        # Alternative method
+        bbox = font.getbbox(text)
+        return bbox[2] - bbox[0]
+    else:
+        raise AttributeError(
+            "Font object has no method to calculate text width.")
+
+
+def create_text_image(text: str, height: int, width: int, file_name: str, color: str = "#000000"):
     """
     Creates an image with the specified text centered within the given dimensions and saves it as a PNG file.
     Args:
@@ -27,14 +45,48 @@ def create_text_image(text: str, height: int, width: int, file_name: str, color:
     best_wrapped_text = None
     best_total_height = None
 
-    # Load the font once to avoid repetitive loading
-    font_path = "arial.ttf"  # Ensure this font is available on your system
+    # Function to find a font that exists on the system
+    def find_font():
+        possible_fonts = [
+            # Common fonts on Mac
+            "/Library/Fonts/Arial.ttf",
+            "/System/Library/Fonts/Supplemental/Arial.ttf",
+            "/Library/Fonts/Helvetica.ttf",
+            "/System/Library/Fonts/Supplemental/Helvetica.ttf",
+            "/Library/Fonts/Times New Roman.ttf",
+            # Common fonts on Windows
+            "C:\\Windows\\Fonts\\Arial.ttf",
+            "C:\\Windows\\Fonts\\times.ttf",
+            "C:\\Windows\\Fonts\\verdana.ttf",
+            # Common fonts on Linux
+            "/usr/share/fonts/truetype/dejavu/DejaVuSans.ttf",
+            # PIL default font
+            "arial.ttf",
+            "DejaVuSans.ttf",  # Comes with matplotlib
+        ]
+        for font_path in possible_fonts:
+            if os.path.exists(font_path):
+                return font_path
+        # If none of the above fonts exist, use PIL's default font
+        return None
+
+    font_path = find_font()
+    if font_path is None:
+        # Use PIL's default font
+        font = ImageFont.load_default()
+    else:
+        font = None  # We'll set the font in the loop
 
     while min_font_size <= max_font_size:
         font_size = (min_font_size + max_font_size) // 2
-        font = ImageFont.truetype(font_path, size=font_size)
+        if font_path is None:
+            # Use default font
+            font = ImageFont.load_default()
+        else:
+            font = ImageFont.truetype(font_path, size=font_size)
 
-        fits, wrapped_text, total_height = does_text_fit(draw, text, font, width, height)
+        fits, wrapped_text, total_height = does_text_fit(
+            draw, text, font, width, height)
         if fits:
             # This font size fits, try a bigger one
             best_font_size = font_size
@@ -47,10 +99,15 @@ def create_text_image(text: str, height: int, width: int, file_name: str, color:
 
     if best_wrapped_text is None:
         # Text does not fit even at the minimum font size
-        raise ValueError("Text cannot fit into the image at the minimum font size.")
+        raise ValueError(
+            "Text cannot fit into the image at the minimum font size.")
 
     # Use the best font size
-    font = ImageFont.truetype(font_path, size=best_font_size)
+    if font_path is None:
+        # Use default font
+        font = ImageFont.load_default()
+    else:
+        font = ImageFont.truetype(font_path, size=best_font_size)
     ascent, descent = font.getmetrics()
     line_spacing = int(best_font_size * 0.2)  # 20% of font size
 
@@ -62,7 +119,8 @@ def create_text_image(text: str, height: int, width: int, file_name: str, color:
 
     # Draw the text on the image
     for line in best_wrapped_text:
-        text_width = font.getlength(line)
+        # Get text width
+        text_width = get_text_width(font, line)
         x = (width - text_width) / 2  # Center horizontally
         draw.text((x, y_offset), line, fill=color, font=font)
         y_offset += ascent + descent + line_spacing  # Move down for the next line
@@ -71,6 +129,7 @@ def create_text_image(text: str, height: int, width: int, file_name: str, color:
     image.save(file_name, "PNG")
     return file_name
 
+
 def does_text_fit(draw, text, font, width, height):
     ascent, descent = font.getmetrics()
     line_spacing = int(font.size * 0.2)  # 20% of font size
@@ -83,7 +142,8 @@ def does_text_fit(draw, text, font, width, height):
         current_line = ""
         for word in words:
             test_line = f"{current_line} {word}".strip()
-            text_width = font.getlength(test_line)
+            # Get text width
+            text_width = get_text_width(font, test_line)
             if text_width <= width:
                 current_line = test_line
             else:
@@ -97,31 +157,40 @@ def does_text_fit(draw, text, font, width, height):
 
     # Calculate total height for the wrapped text
     num_lines = len(wrapped_text)
-    total_height = (ascent + descent) * num_lines + line_spacing * (num_lines - 1)
+    total_height = (ascent + descent) * num_lines + \
+        line_spacing * (num_lines - 1)
 
     # Check if any line exceeds width
-    any_line_too_wide = any(font.getlength(line) > width for line in wrapped_text)
+    any_line_too_wide = any(get_text_width(font, line)
+                            > width for line in wrapped_text)
 
     # Return whether text fits, the wrapped_text, and total_height
     fits = (total_height <= height) and not any_line_too_wide
     return fits, wrapped_text, total_height
 
 
 if __name__ == "__main__":
+    # Create output directory if it doesn't exist
+    output_dir = "../img"
+    if not os.path.exists(output_dir):
+        os.makedirs(output_dir)
+
     # Example usage:
     create_text_image(
         "Lorem ipsum dolor sit amet, consectetur adipiscing elit, sed do eiusmod tempor incididunt ut labore et dolore magna aliqua.",
         800,
         1200,
-        "img/test_1.png",
+        os.path.join(output_dir, "test_1.png"),
         "#000000"
     )
     create_text_image(
         "Lorem ipsum dolor sit amet, consectetur adipiscing elit, sed do eiusmod tempor incididunt ut labore et dolore magna aliqua.",
         800,
         400,
-        "img/test_2.png",
+        os.path.join(output_dir, "test_2.png"),
         "#FF5733"
     )
-    create_text_image("Lorem ipsum dolor sit amet", 800, 1200, "img/test_3.png")
-    create_text_image("Lorem ipsum dolor sit amet", 800, 400, "img/test_4.png", "#FFFFFF")
+    create_text_image("Lorem ipsum dolor sit amet", 800, 1200,
+                      os.path.join(output_dir, "test_3.png"))
+    create_text_image("Lorem ipsum dolor sit amet", 800, 400,
+                      os.path.join(output_dir, "test_4.png"), "#FFFFFF")

img/test_2.png

@@ -0,0 +1,127 @@
+from PIL import Image, ImageDraw, ImageFont
+
+def create_text_image(text: str, height: int, width: int, file_name: str, color: str="#000000"):
+    """
+    Creates an image with the specified text centered within the given dimensions and saves it as a PNG file.
+    Args:
+        text (str): The text to be displayed on the image.
+        height (int): The height of the image in pixels.
+        width (int): The width of the image in pixels.
+        file_name (str): The name of the file to save the image as, including the file extension (e.g., 'image.png').
+        color (str, optional): The color of the text in hexadecimal format (default is black, "#000000").
+    Raises:
+        ValueError: If the text cannot fit into the image at the minimum font size.
+    Returns:
+        None
+    """
+
+    # Create a blank transparent image
+    image = Image.new("RGBA", (width, height), (255, 255, 255, 0))
+    draw = ImageDraw.Draw(image)
+
+    # Font parameters
+    min_font_size = 1
+    max_font_size = 500  # Set an upper limit for font size
+
+    best_font_size = min_font_size
+    best_wrapped_text = None
+    best_total_height = None
+
+    # Load the font once to avoid repetitive loading
+    font_path = "arial.ttf"  # Ensure this font is available on your system
+
+    while min_font_size <= max_font_size:
+        font_size = (min_font_size + max_font_size) // 2
+        font = ImageFont.truetype(font_path, size=font_size)
+
+        fits, wrapped_text, total_height = does_text_fit(draw, text, font, width, height)
+        if fits:
+            # This font size fits, try a bigger one
+            best_font_size = font_size
+            best_wrapped_text = wrapped_text
+            best_total_height = total_height
+            min_font_size = font_size + 1
+        else:
+            # Font size too big, try a smaller one
+            max_font_size = font_size - 1
+
+    if best_wrapped_text is None:
+        # Text does not fit even at the minimum font size
+        raise ValueError("Text cannot fit into the image at the minimum font size.")
+
+    # Use the best font size
+    font = ImageFont.truetype(font_path, size=best_font_size)
+    ascent, descent = font.getmetrics()
+    line_spacing = int(best_font_size * 0.2)  # 20% of font size
+
+    # Adjust total height to include the descent of the last line
+    best_total_height += descent
+
+    # Center position for the text
+    y_offset = (height - best_total_height) / 2
+
+    # Draw the text on the image
+    for line in best_wrapped_text:
+        text_width = font.getlength(line)
+        x = (width - text_width) / 2  # Center horizontally
+        draw.text((x, y_offset), line, fill=color, font=font)
+        y_offset += ascent + descent + line_spacing  # Move down for the next line
+
+    # Save the image
+    image.save(file_name, "PNG")
+    return file_name
+
+def does_text_fit(draw, text, font, width, height):
+    ascent, descent = font.getmetrics()
+    line_spacing = int(font.size * 0.2)  # 20% of font size
+
+    # Wrap the text
+    wrapped_text = []
+    lines = text.splitlines()
+    for line in lines:
+        words = line.split()
+        current_line = ""
+        for word in words:
+            test_line = f"{current_line} {word}".strip()
+            text_width = font.getlength(test_line)
+            if text_width <= width:
+                current_line = test_line
+            else:
+                if not current_line:
+                    # Single word longer than width, cannot fit
+                    return False, None, None
+                wrapped_text.append(current_line)
+                current_line = word
+        if current_line:
+            wrapped_text.append(current_line)
+
+    # Calculate total height for the wrapped text
+    num_lines = len(wrapped_text)
+    total_height = (ascent + descent) * num_lines + line_spacing * (num_lines - 1)
+
+    # Check if any line exceeds width
+    any_line_too_wide = any(font.getlength(line) > width for line in wrapped_text)
+
+    # Return whether text fits, the wrapped_text, and total_height
+    fits = (total_height <= height) and not any_line_too_wide
+    return fits, wrapped_text, total_height
+
+
+if __name__ == "__main__":
+    # Example usage:
+    create_text_image(
+        "Lorem ipsum dolor sit amet, consectetur adipiscing elit, sed do eiusmod tempor incididunt ut labore et dolore magna aliqua.",
+        800,
+        1200,
+        "img/test_1.png",
+        "#000000"
+    )
+    create_text_image(
+        "Lorem ipsum dolor sit amet, consectetur adipiscing elit, sed do eiusmod tempor incididunt ut labore et dolore magna aliqua.",
+        800,
+        400,
+        "img/test_2.png",
+        "#FF5733"
+    )
+    create_text_image("Lorem ipsum dolor sit amet", 800, 1200, "img/test_3.png")
+    create_text_image("Lorem ipsum dolor sit amet", 800, 400, "img/test_4.png", "#FFFFFF")