327 lines
15 KiB
Python
327 lines
15 KiB
Python
#########################################################
|
||
## @file : txt_to_image.py
|
||
## @desc : text content convert to image
|
||
## @create : 2025/6/22
|
||
## @author : Chengan，doubao AI
|
||
## @email : douboer@gmail.com
|
||
#########################################################
|
||
|
||
from logger_utils import CommonLogger
|
||
from PIL import Image, ImageDraw, ImageFont, ImageFilter
|
||
import textwrap
|
||
import random
|
||
import os
|
||
from config import fonts_path, templates, styles
|
||
|
||
class TextToImage:
|
||
"""将文字内容转换为美观图片的类"""
|
||
def __init__(self, log_file=None):
|
||
"""初始化字体和模板路径"""
|
||
self.logger = CommonLogger(log_file).get_logger()
|
||
self.fonts_path = fonts_path
|
||
self.templates = templates
|
||
self.target_size = (1244, 1660) # 目标尺寸：1244x1660
|
||
|
||
def create_image(self, title, content, subtitle=None, signature=None,
|
||
template='minimal',
|
||
output_path='output.png'):
|
||
"""创建文字图片"""
|
||
try:
|
||
# 确保模板存在
|
||
if template not in self.templates:
|
||
raise ValueError(f"模板 '{template}' 不存在")
|
||
template_data = self.templates[template]
|
||
|
||
# 处理背景图片尺寸
|
||
background = self._process_background(template_data['background'])
|
||
draw = ImageDraw.Draw(background)
|
||
width, height = background.size
|
||
padding = template_data['padding']
|
||
|
||
# 从config获取styles
|
||
title_style = styles['title']
|
||
subtitle_style = styles['subtitle']
|
||
body_style = styles['body']
|
||
signature_style = styles['signature']
|
||
top_offset = styles['top_offset']
|
||
bottom_offset = styles['bottom_offset']
|
||
paragraph_spacing = styles['paragraph_spacing'] # 获取段间距
|
||
|
||
# 检查字体文件是否存在
|
||
for key, path in self.fonts_path.items():
|
||
if not os.path.exists(path):
|
||
raise FileNotFoundError(f"字体文件 '{path}' 不存在，请确认已放入fonts目录")
|
||
|
||
# 加载字体
|
||
title_font = ImageFont.truetype(self.fonts_path['title'], title_style['size'])
|
||
subtitle_font = ImageFont.truetype(self.fonts_path['subtitle'], subtitle_style['size'])
|
||
body_font = ImageFont.truetype(self.fonts_path['body'], body_style['size'])
|
||
signature_font = ImageFont.truetype(self.fonts_path['signature'], signature_style['size'])
|
||
|
||
# 绘制标题
|
||
title_x = (width - self._get_text_width(title, title_font, title_style['letter_spacing'])) // 2
|
||
title_y = padding + title_style['top_spacing'] # 使用 top_spacing
|
||
self._draw_text_with_spacing(draw, (title_x, title_y), title,
|
||
title_font, title_style['color'], title_style['letter_spacing'])
|
||
|
||
# 计算标题高度
|
||
title_bbox = draw.textbbox((0, 0), title, font=title_font)
|
||
title_height = title_bbox[3] - title_bbox[1]
|
||
|
||
# 绘制副标题
|
||
current_y = title_y + title_height + title_style['bottom_spacing']
|
||
if subtitle:
|
||
subtitle_x = (width - self._get_text_width(subtitle, subtitle_font, subtitle_style['letter_spacing'])) // 2
|
||
self._draw_text_with_spacing(draw, (subtitle_x, current_y), subtitle,
|
||
subtitle_font, subtitle_style['color'], subtitle_style['letter_spacing'])
|
||
subtitle_bbox = draw.textbbox((0, 0), subtitle, font=subtitle_font)
|
||
subtitle_height = subtitle_bbox[3] - subtitle_bbox[1]
|
||
current_y += subtitle_height + subtitle_style['bottom_spacing']
|
||
|
||
# 分割内容为段落
|
||
paragraphs = content.split('\n\n')
|
||
page_num = 1
|
||
output_paths = []
|
||
for i, paragraph in enumerate(paragraphs):
|
||
if i > 0: # 如果不是第一个段落，添加段间距
|
||
current_y += paragraph_spacing
|
||
|
||
if current_y + self._get_paragraph_height(paragraph, body_font, body_style) > height - bottom_offset:
|
||
# 绘制签名
|
||
if signature:
|
||
self._draw_signature(draw, width, height, padding, signature, signature_font, signature_style)
|
||
# 保存当前图片
|
||
output_path_current = output_path.replace('.png', f'_{page_num}.png')
|
||
background.save(output_path_current)
|
||
self.logger.info(f"图片已保存至: {output_path_current}")
|
||
output_paths.append(output_path_current)
|
||
# 创建新的图片
|
||
background = self._process_background(template_data['background'])
|
||
draw = ImageDraw.Draw(background)
|
||
current_y = top_offset
|
||
page_num += 1
|
||
|
||
# 绘制段落
|
||
max_width = width - padding * 2
|
||
wrapped_text = self._wrap_text(paragraph, body_font, max_width, body_style['letter_spacing'])
|
||
self._draw_multiline_text_with_spacing(
|
||
draw, (padding, current_y), wrapped_text,
|
||
body_font, body_style['color'],
|
||
body_style['letter_spacing'], body_style['line_spacing']
|
||
)
|
||
|
||
# 计算段落高度
|
||
text_bbox = draw.textbbox((0, 0), wrapped_text, font=body_font)
|
||
text_height = (text_bbox[3] - text_bbox[1]) + (body_style['line_spacing'] * (wrapped_text.count('\n') or 1))
|
||
current_y += text_height + body_style['line_spacing']
|
||
|
||
# 绘制签名
|
||
if signature:
|
||
self._draw_signature(draw, width, height, padding, signature, signature_font, signature_style)
|
||
|
||
# 保存最后一张图片
|
||
output_path_current = output_path.replace('.png', f'_{page_num}.png')
|
||
background.save(output_path_current)
|
||
self.logger.info(f"图片已保存至: {output_path_current}")
|
||
output_paths.append(output_path_current)
|
||
|
||
return output_paths
|
||
except Exception as e:
|
||
self.logger.exception(f"创建图片时发生错误: {e}")
|
||
return None
|
||
|
||
def _get_text_width(self, text, font, letter_spacing):
|
||
"""计算包含字间距的文本总宽度"""
|
||
if not text:
|
||
return 0
|
||
bbox = font.getbbox(text)
|
||
return bbox[2] - bbox[0] + (letter_spacing * (len(text) - 1))
|
||
|
||
def _draw_text_with_spacing(self, draw, position, text, font, fill, letter_spacing):
|
||
"""绘制包含字间距的文本"""
|
||
x, y = position
|
||
for char in text:
|
||
draw.text((x, y), char, font=font, fill=fill)
|
||
# 获取字符宽度并加上字间距
|
||
char_width = font.getlength(char)
|
||
x += char_width + letter_spacing
|
||
|
||
def _draw_multiline_text_with_spacing(self, draw, position, text, font, fill,
|
||
letter_spacing, line_spacing):
|
||
"""绘制包含字间距和行间距的多行文本"""
|
||
x, y = position
|
||
lines = text.split('\n')
|
||
for line in lines:
|
||
self._draw_text_with_spacing(draw, (x, y), line, font, fill, letter_spacing)
|
||
# 获取行高并加上行间距
|
||
line_bbox = draw.textbbox((0, 0), line, font=font)
|
||
line_height = line_bbox[3] - line_bbox[1]
|
||
y += line_height + line_spacing
|
||
|
||
def _add_decorations(self, draw, width, height, template_data):
|
||
"""添加装饰元素"""
|
||
padding = template_data['padding']
|
||
color = (30, 30, 30) # 默认装饰颜色
|
||
# 顶部和底部添加细线条
|
||
#draw.line([(padding, padding//2), (width-padding, padding//2)], fill=color, width=1)
|
||
#draw.line([(padding, height-padding//2), (width-padding, height-padding//2)], fill=color, width=1)
|
||
|
||
# 添加几何装饰点
|
||
for _ in range(8):
|
||
x = random.randint(padding, width-padding)
|
||
y = random.randint(padding, height-padding)
|
||
draw.ellipse([(x-2, y-2), (x+2, y+2)], fill=color)
|
||
|
||
def _wrap_text(self, text, font, max_width, letter_spacing):
|
||
"""根据最大宽度和字间距智能换行"""
|
||
if not text:
|
||
return ""
|
||
|
||
avg_char_width = font.getlength('a') + letter_spacing
|
||
max_chars_per_line = int(max_width // avg_char_width)
|
||
|
||
wrapped_text = ""
|
||
lines = text.split('\n')
|
||
|
||
for line in lines:
|
||
if not line:
|
||
wrapped_text += "\n\n"
|
||
continue
|
||
|
||
words = list(line)
|
||
current_line = ""
|
||
current_width = 0
|
||
|
||
for word in words:
|
||
word_width = font.getlength(word) + letter_spacing
|
||
if current_width + word_width > max_width:
|
||
if current_line:
|
||
wrapped_text += current_line + "\n"
|
||
current_line = word
|
||
current_width = word_width
|
||
else:
|
||
wrapped_text += word + "\n"
|
||
current_line = ""
|
||
current_width = 0
|
||
else:
|
||
current_line += word
|
||
current_width += word_width
|
||
if current_line:
|
||
wrapped_text += current_line + "\n"
|
||
|
||
return wrapped_text.strip()
|
||
|
||
def _process_background(self, background_path):
|
||
"""处理背景图片，将宽度调整为1440并等比例缩放高度，然后裁剪或扩展为目标尺寸1244x1660"""
|
||
target_width, target_height = self.target_size
|
||
try:
|
||
# 打开背景图片
|
||
background = Image.open(background_path).convert("RGBA")
|
||
bg_width, bg_height = background.size
|
||
|
||
# 调整宽度为1440并等比例缩放高度
|
||
if bg_width != 1440:
|
||
ratio = 1440 / bg_width
|
||
new_height = int(bg_height * ratio)
|
||
background = background.resize((1440, new_height), Image.LANCZOS)
|
||
|
||
bg_width, bg_height = background.size
|
||
|
||
# 如果图片尺寸大于目标尺寸，进行裁剪
|
||
if bg_width >= target_width and bg_height >= target_height:
|
||
# 计算裁剪区域（居中裁剪）
|
||
left = (bg_width - target_width) // 2
|
||
top = (bg_height - target_height) // 2
|
||
right = left + target_width
|
||
bottom = top + target_height
|
||
background = background.crop((left, top, right, bottom))
|
||
|
||
# 如果图片尺寸小于目标尺寸，进行扩展（白色填充）
|
||
else:
|
||
new_background = Image.new("RGBA", self.target_size, (255, 255, 255, 255))
|
||
# 计算粘贴位置（居中）
|
||
paste_x = (target_width - bg_width) // 2
|
||
paste_y = (target_height - bg_height) // 2
|
||
new_background.paste(background, (paste_x, paste_y))
|
||
background = new_background
|
||
|
||
except Exception as e:
|
||
# 如果背景图片处理失败，创建默认尺寸的白色背景
|
||
self.logger.exception(f"背景图片处理失败: {e}")
|
||
background = Image.new("RGBA", self.target_size, (255, 255, 255, 255))
|
||
|
||
return background
|
||
|
||
def _get_paragraph_height(self, paragraph, font, style):
|
||
"""计算段落的高度"""
|
||
max_width = self.target_size[0] - 2 * self.templates['minimal']['padding']
|
||
wrapped_text = self._wrap_text(paragraph, font, max_width, style['letter_spacing'])
|
||
text_bbox = ImageDraw.Draw(Image.new("RGBA", self.target_size)).textbbox((0, 0), wrapped_text, font=font)
|
||
text_height = (text_bbox[3] - text_bbox[1]) + (style['line_spacing'] * (wrapped_text.count('\n') or 1))
|
||
return text_height
|
||
|
||
def _draw_signature(self, draw, width, height, padding, signature, signature_font, signature_style):
|
||
"""绘制签名"""
|
||
signature_width = self._get_text_width(signature, signature_font, signature_style['letter_spacing'])
|
||
signature_bbox = draw.textbbox((0, 0), signature, font=signature_font)
|
||
signature_height = signature_bbox[3] - signature_bbox[1]
|
||
|
||
# 根据配置确定签名垂直位置
|
||
if signature_style['vertical_position'] == 'bottom':
|
||
# 固定在底部
|
||
signature_y = height - padding - signature_height - signature_style['offset']
|
||
else:
|
||
# 跟随内容流动（原逻辑）
|
||
signature_y = current_y + 40
|
||
|
||
# 根据配置确定签名水平位置
|
||
if signature_style['position'] == 'left':
|
||
signature_x = padding
|
||
elif signature_style['position'] == 'center':
|
||
signature_x = (width - signature_width) // 2
|
||
else: # 'right'
|
||
signature_x = width - padding - signature_width
|
||
|
||
self._draw_text_with_spacing(draw, (signature_x, signature_y), signature,
|
||
signature_font, signature_style['color'], signature_style['letter_spacing'])
|
||
|
||
if __name__ == "__main__":
|
||
# 示例内容
|
||
title = "夏日阅读清单"
|
||
subtitle = "2025年必读书单推荐"
|
||
content = """日本寺院有抽签的风尚，大概也是收入不菲的创收项目吧。一个个签位，标了价格，无人售票，这种方式挺好，没有买卖的功利和压力。
|
||
|
||
不免俗，抽了支，下下，中间有句，大意是人财分离。虽不作兴这个，还是略有不悦。
|
||
|
||
逛完突然想着去吃抹茶小点，老杨发现转角处就有一家抹茶专业店，可惜刚打烊。边上小店点了一个撒着浅草字样的冰激凌，无惊喜。
|
||
|
||
途经一家电玩店，别有洞天，一排排整齐划一，机器挨着机器，每一排机器造型不同，应该超过500台，空间被利用到极致。充斥着机器电音，令人想起工业朋
|
||
克。一簇簇坐着的玩家，电子烟是标配，60%是中老年人，秃头大爷和白发老奶不少见，并无少年，颠覆我对游戏厅的印象。
|
||
|
||
这里机器是掌控者，人是机器的有机体延伸，机器设定规则，发出吼叫，刺激神经，填满有机体空虚孤独的心。比起从站台跳下去，被机器奴役也并不算太差，这样想来，风月场也好，游戏厅也好，经营者都有功德。"""
|
||
signature = "@刀波儿"
|
||
|
||
# 确保目录存在
|
||
os.makedirs('fonts', exist_ok=True)
|
||
os.makedirs('backgrounds', exist_ok=True)
|
||
|
||
# 创建简单背景（如果模板不存在）
|
||
if not os.path.exists('backgrounds/minimal.jpg'):
|
||
img = Image.new('RGB', (1244, 1660), color=(240, 240, 240))
|
||
draw = ImageDraw.Draw(img)
|
||
for i in range(0, 1660, 20):
|
||
draw.line([(0, i), (1244, i)], fill=(230, 230, 230))
|
||
img.save('backgrounds/minimal.jpg')
|
||
|
||
# 生成图片
|
||
converter = TextToImage(log_file='logs/txt_to_image.log')
|
||
output_paths = converter.create_image(
|
||
title=title,
|
||
subtitle=subtitle,
|
||
content=content,
|
||
signature=signature,
|
||
template='minimal',
|
||
output_path='temp/reading_list.png'
|
||
)
|
||
print("生成的图片路径:", output_paths)
|