带文字海报流程自动化

上一篇文章：

带文字海报流程自动化 - 知乎

项目代码整理在：

https://github.com/liangwq/Chatglm_lora_multi-gpu​github.com/liangwq/Chatglm_lora_multi-gpu

根据用户的输入生成图片prompt模块代码封装：

from openai import OpenAI 

import json
def json_parse(image_prompt):
    image_prompt = image_prompt.replace("```\n","")
    image_prompt = image_prompt.replace("\n```","")
    # 您提供的字符串
    data_str =image_prompt #completion.choices[0].message.content


    # 将字符串转换为JSON对象
    try:
        data_json = json.loads(data_str)
        return data_json
    except json.JSONDecodeError as e:
        print("字符串不是有效的JSON格式：", e)
        
def gen_text_to_image_prompt(user_input): 
    client = OpenAI(api_key="你的阶跃APIkey", base_url="https://api.stepfun.com/v1")

    completion = client.chat.completions.create(
      model="step-1-200k",
      messages=[
          {
              "role": "system",
              "content": "你是由阶跃星辰提供的带文字海报生成助手，你擅长中文，英文，以及多种其他语言的对话。在保证用户数据安全的前提下，你能对用户的问题和请求，作出快速和精准的回答。同时，你的回答和建议应该拒绝黄赌毒，暴力恐怖主义的内容",
          },
          {"role": "user", "content": "你是带文本海报生成智能助理. \
                                       1.根据用户输入内容生成做图需要的描述prompt \