把这个脚本加上tkinter库
时间: 2023-11-15 11:48:59 浏览: 25
好的,以下是加上 tkinter 库的脚本示例:
```python
import openai
import tkinter as tk
# 设置 OpenAI API 密钥
openai.api_key = "YOUR_API_KEY"
# 创建与 OpenAI 的连接
def create_openai_session():
return openai.CompletionClient()
# 创建一个类 ChatGPT 的 AI
class ChatGPT:
def __init__(self):
self.session = create_openai_session()
# 定义一个方法用于获取 AI 的回复
def generate_text(self, prompt):
response = self.session.generate(
engine="text-davinci-002",
prompt=prompt,
temperature=0.5,
max_tokens=1024,
n_context=1
)
message = response.choices[0].text.strip()
return message
# 创建一个 Tkinter 窗口
root = tk.Tk()
# 设置窗口标题
root.title("ChitGPT Chatbot")
# 创建聊天记录框
chat_log = tk.Text(root, height=20, width=50)
chat_log.pack()
# 创建输入框和发送按钮
input_frame = tk.Frame(root)
input_frame.pack(side=tk.BOTTOM, fill=tk.X)
input_box = tk.Entry(input_frame, width=40)
input_box.pack(side=tk.LEFT, padx=5)
submit_button = tk.Button(input_frame, text="Send", command=lambda: send_message())
submit_button.pack(side=tk.RIGHT, padx=5)
# 创建 ChatGPT AI 实例
chatbot = ChatGPT()
# 定义发送消息的方法
def send_message():
message = input_box.get()
input_box.delete(0, tk.END)
chat_log.insert(tk.END, "You: " + message + "\n")
response = chatbot.generate_text(message)
chat_log.insert(tk.END, "ChitGPT: " + response + "\n")
# 运行窗口
root.mainloop()
```
注意,如果你要使用此脚本,需要将 YOUR_API_KEY 替换为你的 OpenAI API 密钥。