import tkinter as tk
from tkinter import messagebox, scrolledtext
from openai import OpenAI
# 定义调用AI API的函数
def get_ai_answer(question):
    client = OpenAI(api_key="??????????", base_url="https://api.deepseek.com")
    response = client.chat.completions.create(
        model="deepseek-chat",
        messages=[
            {"role": "user", "content": question},
        ],
        stream=False
    )
    return response.choices[0].message.content


def on_ask():
    question = entry.get()
    if not question.strip():
        messagebox.showerror("Input Error", "Please enter a question")
        return

    try:
        answer = get_ai_answer(question)
        display_answer(answer)
    except Exception as e:
        messagebox.showerror("Error", f"An error occurred: {e}")

# 显示答案
def display_answer(answer):
    result_text.config(state=tk.NORMAL)  # 允许修改
    result_text.delete(1.0, tk.END)  # 清空当前内容
    result_text.insert(tk.END, answer)  # 插入新的答案
    result_text.config(state=tk.DISABLED)  # 禁止修改

# 创建主窗口
root = tk.Tk()
root.title("AI Question Answering")

# 设置窗口大小
root.geometry("400x300")

# 创建并放置输入框和标签
input_label = tk.Label(root, text="Enter your question:")
input_label.pack(pady=5)

entry = tk.Entry(root, width=50)
entry.pack(pady=5)

# 创建并放置计算按钮
ask_button = tk.Button(root, text="Ask", command=on_ask)
ask_button.pack(pady=5)

# 创建并放置结果文本框
result_text = scrolledtext.ScrolledText(root, wrap=tk.WORD, width=50, height=10, state=tk.DISABLED)
result_text.pack(pady=5)

# 运行主循环
root.mainloop()