You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
120 lines
3.5 KiB
120 lines
3.5 KiB
import http.client
|
|
import json
|
|
import tkinter as tk
|
|
import time
|
|
import chat2
|
|
|
|
# 创建HTTPS连接和请求头
|
|
conn = http.client.HTTPSConnection("api.chatanywhere.com.cn")
|
|
headers = {
|
|
'Authorization': 'Bearer sk-T2uMUUuvXHFTpVo0M96rOMkLCygv6BODpwBozgXjmzU9c2yi',
|
|
'Content-Type': 'application/json'
|
|
}
|
|
|
|
# 初始化暂停标志
|
|
paused = False
|
|
|
|
# 对话历史列表
|
|
dialogue_history = []
|
|
|
|
# 发送消息函数
|
|
def send_message(event=None):
|
|
global paused
|
|
if paused:
|
|
return
|
|
|
|
# 获取用户输入的消息
|
|
user_message = user_input.get()
|
|
user_input.delete(0, tk.END)
|
|
|
|
# 构建请求payload
|
|
payload = json.dumps({
|
|
"model": "gpt-3.5-turbo",
|
|
"messages": [
|
|
{
|
|
"role": "user",
|
|
"content": user_message
|
|
}
|
|
]
|
|
})
|
|
|
|
# 发送POST请求获取助手回复
|
|
conn.request("POST", "/v1/chat/completions", payload, headers)
|
|
res = conn.getresponse()
|
|
data = res.read()
|
|
|
|
# 解析助手回复数据
|
|
response_data = json.loads(data.decode("utf-8"))
|
|
assistant_response = response_data["choices"][0]["message"]["content"]
|
|
|
|
# 将助手回复按行拆分
|
|
assistant_lines = assistant_response.split("\n")
|
|
|
|
# 保存对话时间标识
|
|
current_time = time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())
|
|
|
|
# 更新对话历史列表
|
|
dialogue_history.append((current_time, "我", user_message))
|
|
dialogue_history.extend([(current_time, "下等外国马", line) for line in assistant_lines])
|
|
|
|
# 保存对话到指定的txt文件
|
|
with open("dialogue.txt", "a") as file:
|
|
for time_stamp, role, message in dialogue_history:
|
|
file.write(f"[{time_stamp}] {role}: {message}\n")
|
|
|
|
# 更新对话历史文本框并显示助手回复
|
|
message_history.config(state=tk.NORMAL)
|
|
message_history.insert(tk.END, "我: " + user_message + "\n", "user_message")
|
|
for line in assistant_lines:
|
|
if paused:
|
|
break
|
|
message_history.insert(tk.END, "下等外国马: " + line + "\n", "assistant_message")
|
|
root.update()
|
|
time.sleep(1)
|
|
message_history.config(state=tk.DISABLED)
|
|
message_history.yview(tk.END)
|
|
|
|
# 控制暂停和继续回答的函数
|
|
def pause_response():
|
|
global paused
|
|
paused = not paused
|
|
|
|
# 打开新的tk界面
|
|
def open_new_window():
|
|
chat2.run()
|
|
|
|
# 创建主窗口
|
|
root = tk.Tk()
|
|
root.title("Chat_fk")
|
|
root.geometry("660x480")
|
|
|
|
# 创建对话历史文本框
|
|
message_history = tk.Text(root, width=60, height=20, wrap="word")
|
|
message_history.pack(pady=10)
|
|
message_history.tag_config("user_message", justify="right", foreground="blue")
|
|
message_history.tag_config("assistant_message", justify="left", foreground="green")
|
|
message_history.config(state=tk.DISABLED)
|
|
|
|
# 创建用户输入框和发送按钮
|
|
input_frame = tk.Frame(root)
|
|
input_frame.pack()
|
|
|
|
user_input = tk.Entry(input_frame, width=50)
|
|
user_input.pack(side=tk.LEFT, padx=5)
|
|
|
|
send_button = tk.Button(input_frame, text="发送", command=send_message)
|
|
send_button.pack(side=tk.LEFT, padx=5)
|
|
|
|
# 创建暂停按钮
|
|
pause_button = tk.Button(input_frame, text="暂停", command=pause_response)
|
|
pause_button.pack(side=tk.LEFT, padx=5)
|
|
|
|
# 创建打开新界面的按钮
|
|
open_button = tk.Button(input_frame, text="打开历史对话", command=open_new_window)
|
|
open_button.pack(side=tk.LEFT, padx=5)
|
|
|
|
|
|
|
|
root.bind('<Return>', send_message)
|
|
|
|
root.mainloop() |