red ebalo
This commit is contained in:
parent
a2fa597921
commit
023fd3207d
8 changed files with 150 additions and 0 deletions
70
render.py
Normal file
70
render.py
Normal file
|
@ -0,0 +1,70 @@
|
|||
from PIL import Image, ImageDraw, ImageFont
|
||||
import sys
|
||||
import io
|
||||
|
||||
|
||||
def render_text_on_image(input_image_path, text_to_draw, output_image_path=None, font_path="arial-unicode-ms-regular.ttf", font_size=52):
|
||||
image = Image.open(input_image_path)
|
||||
|
||||
draw = ImageDraw.Draw(image)
|
||||
|
||||
font = ImageFont.truetype(font_path, font_size)
|
||||
|
||||
text_color = (255, 255, 255)
|
||||
|
||||
image_width, image_height = image.size
|
||||
|
||||
def wrap_text(text, draw, font, max_width):
|
||||
words = text.split()
|
||||
lines = []
|
||||
current_line = []
|
||||
current_line_width = 0
|
||||
|
||||
for word in words:
|
||||
word_width, word_height = draw.textbbox((0, 0), word, font=font)[2:4]
|
||||
space_width = draw.textbbox((0, 0), ' ', font=font)[2]
|
||||
|
||||
if current_line and (current_line_width + word_width + space_width) > max_width:
|
||||
lines.append(' '.join(current_line))
|
||||
current_line = [word]
|
||||
current_line_width = word_width
|
||||
else:
|
||||
current_line.append(word)
|
||||
current_line_width += word_width + space_width
|
||||
|
||||
if current_line:
|
||||
lines.append(' '.join(current_line))
|
||||
|
||||
return lines
|
||||
|
||||
lines = wrap_text(text_to_draw, draw, font, image_width - 20)
|
||||
|
||||
total_text_height = sum([draw.textbbox((0, 0), line, font=font)[3] for line in lines])
|
||||
|
||||
current_y = 10
|
||||
|
||||
for line in lines:
|
||||
draw.text((10, current_y), line, font=font, fill=text_color)
|
||||
|
||||
current_y += draw.textbbox((0, 0), line, font=font)[3] + 5
|
||||
|
||||
if current_y > image_height:
|
||||
break
|
||||
|
||||
if output_image_path:
|
||||
image.save(output_image_path)
|
||||
|
||||
else:
|
||||
file = io.BytesIO()
|
||||
|
||||
image.thumbnail((512, 512))
|
||||
|
||||
image.save(file, format="WEBP")
|
||||
|
||||
file.seek(0)
|
||||
|
||||
return file
|
||||
|
||||
if __name__ == "__main__":
|
||||
|
||||
render_text_on_image("red_ebalo.png", " ".join(sys.argv[1:]), output_image_path="red_ebalo_text.png")
|
Loading…
Add table
Add a link
Reference in a new issue