Add File

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)
+