diff --git a/main/fun_calls.py b/main/fun_calls.py new file mode 100644 index 0000000..712585f --- /dev/null +++ b/main/fun_calls.py @@ -0,0 +1,44 @@ +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) +