v1.0.7.py

import tkinter as tk
from tkinter import scrolledtext
from tkinter import messagebox
from PIL import Image, ImageTk
from openai import OpenAI
import json
import threading

class ChatApp:
    def __init__(self, root):
        self.root = root
        self.root.title("Chat Application")
        self.root.geometry("800x600")

        # 配置根窗口的网格权重
        self.root.grid_columnconfigure(0, weight=1)
        self.root.grid_rowconfigure(0, weight=1)

        # 创建并配置消息框架
        self.message_frame = tk.Frame(self.root)
        self.message_frame.grid(column=0, row=0, padx=10, pady=10, sticky="nsew")
        self.message_frame.grid_columnconfigure(0, weight=1)
        self.message_frame.grid_rowconfigure(0, weight=1)

        # 创建并配置画布和滚动条
        self.canvas = tk.Canvas(self.message_frame)
        self.scrollbar = tk.Scrollbar(self.message_frame, orient="vertical", command=self.canvas.yview)
        self.scrollable_frame = tk.Frame(self.canvas)

        self.scrollable_frame.bind(
            "<Configure>",
            lambda e: self.canvas.configure(
                scrollregion=self.canvas.bbox("all")
            )
        )

        self.canvas.create_window((0, 0), window=self.scrollable_frame, anchor="nw")
        self.canvas.configure(yscrollcommand=self.scrollbar.set)

        self.canvas.pack(side="left", fill="both", expand=True)
        self.scrollbar.pack(side="right", fill="y")

        # 绑定鼠标滚轮事件
        self.canvas.bind_all("<MouseWheel>", self._on_mousewheel)

        # 输入框和发送按钮框架
        input_frame = tk.Frame(self.root)
        input_frame.grid(column=0, row=1, padx=10, pady=5, sticky="ew")
        input_frame.grid_columnconfigure(0, weight=1)

        # 输入框
        self.entry = tk.Entry(input_frame, width=50)
        self.entry.grid(column=0, row=0, padx=(0, 5), pady=5, sticky="ew")

        # 发送按钮
        self.send_button = tk.Button(input_frame, text="发送", command=self.send_message)
        self.send_button.grid(column=1, row=0, padx=5, pady=5, sticky="e")

        # 控制按钮框架
        control_frame = tk.Frame(self.root)
        control_frame.grid(column=0, row=2, padx=10, pady=5, sticky="ew")

        # 新对话按钮
        self.new_chat_button = tk.Button(control_frame, text="新对话", command=self.new_chat)
        self.new_chat_button.grid(column=0, row=0, padx=(0, 5), pady=5, sticky="w")

        # Debug按钮
        self.debug_button = tk.Button(control_frame, text="Debug", command=self.show_debug_info)
        self.debug_button.grid(column=1, row=0, padx=5, pady=5, sticky="w")

        # 流式输出复选框
        self.stream_var = tk.BooleanVar(value=True)
        self.stream_checkbox = tk.Checkbutton(control_frame, text="流式输出", variable=self.stream_var)
        self.stream_checkbox.grid(column=2, row=0, padx=5, pady=5, sticky="w")

        # 模式选择下拉菜单
        self.context_mode_var = tk.StringVar(value="上下文连续对话")
        self.context_mode_menu = tk.OptionMenu(control_frame, self.context_mode_var, "独立对话", "上下文连续对话")
        self.context_mode_menu.grid(column=3, row=0, padx=5, pady=5, sticky="w")

        # 加载用户和助手头像
        self.user_img = self.load_and_resize_image("images/user_icon.png", (25, 25))
        self.assistant_img = self.load_and_resize_image("images/chatgpt_icon.png", (25, 25))

        # 初始化OpenAI API
        self.client = OpenAI(
            api_key="your api-key",
            base_url="https://Proxy address/v1/"
        )

        self.debug_info = ""
        self.messages = [{"role": "system", "content": "You are a helpful assistant."}]

        # 绑定窗口大小变化事件以调整wraplength
        self.root.bind('<Configure>', self.update_wraplengths)

    def load_and_resize_image(self, path, size):
        img = Image.open(path)
        img = img.resize(size, Image.LANCZOS)
        return ImageTk.PhotoImage(img)

    def new_chat(self):
        for widget in self.scrollable_frame.winfo_children():
            widget.destroy()
        self.messages = [{"role": "system", "content": "You are a helpful assistant."}]
        # 重新配置画布的滚动区域为初始状态
        self.canvas.configure(scrollregion=(0, 0, 0, 0))

    def send_message(self):
        user_message = self.entry.get()
        if user_message:
            self.add_message_to_frame("user", user_message)
            self.entry.delete(0, tk.END)

            if self.context_mode_var.get() == "独立对话":
                self.messages = [{"role": "system", "content": "You are a helpful assistant."}]

            self.messages.append({"role": "user", "content": user_message})

            if self.stream_var.get():
                threading.Thread(target=self.get_ai_reply_stream, daemon=True).start()
            else:
                threading.Thread(target=self.get_ai_reply, daemon=True).start()

    def add_message_to_frame(self, role, message):
        frame = tk.Frame(self.scrollable_frame)
        frame.pack(fill=tk.X, padx=5, pady=5)
        frame.grid_columnconfigure(1, weight=1)

        if role == "user":
            img_label = tk.Label(frame, image=self.user_img)
        else:
            img_label = tk.Label(frame, image=self.assistant_img)
        img_label.grid(row=0, column=0, sticky="nw", padx=(0, 5))

        # 动态设置wraplength
        wraplength = self.get_wraplength()

        text_widget = tk.Label(frame, text=message, wraplength=wraplength, justify="left", anchor="w")
        text_widget.grid(row=0, column=1, sticky="nsew")

        button_frame = tk.Frame(frame)
        button_frame.grid(row=1, column=1, sticky="nw", pady=(5, 0))

        copy_button = tk.Label(button_frame, text="复制", fg="gray", cursor="hand2")
        copy_button.pack(side="left", padx=(0, 5))
        copy_button.bind("<Button-1>", lambda e: self.copy_to_clipboard(message, copy_button))

        edit_button = tk.Label(button_frame, text="编辑", fg="gray", cursor="hand2")
        edit_button.pack(side="left")
        edit_button.bind("<Button-1>", lambda e: self.toggle_edit_mode(frame, text_widget, edit_button, role, message))

        self.adjust_scroll_region()

    def copy_to_clipboard(self, text, copy_button):
        self.root.clipboard_clear()
        self.root.clipboard_append(text)
        self.root.update()  # 这将更新剪贴板
        # 更改按钮文字为"✔"
        copy_button.config(text="✔")
        # 1秒后恢复按钮文字为"复制"
        self.root.after(1000, lambda: copy_button.config(text="复制"))

    def toggle_edit_mode(self, frame, widget, edit_button, role, original_message):
        if edit_button["text"] == "编辑":
            # 切换到编辑模式
            text_widget = tk.Text(frame, wrap=tk.WORD)

            # 使用之前定义的 get_text_width 方法计算字符宽度
            a = self.get_text_width()
            text_widget.config(width=a)

            text_widget.insert(tk.END, widget["text"])
            text_widget.grid(row=0, column=1, sticky="nsew")
            widget.grid_remove()
            edit_button.configure(text="完成")
            # 保存对 text_widget 的引用
            frame.text_widget = text_widget

            # 保存原始消息内容
            frame.original_message = original_message

            # 自动调整Text高度以适应内容
            max_height = 20  # 设置最大行数为20行，可以根据需要调整
            extra_lines = 10
            num_lines = int(text_widget.index('end-1c').split('.')[0]) + extra_lines
            text_widget.configure(height=num_lines)

            # 创建取消按钮，并添加到edit_button后面
            cancel_button = tk.Label(frame, text="取消", fg="gray", cursor="hand2")
            cancel_button.grid(row=0, column=2, padx=(5, 0), sticky="w")
            frame.cancel_button = cancel_button

            # 绑定取消按钮的事件
            cancel_button.bind("<Button-1>", lambda e: self.cancel_edit_mode(frame, widget, edit_button))

        else:
            # 保存编辑并切换回显示模式
            new_text = frame.text_widget.get("1.0", tk.END).strip()
            widget.configure(text=new_text)
            frame.text_widget.grid_remove()
            widget.grid()
            edit_button.configure(text="编辑")
            frame.cancel_button.grid_remove()  # 移除取消按钮
            # 删除 text_widget 和 cancel_button 的引用
            del frame.text_widget
            del frame.cancel_button

            # 更新消息历史，替换原始消息内容
            for msg in self.messages:
                if msg["role"] == role and msg["content"] == original_message:
                    msg["content"] = new_text
                    break

        self.adjust_scroll_region()

    def cancel_edit_mode(self, frame, widget, edit_button):
        # 恢复到编辑前的内容
        widget.configure(text=frame.original_message)
        frame.text_widget.grid_remove()
        widget.grid()
        edit_button.configure(text="编辑")
        frame.cancel_button.grid_remove()  # 移除取消按钮
        # 删除 text_widget 和 cancel_button 的引用
        del frame.text_widget
        del frame.cancel_button
        self.adjust_scroll_region()

    def adjust_scroll_region(self):
        self.scrollable_frame.update_idletasks()
        bbox = self.canvas.bbox("all")

        # 获取画布的高度
        canvas_height = self.canvas.winfo_height()

        # 如果内容高度小于画布高度，将内容固定在顶部
        if (bbox[3] - bbox[1]) < canvas_height:
            self.canvas.configure(scrollregion=(bbox[0], bbox[1], bbox[2], canvas_height))
            self.canvas.yview_moveto(0)
        else:
            self.canvas.configure(scrollregion=bbox)
            self.canvas.yview_moveto(1)

    def get_ai_reply_stream(self):
        response = self.client.chat.completions.create(
            model="gpt-4o-mini",
            messages=self.messages,
            stream=True
        )

        debug_info_list = []
        ai_reply = ""
        temp_frame = tk.Frame(self.scrollable_frame)
        temp_frame.pack(fill=tk.X, padx=5, pady=5)
        temp_frame.grid_columnconfigure(1, weight=1)

        img_label = tk.Label(temp_frame, image=self.assistant_img)
        img_label.grid(row=0, column=0, sticky="nw", padx=(0, 5))

        wraplength = self.get_wraplength()

        text_widget = tk.Label(temp_frame, wraplength=wraplength, justify="left", anchor="w")
        text_widget.grid(row=0, column=1, sticky="nsew")

        button_frame = tk.Frame(temp_frame)
        button_frame.grid(row=1, column=1, sticky="nw", pady=(5, 0))

        copy_button = tk.Label(button_frame, text="复制", fg="gray", cursor="hand2")
        copy_button.pack(side="left", padx=(0, 5))
        copy_button.bind("<Button-1>", lambda e: self.copy_to_clipboard(ai_reply, copy_button))

        edit_button = tk.Label(button_frame, text="编辑", fg="gray", cursor="hand2")
        edit_button.pack(side="left")

        for chunk in response:
            debug_info_list.append(chunk)
            content = chunk.choices[0].delta.content
            if content:
                ai_reply += content
                text_widget.config(text=ai_reply)

            self.adjust_scroll_region()
            self.root.update()

        edit_button.bind("<Button-1>",
                         lambda e: self.toggle_edit_mode(temp_frame, text_widget, edit_button, "assistant", ai_reply))

        self.debug_info = json.dumps([chunk.to_dict() for chunk in debug_info_list], indent=4, ensure_ascii=False)
        self.messages.append({"role": "assistant", "content": ai_reply})

    def get_ai_reply(self):
        response = self.client.chat.completions.create(
            model="gpt-4o-mini",
            messages=self.messages
        )

        ai_reply = response.choices[0].message.content

        # 创建新的框架来显示AI回复
        temp_frame = tk.Frame(self.scrollable_frame)
        temp_frame.pack(fill=tk.X, padx=5, pady=5)
        temp_frame.grid_columnconfigure(1, weight=1)

        img_label = tk.Label(temp_frame, image=self.assistant_img)
        img_label.grid(row=0, column=0, sticky="nw", padx=(0, 5))

        wraplength = self.get_wraplength()

        text_widget = tk.Label(temp_frame, text=ai_reply, wraplength=wraplength, justify="left", anchor="w")
        text_widget.grid(row=0, column=1, sticky="nsew")

        button_frame = tk.Frame(temp_frame)
        button_frame.grid(row=1, column=1, sticky="nw", pady=(5, 0))

        copy_button = tk.Label(button_frame, text="复制", fg="gray", cursor="hand2")
        copy_button.pack(side="left", padx=(0, 5))
        copy_button.bind("<Button-1>", lambda e: self.copy_to_clipboard(ai_reply, copy_button))

        edit_button = tk.Label(button_frame, text="编辑", fg="gray", cursor="hand2")
        edit_button.pack(side="left")
        edit_button.bind("<Button-1>",
                         lambda e: self.toggle_edit_mode(temp_frame, text_widget, edit_button, "assistant", ai_reply))

        self.adjust_scroll_region()

        self.debug_info = json.dumps(response.to_dict(), indent=4, ensure_ascii=False)
        self.messages.append({"role": "assistant", "content": ai_reply})

    def show_debug_info(self):
        debug_window = tk.Toplevel(self.root)
        debug_window.title("Debug Information")
        debug_text = scrolledtext.ScrolledText(debug_window, wrap=tk.WORD, width=60, height=20)
        debug_text.pack(padx=10, pady=10, fill=tk.BOTH, expand=True)
        debug_text.insert(tk.END, self.debug_info)

    def _on_mousewheel(self, event):
        # 获取当前的滚动位置
        current_position = self.canvas.yview()[0]

        # 如果当前位置已经在顶部，并且尝试向上滚动，则不执行滚动
        if (current_position <= 0 and event.delta > 0):
            return

        self.canvas.yview_scroll(int(-1 * (event.delta / 120)), "units")

    def update_wraplengths(self, event=None):
        # 获取wraplength和Text部件宽度
        wraplength = self.get_wraplength()
        text_width = self.get_text_width()

        # 更新所有Label部件的wraplength和Text部件的width
        for frame in self.scrollable_frame.winfo_children():
            for widget in frame.winfo_children():
                if isinstance(widget, tk.Label):
                    widget.config(wraplength=wraplength)
                elif isinstance(widget, tk.Text):
                    widget.config(width=text_width)

    def get_wraplength(self):
        # 获取窗口的当前宽度
        window_width = self.root.winfo_width()
        # 假设图片宽度为30像素，设置Label的wraplength为窗口宽度减去图片宽度
        return window_width - 110  # 50包含了图片宽度和一些边距

    def get_text_width(self):
        # 获取窗口的当前宽度
        window_width = self.root.winfo_width()
        # 假设每个字符的平均宽度为7像素，图片宽度为30像素
        return (window_width - 130) // 9  # 减去图片宽度和一些边距

if __name__ == "__main__":
    root = tk.Tk()
    app = ChatApp(root)
    root.mainloop()