13315423919/k3GPT
Template
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity
Files
b5e60059ac8f7890231e67197a64afb63e821443
k3GPT/main/make_poster.py

1378 lines
45 KiB
Python
Raw Normal View History

Add File
2025-11-19 19:42:44 +08:00
from PIL import Image, ImageDraw, ImageFont
import os
from parse_html import parse_html_to_ppt,parse_html_to_poster
# 加载字体 - 使用系统路径
font_path_1 = "/usr/share/fonts/opentype/noto/NotoSansCJK-Thin.ttc"
font_path_2 = "/usr/share/fonts/opentype/noto/NotoSansCJK-Regular.ttc"
font_path_3 = "/usr/share/fonts/opentype/noto/NotoSansCJK-Black.ttc"
font_path_4 = "/usr/share/fonts/opentype/noto/NotoSansCJK-Bold.ttc"
try:
title_font = ImageFont.truetype(font_path_3, 100) # 粗体
subtitle_font = ImageFont.truetype(font_path_1, 60) # 细
end_font = ImageFont.truetype(font_path_1, 50) # 细
section_font = ImageFont.truetype(font_path_2, 100) # 粗体
section_title_font = ImageFont.truetype(font_path_4, 40) # 粗体
section_subtitle_font = ImageFont.truetype(font_path_4, 60) # 粗体
content_bold_font = ImageFont.truetype(font_path_4, 35) # 粗体
keyword_font = ImageFont.truetype(font_path_2, 40) # 中等
content_font = ImageFont.truetype(font_path_2, 35) # 中等
footer_font = ImageFont.truetype(font_path_1, 35) # 常规
except:
raise Exception("缺少思源黑体字体文件")
#颜色组
card_bg_color = None
#颜色转换
def hex_to_rgb(hex_color):
# 去掉可能的 '#' 字符
hex_color = hex_color.lstrip('#')
# 使用 int 和 base=16 将十六进制字符串转换为整数，然后用 ',' 分隔这些值，并创建一个元组
rgb = tuple(int(hex_color[i:i+2], 16) for i in (0, 2, 4))
return rgb
#获取相近色
def get_similar_hsv_color(rgb, dh=0.05, ds=0.1, dv=0.1):
import colorsys
r,g,b = rgb
h, s, v = colorsys.rgb_to_hsv(r/255., g/255., b/255.)
# 调整色调、饱和度和亮度
h = (h + dh) % 1 # 循环
s = max(0, min(1, s + ds))
v = max(0, min(1, v + dv))
r_new, g_new, b_new = [int(x*255) for x in colorsys.hsv_to_rgb(h, s, v)]
return (r_new, g_new, b_new)
import re
def is_english_alpha(s):
return bool(re.fullmatch(r'[A-Za-z]+', s))
#处理多行文本
def draw_multiline_text(draw, text, position, max_width, font, fill, line_spacing=10, only_calc=False):
"""
在指定区域内绘制自动换行的文本
参数:
draw - ImageDraw对象
text - 要绘制的文本
position - 起始位置 (x, y)
max_width - 最大行宽
font - 字体对象
fill - 文本颜色
line_spacing - 行间距 (默认10)
"""
x, y = position
lines = []
# 如果文本包含换行符，先按换行符分割
paragraphs = text.split('\n')
for paragraph in paragraphs:
words = []
# 中英文混合处理：将每个字符视为独立的"词"
for char in paragraph:
if char !='\n' and char !='\r' and char !="\xa0" and not char.isspace():
words.append(char)
# # 添加空格字符作为可能的断点
# if char.isspace():
# words.append('')
space_width = draw.textlength(' ', font=font)
current_line = []
current_width = 0
for i,word in enumerate(words):
word_width = draw.textlength(word, font=font)
#中文超宽符号的处理
if word in ['“','”','《','》','『','』','·']:
#print("中文",word,word_width)
word_width += word_width
elif word.islower() or word.isdigit(): #小写字母和数字要窄一点
#print("字母数字",word,word_width)
word_width -= word_width//5
elif word=="-":
word_width = word_width//2
# 如果当前行宽度加上新词宽度超过最大宽度
if int(current_width) + int(word_width) > int(max_width) and current_line:
#print(''.join(current_line),int(current_width),int(word_width),int(max_width),word)
#如果只剩下最后一个字，就放置在末尾
if len(words) -i ==1:
current_line.append(word)
break
# 将当前行添加到行列表
lines.append(''.join(current_line))
#下一行计数
current_line = [word]
current_width = word_width
else:
current_line.append(word)
current_width += word_width
# 添加最后一行
if current_line:
lines.append(''.join(current_line))
# 计算行高
_, _, _, line_height = font.getbbox("A")
# 绘制每一行文本
for line in lines:
if not only_calc:
draw.text((x, y), line, fill=fill, font=font)
y += line_height + line_spacing
return y
#绘制图片
def draw_image(path_data,image_width):
# 如果有 data:image/...;base64, 前缀，需要去掉
if path_data.startswith('data:image'):
base64_str = path_data.split(',', 1)[1] # 取逗号后面的部分
import base64
from io import BytesIO
# 解码 Base64 为二进制
image_data = base64.b64decode(base64_str)
# 用 BytesIO 包装成“文件对象”
image_file = BytesIO(image_data)
content_img = Image.open(image_file)
else:
content_img = Image.open(path_data)
# 获取原始尺寸
original_width, original_height = content_img.size
# 计算高度的比例
width_ratio = image_width / original_width
target_height = int(original_height * width_ratio)
# 调整图像大小
resized_img = content_img.resize((image_width, target_height), Image.Resampling.LANCZOS)
return resized_img,target_height
#绘制表格
def draw_table(draw,data,table_top,table_left,table_width,header_font,data_font,text_color,header_color,tbc):
if data is None: return table_top
cell_height = 65 # 默认单元格高度
headers=data[0]
data=data[1:] #去掉第一行的空行
rows = len(data)
cols = len(headers)
#table_height = cell_height * rows
cell_width = table_width // cols
# 绘制表头圆角矩形背景
draw.rounded_rectangle(
(table_left-10, table_top, table_left+table_width+10, table_top+cell_height+15),
radius=40,
fill=header_color, # 深蓝色背景
outline=header_color
)
# # 绘制表头分割线
for i in range(1, cols):
x = table_left + i * cell_width
# 绘制垂直线条，稍微短一点以避开圆角
draw.line(
[(x, table_top + 10), (x, table_top + cell_height)],
fill=(255, 255, 255, 180), # 半透明白色
width=2
)
for col in range(cols):
text = headers[col]
# 使用textbbox计算文本尺寸
text_bbox = draw.textbbox((0, 0), text, font=header_font)
text_width = text_bbox[2] - text_bbox[0]
text_height = text_bbox[3] - text_bbox[1]
x = table_left + col * cell_width + (cell_width - text_width) // 2
y = table_top + (cell_height - text_height) // 2
draw.text((x, y), text, fill=(255, 255, 230), font=header_font)
#微调一下数据表格的大小和位置，下移15
table_top +=15
table_width = table_width - 40
table_left +=20
cell_width = table_width // cols
# 绘制数据行的起始位置
y = table_top + cell_height
for row in range(rows):
#计算最大行高
max_y = y + cell_height
for col in range(cols):
text = data[row][col]
y1 = draw_multiline_text(draw, text, (table_left,y), cell_width-20, data_font, text_color, line_spacing=10, only_calc=True)
if y1>max_y:
max_y=y1
row_height = max_y - y
for col in range(cols):
# 计算单元格位置
x1 = table_left + col * cell_width
#y1 = table_top + (row + 1) * cell_height
y1 = y
x2 = x1 + cell_width
y2 = y1 + row_height
# 交替行颜色
if row % 2 == 0:
#bg_color = (248, 250, 253) # 浅色行
bg_color = (tbc[0],tbc[1],tbc[2],100)
else:
#bg_color = (240, 245, 251) # 更浅的蓝色行
bg_color = (tbc[0],tbc[1],tbc[2],200)
# 绘制单元格背景
draw.rectangle([x1, y1, x2, y2], fill=bg_color)
# 添加数据
text = data[row][col]
draw_multiline_text(draw, text, (x1+15,y+10), cell_width-15, data_font, text_color, line_spacing=10, only_calc=False)
# # 使用textbbox计算文本尺寸
# text_bbox = draw.textbbox((0, 0), text, font=data_font)
# text_width = text_bbox[2] - text_bbox[0]
# text_height = text_bbox[3] - text_bbox[1]
# x = x1 + 15
# # 默认y位置
# y = y1 + (cell_height - text_height) // 2
# draw.text((x, y), text, fill=text_color, font=data_font)
#一行的横线
y = max_y+10
draw.line(
[(table_left, y), (table_left+table_width, y)],
fill=(225, 232, 243),
width=1
)
y +=1
#表格高度
table_height = y - table_top
# 固定高度的横线
# for i in range(1, rows):
# y = table_top + i * cell_height
# draw.line(
# [(table_left, y), (table_left+table_width, y)],
# fill=(225, 232, 243),
# width=1
# )
#竖线
for i in range(1, cols):
x = table_left + i * cell_width
draw.line(
[(x, table_top + cell_height), (x, table_top+table_height)],
fill=(225, 232, 243),
width=1
)
#表格边线，三条，, # 深蓝色背景
draw.line(
[(table_left, table_top + cell_height-5), (table_left, table_top+table_height)],
fill=header_color,
width=2
)
draw.line(
[(table_left+table_width, table_top + cell_height-5), (table_left+table_width, table_top+table_height)],
fill=header_color,
width=2
)
draw.line(
[(table_left, table_top + table_height), (table_left+table_width, table_top+table_height)],
fill=header_color,
width=2
)
return table_top+table_height+20
# 辅助函数：获取文本尺寸
def get_text_size(draw,text, font):
bbox = draw.textbbox((0, 0), text, font=font)
return bbox[2] - bbox[0], bbox[3] - bbox[1]
#绘制头部样式1： 大logo+自动居中布局的标题
def draw_header_1(img,draw,poster_data,width,title_color,subtitle_color):
# 1. 顶部标题区域,高度
title = poster_data["主标"]
subtitle = poster_data["副标"]
#主标题和副标题的宽度
title_w, title_h = get_text_size(draw,title, title_font)
subtitle_w, subtitle_h = get_text_size(draw,subtitle, subtitle_font)
# 复制logo
logo_path = poster_data["logo_path"] # logo图片路径
logo = Image.open(logo_path)
logo = logo.resize((400, 150), Image.Resampling.LANCZOS) # 调整logo大小
#标题很长，需要改变布局
if title_w > width-450 or subtitle_w > width-450:
#logo和标题上下布局
# 这里位置是以左上角为起点的坐标(x, y)
position = ((width - 400) // 2, 10) # 上面居中的位置
# 将logo复制到当前图像上
# 使用Image.paste方法，注意如果你的logo有透明度，需要使用mask参数
img.paste(logo, position, logo.convert('RGBA'))
#主标
draw.text(((width - title_w) // 2, 180), title, fill=title_color, font=title_font)
draw.text(((width - subtitle_w) // 2, 180 + title_h + 30),
subtitle, fill=subtitle_color, font=subtitle_font)
y = 180 + title_h + 30 + subtitle_h +60
else:
#logo和标题左右布局
draw.text(((width - title_w) // 2+200, 50), title, fill=title_color, font=title_font)
# 绘制副标题
subtitle_w, subtitle_h = get_text_size(draw,subtitle, subtitle_font)
draw.text(((width - subtitle_w) // 2+200, 50 + title_h + 30),
subtitle, fill=subtitle_color, font=subtitle_font)
# 这里位置是以左上角为起点的坐标(x, y)
position = (10, 10) # 左上角的位置
# 将logo复制到当前图像上
# 使用Image.paste方法，注意如果你的logo有透明度，需要使用mask参数
img.paste(logo, position, logo.convert('RGBA'))
#固定值
y=300
# 装饰元素 - 金色圆点
for i in range(5):
x = width * (0.1 + 0.2 * i)
size = 10 if i % 2 == 0 else 15
draw.ellipse([(x, y),
(x+size, y + size)],
fill=title_color)
return y
#绘制头部样式2： 小logo+ 标题居左
def draw_header_2(img,draw,poster_data,width,title_color,subtitle_color):
# 1. 顶部标题区域,高度
title = poster_data["主标"]
subtitle = poster_data["副标"]
#主标题和副标题的宽度
title_w, title_h = get_text_size(draw,title, title_font)
subtitle_w, subtitle_h = get_text_size(draw,subtitle, subtitle_font)
# 复制logo
image_width = 250 #总宽度的4分之一
logo_path = poster_data["logo_path"] # logo图片路径
logo = Image.open(logo_path)
# 获取原始尺寸
original_width, original_height = logo.size
# 计算高度的比例
width_ratio = 250 / original_width
target_height = int(original_height * width_ratio)
# 调整图像大小
logo = logo.resize((image_width, target_height), Image.Resampling.LANCZOS)
#logo和标题上下布局
# 这里位置是以左上角为起点的坐标(x, y)
position = ((width - image_width)-50, 10) # 右边的位置
# 将logo复制到当前图像上
# 使用Image.paste方法，注意如果你的logo有透明度，需要使用mask参数
img.paste(logo, position, logo.convert('RGBA'))
y = target_height +20
#主标
#draw.text((50, y), title, fill=title_color, font=title_font)
y = draw_multiline_text(
draw,
title,
(50, y),
width-100,
title_font,
title_color,
line_spacing=15,
only_calc = False
)
draw.text((50, y + 30),
subtitle, fill=subtitle_color, font=subtitle_font)
y += 30 + subtitle_h +60
draw.line(
[(0, y), (width, y)],
fill = title_color,
width=1
)
return y
#绘制头部样式2： 无logo+ 标题居左+ 日期
def draw_header_3(img,draw,poster_data,width,title_color,subtitle_color):
# 1. 顶部标题区域,高度
title = poster_data["主标"]
subtitle = poster_data["副标"]
from datetime import date
# 获取当前日期
current_date = date.today()
# 如果需要将日期格式化为特定格式的字符串，可以使用 strftime 方法
formatted_date = current_date.strftime("%Y-%m-%d")
print("格式化后的日期:", formatted_date)
#主标题和副标题的宽度
date_w, date_h = get_text_size(draw,formatted_date, keyword_font)
title_w, title_h = get_text_size(draw,title, title_font)
subtitle_w, subtitle_h = get_text_size(draw,subtitle, subtitle_font)
#日期
y = 20
#主标日期
draw.text(((width-date_w)-40, y), formatted_date, fill=subtitle_color, font=keyword_font)
y = 100
#主标
#draw.text((50, y), title, fill=title_color, font=title_font)
y = draw_multiline_text(
draw,
title,
(50, y),
width-100,
title_font,
title_color,
line_spacing=15,
only_calc = False
)
# 绘制矩形,副标题题点缀
draw.rectangle([(50-10, y+15), (50+subtitle_w+10, y+subtitle_h+25)],
fill=None,
outline=subtitle_color)
draw.text((50, y),
subtitle, fill=subtitle_color, font=subtitle_font)
y += subtitle_h +60
draw.line(
[(0, y), (width, y)],
fill = title_color,
width=4
)
return y
"""
数字卡片，卡片没有标题，有数字，通过后面增加一个区域做标题汇总
"""
def draw_content_card_1(img,draw,poster_data,width,y_start,section_colors,card_bg_color,font_color):
# 2. 内容区域 -多个内容块,流式布局按各自的内容大小来排
section_y_start = y_start
y = section_y_start #内容的绝对值是
#内容
for i,slide in enumerate(poster_data["内容列表"]):
# 计算位置,卡片的位置
y_pos = y
#卡片宽度
card_width = width * 0.9
card_x = int((width - card_width) / 2)
#y从标题颜色后面开始测算
y = y_pos + 50
#文字内容
if slide["content"]:
y = draw_content_list(img,draw,slide,card_x,card_width,card_bg_color,font_color,section_colors,i,y,only_calc= True)
# 卡片高度,保障一个最小值300的空间
if y - y_pos < 150:
y += 100
#卡片高度
card_height = y - y_pos+30
# 创建圆角卡片
card = Image.new('RGBA', (int(card_width), int(card_height)), (0, 0, 0, 0))
card_draw = ImageDraw.Draw(card)
# 圆角矩形填充
card_draw.rounded_rectangle([(0, 0), (card_width, card_height)],
radius=40, fill=card_bg_color)
# 抹去上面的圆角
bar_height = 40
card_draw.rectangle([(0, 0), (card_width, bar_height)],
fill=card_bg_color)
#添加色条
bar_height = 20
card_draw.rectangle([(0, 0), (card_width, bar_height)],
fill=section_colors[i])
# 添加装饰元素
card_draw.ellipse([(card_width - 80, card_height - 80),
(card_width - 20, card_height - 20)],
fill=(255, 255, 255, 30))
# 粘贴卡片到海报
img.paste(card, (int(card_x), int(y_pos)), card)
# 数字标识
num_text = str(i+1)
num_w, num_h = get_text_size(draw,num_text, section_font)
draw.text((card_x + 25, y_pos + 60),
num_text, fill=section_colors[i], font=section_font)
#真正绘制内容
y = y_pos + 50
#文字内容
if slide["content"]:
y = draw_content_list(img,draw,slide,card_x,card_width,card_bg_color,font_color,section_colors,i,y,only_calc= False)
# # 装饰元素 - 小圆点
# for j in range(3):
# draw.ellipse([(card_x + card_width - 180 + j*40, y_pos + card_height/2 - 5),
# (card_x + card_width - 160 + j*40, y_pos + card_height/2 + 15)],
# fill=section_colors[i])
#下一个
y = y_pos + card_height+60
# 3. 底部关键字区域，和内容列表的主标对应
keywords = [content["主标"] for content in poster_data["内容列表"]]
keyword_y = y+10
# 关键字容器
container_height = 180
container = Image.new('RGBA', (int(width*0.90), container_height), (0, 0, 0, 0))
container_draw = ImageDraw.Draw(container)
container_draw.rounded_rectangle([(0, 0), (width*0.90, container_height)],
radius=30, fill=card_bg_color)
# 粘贴容器
img.paste(container, (int(width*0.05), int(keyword_y)), container)
keyword_w = int(width*0.90) // len(keywords)
# 绘制关键字
for idx, kw in enumerate(keywords):
# 计算位置
x_pos = keyword_w //2 + keyword_w*idx + int(width*0.05)
# 绘制图标背景
icon_size = 50
draw.ellipse([(x_pos - icon_size//2, keyword_y + 40),
(x_pos + icon_size//2, keyword_y + 40 + icon_size)],
fill=section_colors[idx])
# 绘制关键字文本
kw_w, kw_h = get_text_size(draw,kw, keyword_font)
draw.text((x_pos - kw_w//2, keyword_y + 100),
kw, fill=font_color, font=keyword_font)
return y+180
"""
带标题卡片，直接在卡片前放置标题
"""
def draw_content_card_2(img,draw,poster_data,width,y_start,section_colors,card_bg_color,font_color):
# 2. 内容区域 -多个内容块,流式布局按各自的内容大小来排
section_y_start = y_start
#卡片宽度
card_width = width * 0.9
card_x = int((width - card_width) / 2)
#内容
y = section_y_start #内容的绝对值是
for i,slide in enumerate(poster_data["内容列表"]):
# 计算位置,卡片的位置
y_pos = y
#y从 主标
y = y_pos + 80 + 30
#文字内容
if slide["content"]:
y = draw_content_list(img,draw,slide,card_x,card_width,card_bg_color,font_color,section_colors,i,y,only_calc= True)
# 卡片高度,保障一个最小值300的空间
if y - y_pos < 150:
y += 100
#卡片高度
card_height = y - y_pos
# 绘制主标,标题
draw.rounded_rectangle([(card_x+card_width*0.055, y_pos), (card_x+card_width*0.95, y_pos+80)],
radius=40, fill=section_colors[i])
num_w, num_h = get_text_size(draw,slide["主标"], section_title_font)
draw.text((card_x + (card_width-num_w)//2, y_pos+5),
slide["主标"], fill=(255,255,255), font=section_title_font)
# 创建圆角卡片
card = Image.new('RGBA', (int(card_width), int(card_height)), (0, 0, 0, 0))
card_draw = ImageDraw.Draw(card)
# 圆角矩形填充
card_draw.rounded_rectangle([(0, 0), (card_width, card_height)],
radius=40, fill=card_bg_color)
# 抹去上面的圆角
bar_height = 50
card_draw.rectangle([(0, 0), (card_width, bar_height)],
fill=card_bg_color)
# 添加装饰元素
card_draw.ellipse([(card_width - 80, card_height - 80),
(card_width - 20, card_height - 20)],
fill=(255, 255, 255, 30))
# 粘贴卡片到海报
img.paste(card, (int(card_x), int(y_pos+100)), card)
#真正绘制内容
y = y_pos + 80 + 30
#文字内容
if slide["content"]:
y = draw_content_list(img,draw,slide,card_x,card_width,card_bg_color,font_color,section_colors,i,y,only_calc= False)
#下一个
y = y_pos +80+30+card_height+60
return y
"""
彩色标签，直接在卡片前放置标题
"""
def draw_content_card_5(img,draw,poster_data,width,y_start,section_colors,card_bg_color,font_color):
# 2. 内容区域 -多个内容块,流式布局按各自的内容大小来排
section_y_start = y_start
#卡片宽度
card_width = width * 0.9
card_x = int((width - card_width) / 2)
#内容
y = section_y_start #内容的绝对值是
for i,slide in enumerate(poster_data["内容列表"]):
# 计算位置,卡片的位置
y_pos = y
#y从 主标开始，80是主标高度
y = y_pos +80 + 40
#文字内容
if slide["content"]:
y = draw_content_list(img,draw,slide,card_x,card_width,card_bg_color,font_color,section_colors,i,y,only_calc= True)
# 卡片高度,保障一个最小值300的空间
if y - y_pos < 150:
y += 100
#卡片高度
card_height = y - y_pos -80
# 绘制主标,标题
# draw.rounded_rectangle([(card_x+card_width*0.055, y_pos), (card_x+card_width*0.95, y_pos+80)],
# radius=40, fill=section_colors[i])
num_w, num_h = get_text_size(draw,slide["主标"], section_title_font)
draw.text((card_x + (card_width-num_w)//2, y_pos+5),
slide["主标"], fill=section_colors[i], font=section_title_font)
# 创建圆角卡片
card = Image.new('RGBA', (int(card_width), int(card_height)), (0, 0, 0, 0))
card_draw = ImageDraw.Draw(card)
# 圆角矩形填充
card_draw.rounded_rectangle([(0, 0), (card_width, card_height)],
radius=40,
fill=card_bg_color)
#抹去左边的圆角
bar_width = 40
card_draw.rectangle([(0, 0), (bar_width, card_height)],
fill=card_bg_color)
# 左边的彩带
bar_width = 20
card_draw.rectangle([(0, 0), (bar_width, card_height)],
fill=section_colors[i])
# 添加装饰元素
card_draw.ellipse([(card_width - 80, 10),
(card_width - 20, 10+60)],
#fill=(255, 255, 255, 30)
fill=section_colors[i]
)
# 粘贴卡片到海报
img.paste(card, (int(card_x), int(y_pos+100)), card)
#真正绘制内容
y = y_pos +80 + 40
#文字内容
if slide["content"]:
y = draw_content_list(img,draw,slide,card_x,card_width,card_bg_color,font_color,section_colors,i,y,only_calc= False)
#下一个
y = y_pos +80+30+card_height+60
return y
"""
小圆角卡片，标题放置在卡片内部
"""
def draw_content_card_6(img,draw,poster_data,width,y_start,section_colors,card_bg_color,font_color):
# 2. 内容区域 -多个内容块,流式布局按各自的内容大小来排
section_y_start = y_start
#卡片宽度
card_width = width * 0.9
card_x = int((width - card_width) / 2)
#内容
y = section_y_start #内容的绝对值是
for i,slide in enumerate(poster_data["内容列表"]):
# 计算位置,卡片的位置
y_pos = y
#y, 80是主标高度
y = y_pos +80 +30
#文字内容 （含副标题）
if slide["content"]:
y = draw_content_list(img,draw,slide,card_x,card_width,card_bg_color,font_color,section_colors,i,y,only_calc= True)
# 卡片高度,保障一个最小值300的空间
if y - y_pos < 150:
y += 100
#卡片高度
card_height = y - y_pos
# 创建圆角卡片
card = Image.new('RGBA', (int(card_width), int(card_height)), (0, 0, 0, 0))
card_draw = ImageDraw.Draw(card)
# 小圆角矩形填充
card_draw.rounded_rectangle([(0, 0), (card_width, card_height)],
radius=15,
fill=card_bg_color,
outline= section_colors[i])
# 粘贴卡片到海报
img.paste(card, (int(card_x), int(y_pos)), card)
#真正绘制内容
y = y_pos
num_w, num_h = get_text_size(draw,slide["主标"], section_title_font)
draw.rectangle([(card_x + 15, y+20+10), (card_x + 15+10, y+20+10+num_h)],
fill= section_colors[i])
draw.text((card_x + 30, y+20),
slide["主标"], fill=section_colors[i], font=section_title_font)
y += num_h + 50
#文字内容(含副标题)
if "content" in slide:
y = draw_content_list(img,draw,slide,card_x,card_width,card_bg_color,font_color,section_colors,i,y,only_calc= False)
#下一个
y = y_pos + card_height+60
return y
#绘制内容列表
def draw_content_list(img,draw,slide,card_x,card_width,bg_color,font_color,section_colors,i,y,only_calc= True):
ol=1
for j,content in enumerate(slide["content"]):
#文字
if "text" in content:
num_w = 40
if len(slide["content"]) ==1:
num_w = 0
font = content_font
if content["level"]==2:
num_w = 0
font = content_bold_font
if "bullet" in content:#二级标题
num_w= 15
if not only_calc:
# draw.rectangle([(card_x +110, y+15), (card_x +110+20, y+15+20)],
# fill=section_colors[i],
# #outline=font_color,
# #width=4,
# )
# 圆形
# draw.ellipse([(card_x +110, y+15), (card_x +110+25, y+15+25)],
# fill=font_color,
# outline=section_colors[i],
# width = 5,
# )
triangle_points = [
(card_x +90+15, y+10), # 顶部顶点
(card_x +90, y+10+30), # 左下顶点
(card_x +90+30, y+10+30) # 右下顶点
]
draw.polygon(triangle_points,
fill=section_colors[i],
#outline=,
#width = 5,
)
if "square" in content:
num_w= 30
if not only_calc:
# 方形填充
draw.rectangle([(card_x +110, y+15), (card_x +110+20, y+15+20)],
fill=section_colors[i],
#outline=font_color,
#width=4,
)
#顺序号
if "number" in content:
num_w= 50
if not only_calc:
# 圆点填充
draw.ellipse([(card_x +110, y+5), (card_x +110+40, y+5+40)],
fill=section_colors[i],
#outline=font_color
)
#数字
draw.text((card_x +110+10, y-2),
str(ol), fill=card_bg_color, font=content_font)
ol +=1
if content["level"]==1:
#副标题
y = draw_multiline_text(
draw,
content["text"],
(card_x +90, y),
card_width-120,
section_title_font,
section_colors[i],
line_spacing=15,
only_calc = only_calc
)
else:
y = draw_multiline_text(
draw,
content["text"],
(card_x +110+num_w, y),
card_width-120-num_w,
font,
font_color,
line_spacing=15,
only_calc = only_calc
)
y +=20
elif "image" in content:
image = content
image_width= int(card_width)-100
resized_img,target_height= draw_image(image["path"],image_width)
draw.text((card_x + 100, y),
image["caption"], fill=font_color, font=footer_font)
img.paste(resized_img, (card_x+50,y+50), resized_img.convert('RGBA'))
y += target_height+65
elif "table" in content:
y = draw_table(draw,content["data"],
y,card_x+40,card_width-80,
keyword_font,content_font,
font_color,section_colors[i],
bg_color
)
y +=20
return y
"""
无卡片布局
"""
def draw_content_3(img,draw,poster_data,width,y_start,section_colors,title_color,bg_color,font_color):
# 2. 内容区域 -多个内容块,流式布局按各自的内容大小来排
section_y_start = y_start
#卡片宽度
card_width = width * 0.9
card_x = int((width - card_width) / 2)
#内容
y = section_y_start #内容的绝对值是
for i,slide in enumerate(poster_data["内容列表"]):
# 计算位置,卡片的位置
y_pos = y
#y从 主标
y = y_pos + 80 + 30
#内容
if slide["content"]:
y = draw_content_list(img,draw,slide,card_x,card_width,bg_color,font_color,section_colors,i,y,only_calc= True)
# 卡片高度,保障一个最小值300的空间
if y - y_pos < 150:
y += 100
#卡片高度
card_height = y - y_pos
# 绘制主标,标题点缀
draw.rectangle([(card_x, y_pos), (card_x+10, y_pos+70)],
fill=section_colors[i])
num_w, num_h = get_text_size(draw,slide["主标"], section_title_font)
draw.text((card_x+20, y_pos+5),
slide["主标"], fill=title_color, font=section_title_font)
#真正绘制内容
y = y_pos + 80 + 30
#文字内容
if slide["content"]:
y = draw_content_list(img,draw,slide,card_x,card_width,bg_color,font_color,section_colors,i,y,only_calc= False)
#下一个
y = y_pos +80+30+card_height+20
return y
"""
无卡片，网格布局
"""
def draw_content_4(img,draw,poster_data,width,y_start,section_colors,title_color,bg_color,font_color):
# 2. 内容区域 -多个内容块,流式布局按各自的内容大小来排
section_y_start = y_start
#卡片宽度
card_width = width * 0.9
card_x = int((width - card_width) / 2)
#内容
y = section_y_start #内容的绝对值是
for i,slide in enumerate(poster_data["内容列表"]):
# 计算位置,卡片的位置
y_pos = y
#y从 主标
y = y_pos + 80 + 30
#文字内容
if slide["content"]:
y = draw_content_list(img,draw,slide,card_x,card_width,bg_color,font_color,section_colors,i,y,only_calc= True)
# 卡片高度,保障一个最小值300的空间
if y - y_pos < 150:
y += 100
#卡片高度
card_height = y - y_pos
#主标题
num_w, num_h = get_text_size(draw,slide["主标"], section_title_font)
draw.text((card_x+20, y_pos+5),
slide["主标"], fill=section_colors[i], font=section_title_font)
# 绘制主标,标题下方点缀
draw.rectangle([(card_x+20, y_pos+30+num_h), (card_x+20+num_w, y_pos+30+num_h+10)],
fill=section_colors[i])
#真正绘制内容
y = y_pos + 80 + 30
#文字内容
if slide["content"]:
y = draw_content_list(img,draw,slide,card_x,card_width,bg_color,font_color,section_colors,i,y,only_calc= False)
#下一个
y = y_pos +80+30+card_height+20
return y
#总结的card
def draw_submit_card(img,draw,poster_data,width,y,title_color,card_bg_color,subtitle_color,style="round"):
card_width = int(width*0.90)
card_x = int(width*0.05)
container_height = draw_multiline_text(
draw,
poster_data["结语"],
(card_x+40, y),
card_width-40,
end_font,
title_color,
line_spacing=15,
only_calc=True
)-y+20
container = Image.new('RGBA', (card_width, container_height), (0, 0, 0, 0))
container_draw = ImageDraw.Draw(container)
if style=="round":
container_draw.rounded_rectangle([(0, 0), (card_width, container_height)],
radius=20, fill=card_bg_color)
elif style=="rect":
container_draw.rectangle([(0, 0), (card_width, container_height)],
fill=card_bg_color)
# 粘贴容器
img.paste(container, (card_x, y), container)
y +=10
y = draw_multiline_text(
draw,
poster_data["结语"],
(card_x+40, y),
card_width-40,
end_font,
subtitle_color,
line_spacing=15
)
return y
#尾部样式1： 渐变圆点
def draw_footer_1(img,draw,poster_data,width,y,title_color,subtitle_color):
# 装饰元素 - 渐变圆点
footer_dots_y = y
#手机一整屏
if footer_dots_y <1980:#内容少的直接到底部
footer_dots_y = 1980
#13个点，一个点大，一个点小
for i in range(13):
x = width * (0.1 + 0.066 * i)
size = 10 + i % 4
color = subtitle_color
draw.ellipse([(x, footer_dots_y), (x+size, footer_dots_y+size)],
fill=color)
#装饰元素 -线
footer_line_y = footer_dots_y+20
draw.line([(width*0.1, footer_line_y), (width*0.9, footer_line_y)],
fill=title_color, width=2)
#海报尾部文字
y = footer_line_y+30
for i,footer in enumerate(poster_data["报尾"]):
if len(footer)<30:
#单行，居中对齐
footer = footer.replace("."," · ")
footer_w, footer_h = get_text_size(draw,footer, footer_font)
draw.text(((width - footer_w) // 2, y),
footer, fill=subtitle_color, font=footer_font)
y += footer_h+10
else:
#多行，居左对齐
y = draw_multiline_text(
draw,
footer,
(40, y),
width-40,
footer_font,
subtitle_color,
line_spacing=15
)
#海报尾部图片
image_width = width //2 #海报的一半
y = y+30
for i,image in enumerate(poster_data["报尾图片"],1):
# 调整图像大小
resized_img,target_height =draw_image(image["path"],image_width)
#图片文字
if image["caption"] and image["caption"][-3:] not in ["png","jpg","peg"]:
draw.text(((width-image_width)//2+10, y+i*10),
image["caption"], fill=subtitle_color, font=footer_font)
y += i*10+60
#图片居中
img.paste(resized_img, ((width-image_width)//2,y), resized_img.convert('RGBA'))
y += target_height+10
return y+60
#手机适合的宣传图
def create_modern_mobile_poster(poster_data,poster_style={}):
# 海报尺寸 (手机竖屏9:16比例)
width, height = 1080, 20000
accent_color = (100, 180, 180) # 青蓝色点缀
# 主色调：深蓝背景 + 金色点缀
bg_color = (15, 30, 64) # 深蓝背景
title_color = (255, 215, 0) # 金色标题
subtitle_color = (220, 220, 240) #亚白
card_bg_color = (30, 50, 80, 220) #浅蓝
section_colors = [
(255,77,90), #粉红
(70, 130, 180), # 蓝色
(100, 180, 160), # 青色
(180, 120, 70), # 橙色
(184, 66, 55), # 砖红
]
font_color = (240, 240, 240) #接近白色
if poster_style:
try:
bg_color = hex_to_rgb(poster_style["bg_color"])
title_color = hex_to_rgb(poster_style["title_color"])
subtitle_color = hex_to_rgb(poster_style["subtitle_color"])
card_bg_color = hex_to_rgb(poster_style["card_bg_color"])
section_colors = [ hex_to_rgb(color) for color in poster_style["section_colors"]]*5
font_color = hex_to_rgb(poster_style["font_color"])
except:
pass
# 创建画布
img = Image.new('RGBA', (width, height),bg_color)
draw = ImageDraw.Draw(img)
if poster_style["style"]==4:
#网格背景
size=50
bg_color1 = get_similar_hsv_color(bg_color)
#横线
for i in range(0,height//size):
y = size//2 + i * size
draw.line(
[(0, y), (width, y)],
#fill=(int(0.299*subtitle_color[0]),int(0.587*subtitle_color[1]),int(0.114*subtitle_color[2])),
fill = bg_color1,
width=1
)
#竖线
for i in range(0, width//size):
x = size//2 + i * size
draw.line(
[(x, 0), (x, height)],
#fill=(subtitle_color[0],subtitle_color[1],subtitle_color[2],0),
fill = bg_color1,
width=1
)
# 1. 绘制头部
if poster_style["header_style"]==2:
y = draw_header_2(img,draw,poster_data,width,title_color,subtitle_color)
elif poster_style["header_style"]==3:
y = draw_header_3(img,draw,poster_data,width,title_color,subtitle_color)
else:
y = draw_header_1(img,draw,poster_data,width,title_color,subtitle_color)
# 2. 绘制中间内容
if poster_style["style"] ==1: #多彩卡片
# 2. 绘制内容
y +=60
y = draw_content_card_1(img,draw,poster_data,width,y,section_colors,card_bg_color,font_color)
# 3.结语
y += 50
if poster_data["结语"]:
y = draw_submit_card(img,draw,poster_data,width,y,title_color,card_bg_color,subtitle_color)
elif poster_style["style"] ==2: #标题卡片
# 2.结语
y += 40
if poster_data["结语"]:
y = draw_submit_card(img,draw,poster_data,width,y,title_color,card_bg_color,subtitle_color)
# 3. 绘制内容
y +=50
y = draw_content_card_2(img,draw,poster_data,width,y,section_colors,card_bg_color,font_color)
elif poster_style["style"] ==3: #无卡片
# 2.结语
if poster_data["结语"]:
y += 40
y = draw_submit_card(img,draw,poster_data,width,y,title_color,card_bg_color,subtitle_color,"rect")
# 3. 绘制内容
y +=50
y = draw_content_3(img,draw,poster_data,width,y,section_colors,title_color,bg_color,font_color)
elif poster_style["style"] ==4: # 网格背景
# 2.结语
if poster_data["结语"]:
y += 40
y = draw_submit_card(img,draw,poster_data,width,y,title_color,card_bg_color,subtitle_color,"rect")
# 3. 绘制内容
y +=50
y = draw_content_4(img,draw,poster_data,width,y,section_colors,title_color,bg_color,font_color)
elif poster_style["style"] ==5: #彩色标签卡片
# 2.绘制内容
y += 40
y = draw_content_card_5(img,draw,poster_data,width,y,section_colors,card_bg_color,font_color)
# 3. 结语
y +=50
if poster_data["结语"]:
y = draw_submit_card(img,draw,poster_data,width,y,title_color,card_bg_color,subtitle_color)
elif poster_style["style"] ==6: #小圆角卡片
# 2. 结语
if poster_data["结语"]:
y += 40
y = draw_submit_card(img,draw,poster_data,width,y,title_color,card_bg_color,subtitle_color)
# 3. 绘制内容
y +=50
y = draw_content_card_6(img,draw,poster_data,width,y,section_colors,card_bg_color,font_color)
# 4. 底部装饰、版权信息
y +=60
if poster_data["报尾"]:
y = draw_footer_1(img,draw,poster_data,width,y,title_color,subtitle_color)
#裁切图片
if y < height:
new_img = Image.new('RGB', (width, y), bg_color)
new_img.paste(img, (0, 0))
return new_img
else:
return img
#入口函数，创建海报
def create_poster(content,path):
#百科内容转化为结构
poster_data = parse_html_to_poster(content)
from fun_calls import gen_poster_color
#生成主题颜色
try:
poster_style = gen_poster_color(poster_data)
except Exception as e:
print("主题生成出错",e)
poster_style = { "style": 6, "header_style": 3, "bg_color": "#F0F0F0", "title_color": "#003366", "subtitle_color": "#FFA500", "font_color": "#FFFFFF", "card_bg_color": "#D3D3D3", "section_colors": ["#FF0000", "#00FF00", "#0000FF", "#FFFF00", "#FF00FF"] }
poster_style["style"] = 6
poster_style["header_style"] = 3
# #测试使用
# poster_style={
# "style": 6,
# "header_style":3,
# "bg_color": "#FFFFFF",
# "title_color": "#333333",
# "subtitle_color": "#003366",
# "font_color": "#000000",
# "card_bg_color": "#F0F8FF",
# "section_colors": ["#FF6B6B", "#4ECDC4", "#45B7D1", "#96CEB4", "#FFEEAD"],
# # "bg_color": "#000080",
# # "title_color": "#FFFFFF",
# # "subtitle_color": "#CCCCCC",
# # "font_color": "#333333",
# # "card_bg_color": "#ADD8E6",
# # "section_colors": ["#FFFFFF", "#ADD8E6", "#87CEEB", "#B0C4DE", "#D3D3D3"]
# }
# 生成并保存海报
poster = create_modern_mobile_poster(poster_data,poster_style)
poster.save(path)
print('海报已生成: modern_poster.png')
#入口函数，创建海报
def create_poster_style(content,path,poster_style):
#百科内容转化为结构
poster_data = parse_html_to_poster(content)
# 生成并保存海报
poster = create_modern_mobile_poster(poster_data,poster_style)
poster.save(path)
print('海报已生成: modern_poster.png')
Reference in New Issue Copy Permalink