GPT室内提示词生成

from openai import OpenAI

client = OpenAI(
    base_url="https://oneapi.xty.app/v1",
    api_key="sk-gfJsNLc7n7AWUmJG4c2030D10d1d479197400c6c53D58d79"
)

# 定义系统指令（instruction）
instruction = (
    "你是stable diffusion的提示词生成器，专注于室内设计领域(家庭装修)（如客厅，卧室等）。我会给你一个引导词，如“禅意”，“北欧风格等”，请根据这个引导词创造10条英文的stable diffusion提示语，用于创造惊艳的stable diffsuion图像，提示语尽量长，提示语前不要有数字，每条提示语用换行符隔开，提示语尽量用完整的自然语言句子。请只要返还10条提示语，不要有其他的回答，除逗号句号外不要有特殊符号。一行一句，行与行之间不要有空行。提示语前面都加上“Interior design,”"
)

# 初始系统消息
messages = [
    {"role": "system", "content": instruction}
]

# 打印使用说明
print("欢迎使用室内设计领域的stable diffusion提示词生成器！")
print("请输入一个引导词（如'禅意', '北欧风格'），系统将根据您的输入生成用于创造室内设计图像的英文提示词。")
print("如果您想退出程序，请输入 'quit'。\n")

# 循环直到用户输入 'quit'
while True:
    # 获取用户输入
    user_input = input("请输入您的引导词: ")

    # 如果用户输入 'quit', 结束对话
    if user_input.lower() == 'quit':
        print("正在退出程序...")
        break

    # 将用户的消息添加到消息列表
    messages.append({"role": "user", "content": user_input})

    # 获取模型的回答
    completion = client.chat.completions.create(
        model="gpt-4-1106-preview",
        messages=messages
    )

    # 提取 GPT 的回答并打印
    gpt_response = completion.choices[0].message.content
    print(gpt_response)

    # 将 GPT 的回答也添加到消息列表，以便用于下一轮对话
    messages.append({"role": "assistant", "content": gpt_response})