45 lines
1.4 KiB
Python
45 lines
1.4 KiB
Python
|
import datetime
|
|||
|
|
from init import llm
|
|||
|
|
import json
|
|||
|
|
import json5
|
|||
|
|
from utils import found_llm_think_data
|
|||
|
|
|
|||
|
|
|
|||
|
|
#根据海报内容生成主题颜色
|
|||
|
|
def gen_poster_color(data):
|
|||
|
|
# 获取当前日期和星期
|
|||
|
|
today = datetime.datetime.now()
|
|||
|
|
# 获取今天是星期几，使用weekday()方法（周一=0, 周日=6）
|
|||
|
|
weekday_number = today.weekday()
|
|||
|
|
|
|||
|
|
system_prompt=f"""
|
|||
|
|
# 你是一个海报设计助手
|
|||
|
|
今天是{today.strftime('%Y-%m-%d')} 星期{weekday_number+1}
|
|||
|
|
|
|||
|
|
## 工作技能
|
|||
|
|
1. 你可以根据文案的主标题和副标题来设计主题颜色，包括背景色，主标题，副标题，内容卡片的背景色，内容卡片的标题颜色组section_colors不少于5个，内容字体颜色等，
|
|||
|
|
2. 最终输出格式为JSON,具体如下：
|
|||
|
|
|
|||
|
|
{{"bg_color":"背景颜色值","title_color":"主标题颜色值","subtitle_color":"副标题颜色值",
|
|||
|
|
"font_color":"内容字体颜色值","card_bg_color":"卡片背景色","section_colors":["color1","color2",...]}}
|
|||
|
|
|
|||
|
|
海报主标题：
|
|||
|
|
{data['主标']}
|
|||
|
|
海报副标题：
|
|||
|
|
{data['副标']}
|
|||
|
|
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
messages = [{'role': 'user', 'content': system_prompt}]
|
|||
|
|
response = llm.chat(
|
|||
|
|
messages=messages,
|
|||
|
|
stream=False,
|
|||
|
|
) # get a new response from the model where it can see the function response
|
|||
|
|
|
|||
|
|
result = response[0]["content"]
|
|||
|
|
|
|||
|
|
think,data = found_llm_think_data(result)
|
|||
|
|
|
|||
|
|
return json5.loads(data)
|
|||
|
|
|