聊天机器人羲和的代码03

为了使聊天机器人羲和GUI更加美观和人性化，进行了以下几个方面的改进：

***颜色和主题：*选择更柔和、和谐的颜色搭配，使用现代的UI设计元素。
*布局优化：*合理安排控件的位置和大小，确保界面整洁有序。
*交互体验：*增加用户反馈（如加载动画、提示信息等），提高用户体验。
功能增强：添加更多实用功能，如历史记录保存、语音输入等。
以下是一些具体的改进建议和代码实现：

1. 颜色和主题
使用更柔和的颜色和现代的主题风格。可以使用Tkinter的ttk模块来创建更现代的控件。

import ttkbootstrap as ttk

# 主题选择
style = ttk.Style(theme='litera')

# 创建主窗口
root = ttk.Window(themename='litera')
root.title("羲和聊天机器人")
root.geometry("600x400")

2. 布局优化
使用ttk.Frame来组织控件，使布局更加清晰。

# 创建框架
frame = ttk.Frame(root, padding=20)
frame.pack(fill=tk.BOTH, expand=True)

# 问题标签和输入框
question_label = ttk.Label(frame, text="问题:", font=("Arial", 14))
question_label.grid(row=0, column=0, pady=10, sticky=tk.W)

question_entry = ttk.Entry(frame, width=50, font=("Arial", 12))
question_entry.grid(row=0, column=1, pady=10, sticky=tk.W)

# 获取回答按钮
answer_button = ttk.Button(frame, text="获取回答", command=self.get_answer, style='success.TButton')
answer_button.grid(row=1, column=0, pady=10, columnspan=2, sticky=tk.W+tk.E)

# 清除历史记录按钮
clear_button = ttk.Button(frame, text="清除历史记录", command=self.clear_history, style='danger.TButton')
clear_button.grid(row=2, column=0, pady=5, columnspan=2, sticky=tk.W+tk.E)

# 回答标签和文本显示区域
answer_label = ttk.Label(frame, text="回答:", font=("Arial", 14))
answer_label.grid(row=3, column=0, pady=10, sticky=tk.W)

answer_text = tk.Text(frame, height=10, width=50, font=("Arial", 12))
answer_text.grid(row=4, column=0, pady=5, columnspan=2, sticky=tk.W+tk.E)

3. 交互体验
增加加载动画和提示信息，提高用户体验。

def get_answer(self):
    question = self.question_entry.get()
    if not question:
        messagebox.showwarning("输入错误", "请输入问题")
        return

    self.answer_text.insert(tk.END, "正在获取回答...\n")
    self.answer_text.update_idletasks()

    inputs = self.tokenizer(question, return_tensors='pt', padding='max_length', truncation=True, max_length=128)
    with torch.no_grad():
        input_ids = inputs['input_ids'].to(self.device)
        attention_mask = inputs['attention_mask'].to(self.device)
        logits = self.model(input_ids, attention_mask)
    
    if logits.item() > 0:
        answer_type = "羲和回答"
    else:
        answer_type = "零回答"

    specific_answer = self.get_specific_answer(question, answer_type)

    self.answer_text.insert(tk.END, f"问题: {
     question}\n{
     answer_type}\n{
     specific_answer}\n\n")

4. 功能增强
添加历史记录保存和语音输入功能。
历史记录保存

def save_history(self):
    history = self.answer_text.get(1.0, tk.END)
    if history.strip():
        timestamp = datetime.now().strftime('%Y-%m-%d_%H-%M-%S')
        with open(f'history_{
     timestamp}.txt