#.env內(nèi)容為API_KEY=sk-寫(xiě)自己的

import openai

from dotenv import dotenv_values

class ChatBox:

? ? def __init__(self):

? ? ? ? config = dotenv_values(".env")

? ? ? ? openai.api_key = config["API_KEY"]

? ? ? ? self.model = "text-davinci-003"

? ? ? ? self.chat_history = []

? ? def add_chat(self, message):

? ? ? ? response = openai.Completion.create(

? ? ? ? ? ? model=self.model,

? ? ? ? ? ? prompt=message,

? ? ? ? ? ? max_tokens=200

? ? ? ? )

? ? ? ? answer = response.choices[0].text.strip()

? ? ? ? self.chat_history.append((message, answer))

? ? ? ? return answer

? ? def get_chat_history(self):

? ? ? ? return self.chat_history

? ? def clear_chat_history(self):

? ? ? ? self.chat_history = []

chatbox = ChatBox()

while True:

? ? user_input = input("User: ")

? ? if user_input.lower() == "exit":

? ? ? ? break

? ? answer = chatbox.add_chat(user_input)

? ? print("ChatGPT:", answer)