为该代码写一个图形界面

好的，为了给ChitGPT添加一个图形界面，我们可以使用Python的GUI库，例如Tkinter或PyQt5。下面是一个使用Tkinter的示例代码：

import tkinter as tk
from tkinter import scrolledtext
from threading import Thread
from ChitGPT import ChitGPT  # 导入我们的ChitGPT模型

class GUI:
    def __init__(self):
        self.model = ChitGPT()  # 初始化ChitGPT模型
        self.window = tk.Tk()  # 创建一个Tkinter窗口
        self.window.title("ChitGPT Chatbot")  # 设置窗口标题

        # 创建一个滚动文本框，用于显示对话记录
        self.chat_history = scrolledtext.ScrolledText(self.window, width=60, height=20)
        self.chat_history.grid(column=0, row=0, padx=10, pady=10)

        # 创建一个输入框和发送按钮，用于发送对话信息
        self.input_field = tk.Entry(self.window, width=40)
        self.input_field.grid(column=0, row=1, padx=10, pady=10)
        self.send_button = tk.Button(self.window, text="Send", command=self.send_message)
        self.send_button.grid(column=1, row=1, padx=10, pady=10)

        self.window.mainloop()  # 进入Tkinter主循环

    def send_message(self):
        message = self.input_field.get()  # 获取用户输入的信息
        self.input_field.delete(0, tk.END)  # 清空输入框

        # 将用户输入的信息添加到对话记录中
        self.chat_history.insert(tk.END, "You: " + message + "\n")

        # 创建一个线程，用于调用ChitGPT模型生成回复信息
        thread = Thread(target=self.generate_response, args=(message,))
        thread.start()

    def generate_response(self, message):
        response = self.model.generate(message)  # 调用ChitGPT模型生成回复信息

        # 将回复信息添加到对话记录中
        self.chat_history.insert(tk.END, "ChitGPT: " + response + "\n")

if __name__ == "__main__":
    GUI()

运行该代码后，将会弹出一个包含聊天记录和输入框的窗口，用户可以在输入框中输入信息并发送，机器人将会根据用户输入生成回复信息并显示在聊天记录中。