# -*- coding: utf-8 -*-
# from utils.prompt import prompt
# 如果你想直接在命令行中输出assistant_response而不使用PySimpleGUI库，可以简化代码，直接在控制台中打印assistant_response。

from openai import OpenAI

client = OpenAI(api_key="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")

conversation_history = [
    {
        "role": "system",
        "content": prompt
    }
]

user_input = "开始游戏"  # 默认用户输入为"开始游戏"

while user_input.lower() != "stop":
    # 将用户输入添加到对话历史中
    conversation_history.append({"role": "user", "content": user_input})

    # 使用GPT-3.5-turbo模型生成回复
    response = client.chat.completions.create(
        model="gpt-3.5-turbo",  # 指定要使用的模型
        messages=conversation_history,  # 包含对话历史的消息列表
        temperature=0.1,  # 控制生成文本的创造性，值越高则更随机
        max_tokens=4096,  # 生成文本的最大长度限制
        top_p=0.1  # 用于控制采样概率分布的参数，值越低则更保守
    )

    # 获取助手的回复并将其添加到对话历史中
    assistant_response = response.choices[0].message.content
    conversation_history.append({"role": "assistant", "content": assistant_response})

    # 打印助手的回复
    print("Assistant:", assistant_response)

    # 获取用户新的输入
    user_input = input("你的选择是:")
print('游戏结束')