发布时间：2025-06-24 11:48:22 作者：北方职教升学中心 阅读量：247
聊天功能实现细节
用户可以在输入框中键入消息，通过“发送消息”按钮发送给百度大模型。
#ps:12.12修复bug，部分模型不能回应的问题，资源已更新！
在当前人工智能领域，通过与机器进行智能对话的能力变得越来越重要。
源码如下，可以根据注释和需求自行改进：
import datetimeimport threadingimport tkinter as tkfrom tkinter.scrolledtext import ScrolledTextfrom tkinter import ttkimport requestsimport jsonimport osCONFIG_FILE = "config.json"BACKGROUND_COLOR = "#D0E0F0"TEXT_COLOR = "#000000"BUTTON_COLOR = "#AEC6CF"ENTRY_BG = "#F0F8FF"FONT = ("Arial", 10)# 加载配置文件def load_config(): default_config = { "API_KEY": "", "SECRET_KEY": "", "temperature": 0.9, "top_p": 0.7, "penalty_score": 1.0, "system": "You are a helpful assistant", "selected_model": "ERNIE-Bot-turbo" } if os.path.exists(CONFIG_FILE): with open(CONFIG_FILE, "r") as file: config = json.load(file) default_config.update(config) return default_config else: return default_config# 保存所有配置def save_config(api_key, secret_key, selected_model, temperature, top_p, penalty_score, system): with open(CONFIG_FILE, "w") as file: config = { "API_KEY": api_key, "SECRET_KEY": secret_key, "selected_model": selected_model, "temperature": temperature, "top_p": top_p, "penalty_score": penalty_score, "system": system } json.dump(config, file)def get_access_token(api_key, secret_key): url = "https://aip.baidubce.com/oauth/2.0/token" params = {"grant_type": "client_credentials", "client_id": api_key, "client_secret": secret_key} return str(requests.post(url, params=params).json().get("access_token"))# 在 chat_with_bot 函数中添加参数，从配置文件读取相应的值def chat_with_bot(prompt, api_key, secret_key, model_url, temperature, top_p, penalty_score, system): access_token = get_access_token(api_key, secret_key) url = f"https://aip.baidubce.com{model_url}?access_token={access_token}" payload = json.dumps({ "messages": [{"role": "user", "content": prompt}], #部分模型必要输入参数，1.最后一个message的content长度（即此轮对话的问题）不能超过4800个字符，且不能超过2000 tokens.2.messages中content总长度大于4800个字符或2000 tokens，系统会依次遗忘最早的历史会话，直到content的总长度不超过4800个字符且不超过2000 tokens "stream": True, #流式输出 "temperature": temperature, #（1）较高的数值会使输出更加随机，而较低的数值会使其更加集中和确定（2）默认0.8，范围 (0, 1.0]，不能为0（3）建议该参数和top_p只设置1个（4）建议top_p和temperature不要同时更改 "top_p": top_p, #（1）影响输出文本的多样性，取值越大，生成文本的多样性越强（2）默认0.8，取值范围 [0, 1.0]（3）建议该参数和temperature只设置1个（4）建议top_p和temperature不要同时更改 "penalty_score": penalty_score, #通过对已生成的token增加惩罚，减少重复生成的现象。模型返回的结果通过JSON格式解析，最终在界面上显示。同时，添加了保存对话消息的功能，让用户随时保存他们感兴趣的对话记录。说明：（1）值越大表示惩罚越大（2）默认1.0，取值范围：[1.0, 2.0] "system": system, #模型人设，主要用于人设设定，例如，你是xxx公司制作的AI助手，长度限制1024个字符 "extra_parameters": { "use_keyword": True, "use_reference": True }, #Chatlaw必要参数第三方大模型推理高级参数，依据第三方大模型厂商不同而变化 "prompt": prompt, #部分模型必要输入参数，请求信息 }) headers = {'Content-Type': 'application/json'} response = requests.post(url, headers=headers, data=payload, stream=True) replies = [] for line in response.iter_lines(): if line: line_str = line.decode('utf-8') if line_str.startswith('data: '): try: json_str = line_str[6:] data = json.loads(json_str) if 'result' in data and data['result']: replies.append(data['result']) except json.JSONDecodeError as e: print("JSON decoding error:", e) return "\n".join(replies).replace("\n", "")# 传递参数def send_message(event=None): api_key = api_key_entry.get() secret_key = secret_key_entry.get() user_input = user_input_entry.get("1.0", "end-1c") selected_model = model_var.get() if user_input and api_key and secret_key and selected_model: # 从配置文件加载参数 config = load_config() temperature = config["temperature"] top_p = config["top_p"] penalty_score = config["penalty_score"] system = config["system"] # 创建一个新的线程来执行聊天请求，避免卡死 chat_thread = threading.Thread(target=perform_chat, args=(user_input, api_key, secret_key, model_urls[selected_model], temperature, top_p, penalty_score, system)) chat_thread.start() user_input_entry.delete("1.0", tk.END)# 执行聊天请求def perform_chat(user_input, api_key, secret_key, model_url, temperature, top_p, penalty_score, system): response = chat_with_bot(user_input, api_key, secret_key, model_url, temperature, top_p, penalty_score, system) update_chat_history("You: " + user_input + "\n" + "Bot: " + response + "\n\n")# 更新对话消息def update_chat_history(message): chat_history.config(state=tk.NORMAL) chat_history.insert(tk.END, message) chat_history.config(state=tk.DISABLED) chat_history.yview(tk.END)# 保存对话消息def save_chat_history_with_timestamp(): """ Save the chat history to a text file with a timestamp in the file name. """ chat_content = chat_history.get("1.0", tk.END).strip() if chat_content: timestamp = datetime.datetime.now().strftime("%Y-%m-%d_%H-%M-%S") filename = f"chat_history_{timestamp}.txt" with open(filename, "w") as file: file.write(chat_content)# 关闭脚本时保存配置并输出对话消息def on_closing(): api_key = api_key_entry.get() secret_key = secret_key_entry.get() selected_model = model_var.get() temperature = float(temperature_entry.get()) top_p = float(top_p_entry.get()) penalty_score = float(penalty_score_entry.get()) system = system_entry.get() save_config(api_key, secret_key, selected_model, temperature, top_p, penalty_score, system) chat_content = chat_history.get("1.0", tk.END) save_chat_history_with_timestamp() root.destroy()# 隐藏密码，保护隐私def toggle_show_hide(entry, show): entry.config(show="" if show.get() else "*")# 获取对应消息def get_last_message(): """ Get the last user message and bot response from the chat history. """ content = chat_history.get("1.0", tk.END).strip().split("\n") last_user_message = "" last_bot_response = "" for line in content: if line.startswith("You: "): last_user_message = line[len("You: "):] elif line.startswith("Bot: "): last_bot_response = line[len("Bot: "):] return last_user_message, last_bot_response# 复制消息def copy_to_clipboard(text): """ Copy a given text to the system clipboard. """ root.clipboard_clear() root.clipboard_append(text)# 复制用户消息def copy_user_message(): user_message, _ = get_last_message() copy_to_clipboard(user_message)#复制bot消息def copy_bot_response(): _, bot_response = get_last_message() copy_to_clipboard(bot_response)# 发送后清空输出def clear_input(): """ Clear the user input field. """ user_input_entry.delete("1.0", tk.END) chat_history.config(state=tk.NORMAL) chat_history.delete("1.0", tk.END) chat_history.config(state=tk.DISABLED)#保存配置def save_configurations(): """ Save the current API Key and Secret Key to the configuration file. """ api_key = api_key_entry.get() secret_key = secret_key_entry.get() selected_model = model_var.get() save_config(api_key, secret_key,selected_model)# 加载配置config = load_config()# 设置应用窗口root = tk.Tk()root.title("百度大模型(18个模型，其中5个模型官方限时免费，速来体验！！！)")root.geometry("650x700")root.configure(bg=BACKGROUND_COLOR)root.protocol("WM_DELETE_WINDOW", on_closing)# API Key输入框api_key_label = tk.Label(root, text="API Key", bg=BACKGROUND_COLOR, font=FONT)api_key_label.grid(row=0, column=0, padx=10, pady=(10, 0), sticky="w")api_key_entry = tk.Entry(root, width=40, font=FONT, bg=ENTRY_BG, show="*")api_key_entry.insert(0, config["API_KEY"])api_key_entry.grid(row=0, column=1, padx=10, pady=5, columnspan=2)# 隐藏API Keyshow_api_key = tk.BooleanVar(value=False)api_key_toggle = tk.Checkbutton(root, text="Show", variable=show_api_key, command=lambda: toggle_show_hide(api_key_entry, show_api_key), bg=BACKGROUND_COLOR, font=FONT)api_key_toggle.grid(row=0, column=2, sticky="w")# Secret Key输入框secret_key_label = tk.Label(root, text="Secret Key", bg=BACKGROUND_COLOR, font=FONT)secret_key_label.grid(row=1, column=0, padx=10, pady=(10, 0), sticky="w")secret_key_entry = tk.Entry(root, width=40, font=FONT, bg=ENTRY_BG, show="*")secret_key_entry.insert(0, config["SECRET_KEY"])secret_key_entry.grid(row=1, column=1, padx=10, pady=5, columnspan=2)# 隐藏Secret Keyshow_secret_key = tk.BooleanVar(value=False)secret_key_toggle = tk.Checkbutton(root, text="Show", variable=show_secret_key, command=lambda: toggle_show_hide(secret_key_entry, show_secret_key), bg=BACKGROUND_COLOR, font=FONT)secret_key_toggle.grid(row=1, column=2, sticky="w")# 模型网址model_urls = { "ERNIE-Bot-turbo": "/rpc/2.0/ai_custom/v1/wenxinworkshop/chat/eb-instant", "ERNIE-Bot": "/rpc/2.0/ai_custom/v1/wenxinworkshop/chat/completions", "ERNIE-Bot-4.0": "/rpc/2.0/ai_custom/v1/wenxinworkshop/chat/completions_pro", "ERNIE-Bot-8k": "/rpc/2.0/ai_custom/v1/wenxinworkshop/chat/ernie_bot_8k", "ERNIE-Bot-turbo-AI原生应用工作台": "/rpc/2.0/ai_custom/v1/wenxinworkshop/chat/ai_apaas", "BLOOMZ-7B": "/rpc/2.0/ai_custom/v1/wenxinworkshop/chat/bloomz_7b1", "Qianfan-BLOOMZ-7B-compressed": "/rpc/2.0/ai_custom/v1/wenxinworkshop/chat/qianfan_bloomz_7b_compressed", "Llama-2-7b-chat": "/rpc/2.0/ai_custom/v1/wenxinworkshop/chat/llama_2_7b", "Llama-2-13b-chat": "/rpc/2.0/ai_custom/v1/wenxinworkshop/chat/llama_2_13b", "Llama-2-70b-chat": "/rpc/2.0/ai_custom/v1/wenxinworkshop/chat/llama_2_70b", "Qianfan-Chinese-Llama-2-7B": "/rpc/2.0/ai_custom/v1/wenxinworkshop/chat/qianfan_chinese_llama_2_7b", "Qianfan-Chinese-Llama-2-13B(free)": "/rpc/2.0/ai_custom/v1/wenxinworkshop/chat/qianfan_chinese_llama_2_13b", "ChatGLM2-6B-32K": "/rpc/2.0/ai_custom/v1/wenxinworkshop/chat/chatglm2_6b_32k", "XuanYuan-70B-Chat-4bit(free)": "/rpc/2.0/ai_custom/v1/wenxinworkshop/chat/xuanyuan_70b_chat", "ChatLaw(free)": "/rpc/2.0/ai_custom/v1/wenxinworkshop/chat/chatlaw", "AquilaChat-7B": "/rpc/2.0/ai_custom/v1/wenxinworkshop/chat/aquilachat_7b", "SQLCoder-7B(free)": "/rpc/2.0/ai_custom/v1/wenxinworkshop/completions/sqlcoder_7b", "CodeLlama-7b-Instruct(free)": "/rpc/2.0/ai_custom/v1/wenxinworkshop/completions/codellama_7b_instruct",}# 设置模型及默认数值model_var = tk.StringVar(value=config["selected_model"])model_label = tk.Label(root, text="Selected model(选择模型)", bg=BACKGROUND_COLOR, font=FONT)model_label.grid(row=2, column=0, padx=10, pady=(10, 0), sticky="w")model_combobox = ttk.Combobox(root, textvariable=model_var, values=list(model_urls.keys()), state="readonly",width=29)model_combobox.grid(row=2, column=1, padx=10, pady=5,columnspan=1)# 温度输入框temperature_label = tk.Label(root, text="Temperature(温度)", bg=BACKGROUND_COLOR, font=FONT)temperature_label.grid(row=3, column=0, padx=10, pady=(10, 0), sticky="w")temperature_entry = tk.Entry(root, width=15, font=FONT, bg=ENTRY_BG)temperature_entry.insert(0, config["temperature"])temperature_entry.grid(row=3, column=1, padx=10, pady=5, columnspan=1)# 核采样输入框top_p_label = tk.Label(root, text="Top P(核采样)", bg=BACKGROUND_COLOR, font=FONT)top_p_label.grid(row=4, column=0, padx=10, pady=(10, 0), sticky="w")top_p_entry = tk.Entry(root, width=15, font=FONT, bg=ENTRY_BG)top_p_entry.insert(0, config["top_p"])top_p_entry.grid(row=4, column=1, padx=10, pady=5, columnspan=1)# 处罚成绩输入框penalty_score_label = tk.Label(root, text="Penalty Score(处罚成绩)", bg=BACKGROUND_COLOR, font=FONT)penalty_score_label.grid(row=5, column=0, padx=10, pady=(10, 0), sticky="w")penalty_score_entry = tk.Entry(root, width=15, font=FONT, bg=ENTRY_BG)penalty_score_entry.insert(0, config["penalty_score"])penalty_score_entry.grid(row=5, column=1, padx=10, pady=5, columnspan=1)# 系统角色输入框system_label = tk.Label(root, text="System role(系统角色)", bg=BACKGROUND_COLOR, font=FONT)system_label.grid(row=6, column=0, padx=10, pady=(10, 0), sticky="w")system_entry = tk.Entry(root, width=45, font=FONT, bg=ENTRY_BG)system_entry.insert(0, config["system"])system_entry.grid(row=6, column=1, padx=10, pady=5, columnspan=2)# 创建一个历史对话消息框chat_history = ScrolledText(root, state='disabled', width=85, height=15, bg=ENTRY_BG, font=FONT)chat_history.grid(row=8, column=0, columnspan=3, padx=10, pady=10,sticky="w")# 用户输入框user_input_entry = tk.Text(root, height=3, width=80, bd=3, font=("Arial", 10), bg="white")user_input_entry.grid(row=10, column=0, padx=10, pady=5, columnspan=2)user_input_entry.bind("<Return>", lambda event: send_message())# 发送按钮send_button = tk.Button(root, text="发送消息", command=send_message, bg=BUTTON_COLOR, font=FONT,width=33)send_button.grid(row=11, column=0, padx=10, pady=10,sticky="nsew",columnspan=1)# 清空按钮clear_button = tk.Button(root, text="清空输入", command=clear_input, bg=BUTTON_COLOR, font=FONT)clear_button.grid(row=11, column=1, padx=10, pady=5, sticky="nsew", columnspan=1)# 保存消息按钮save_with_timestamp_button = tk.Button(root, text="保存对话", command=save_chat_history_with_timestamp, bg=BUTTON_COLOR, font=FONT)save_with_timestamp_button.grid(row=12, column=0, padx=10, pady=10, sticky="nsew", columnspan=1)# 保存配置按钮save_config_button = tk.Button(root, text="保存配置", command=save_configurations, bg=BUTTON_COLOR, font=FONT)save_config_button.grid(row=12, column=1, padx=10, pady=10, sticky="nsew", columnspan=1)# 复制用户消息按钮copy_user_message_button = tk.Button(root, text="复制用户消息", command=copy_user_message, bg=BUTTON_COLOR, font=FONT)copy_user_message_button.grid(row=13, column=0, padx=5, pady=5,sticky="nsew")# 复制bot信息按钮copy_bot_response_button = tk.Button(root, text="复制bot消息", command=copy_bot_response, bg=BUTTON_COLOR, font=FONT)copy_bot_response_button.grid(row=13, column=1, padx=5, pady=5,sticky="nsew")# 全局定义用户/bot消息last_user_message = ""last_bot_response = ""root.mainloop()本文展示了如何使用Python和Tkinter库构建一个简单但功能强大的交互式界面，使用户能够与百度大模型进行智能对话，并提供了保存对话消息的功能。参数设置和聊天历史记录显示和清除等功能。
效果如图：
加载和保存配置文件功能
利用config.json文件，实现了用户配置参数的读取和保存。
与百度API通信过程
通过百度的API，代码可以获取访问令牌，并将用户输入的消息发送给百度大模型。
用户界面构建
使用Tkinter库构建了一个简单但功能完善的用户界面，其中包括输入API密钥的文本框、这确保了用户下次启动应用时能够继续使用先前的参数，如图：
如上软件集成了百度大模型的18个模型，其中有5个模型正在限时免费使用，快去注册登录获取属于自己的API密钥开启对话吧！！！
步骤如下：百度智能云千帆大模型平台
软件资源：
已更新！百度大模型软件大小9.3MB，百度网盘下载地址如下，永久有效：
百度大模型
https://pan.baidu.com/share/init?surl=Ojs0W00E0L5pteGqPPpecw&pwd=6666
代码功能概述：
代码实现了以下主要功能：
- 加载和保存配置文件： 提供了一种方法，使用户可以保存他们的API密钥和其他参数，以便于下次使用。模型返回的消息将会显示在界面上，并且整个对话历史都会保留在界面上。模型选择的下拉菜单、本文将介绍如何使用Python编程语言和Tkinter库构建一个交互式界面，使用户能够与百度大模型进行智能对话，展示了人机交互的潜力与未来发展。参数设置的输入框以及聊天历史记录的显示。
- 实现聊天功能： 用户可以在界面上输入文本消息并发送给模型，模型将生成回复并显示在界面上。通过对代码的解析和功能概述，读者可以更好地了解这个项目的意义和实现细节，并在人机交互领域中探索更多可能性。
感谢各位朋友们阅读，下期再见！！！
