import tkinter as tk
from tkinter import filedialog, messagebox, scrolledtext
import requests
import json
import time

# 替换为你的 Deepseek API 地址和密钥
API_URL = "https://api.deepseek.com/v1/chat/completions"
API_KEY = "your_deepseek_api_key_here"

HEADERS = {
    "Authorization": f"Bearer {API_KEY}",
    "Content-Type": "application/json"
}

def split_code(code, max_tokens=3000):
    """将大段代码分割成适合 API 调用的块"""
    lines = code.splitlines()
    chunks = []
    current_chunk = []
    current_length = 0

    for line in lines:
        line_length = len(line.encode('utf-8'))
        if current_length + line_length > max_tokens:
            chunks.append("\n".join(current_chunk))
            current_chunk = []
            current_length = 0
        current_chunk.append(line)
        current_length += line_length

    if current_chunk:
        chunks.append("\n".join(current_chunk))

    return chunks

def analyze_code(code_chunk):
    """调用 Deepseek API 分析代码"""
    prompt = f"请检查以下代码中的潜在问题，包括语法错误、风格问题、逻辑漏洞等：\n\n{code_chunk}"
    data = {
        "model": "deepseek-chat",
        "messages": [
            {"role": "system", "content": "你是一个专业的代码审查助手。"},
            {"role": "user", "content": prompt}
        ],
        "temperature": 0.3
    }

    try:
        response = requests.post(API_URL, headers=HEADERS, json=data)
        response.raise_for_status()
        result = response.json()
        return result['choices'][0]['message']['content']
    except Exception as e:
        return f"API 调用失败: {str(e)}"

def load_file():
    file_path = filedialog.askopenfilename(filetypes=[("Python 文件", "*.py"), ("所有文件", "*.*")])
    if file_path:
        with open(file_path, 'r', encoding='utf-8') as f:
            code_text.delete(1.0, tk.END)
            code_text.insert(tk.END, f.read())

def analyze():
    code = code_text.get(1.0, tk.END)
    if not code.strip():
        messagebox.showwarning("警告", "请先加载代码")
        return

    result_text.delete(1.0, tk.END)
    result_text.insert(tk.END, "正在分析代码...\n")
    root.update()

    chunks = split_code(code)
    all_results = []

    for i, chunk in enumerate(chunks):
        result = analyze_code(chunk)
        all_results.append(f"=== 第 {i+1} 段代码分析结果 ===\n{result}\n")
        time.sleep(1)  # 避免 API 请求过快

    result_text.delete(1.0, tk.END)
    result_text.insert(tk.END, "\n".join(all_results))

# GUI 构建
root = tk.Tk()
root.title("Deepseek 代码检查器")

frame_top = tk.Frame(root)
frame_top.pack(fill=tk.BOTH, expand=True, padx=10, pady=10)

code_label = tk.Label(frame_top, text="代码内容：")
code_label.pack(anchor='w')

code_text = scrolledtext.ScrolledText(frame_top, height=20, width=100)
code_text.pack(fill=tk.BOTH, expand=True)

frame_btn = tk.Frame(root)
frame_btn.pack(pady=5)

load_btn = tk.Button(frame_btn, text="加载代码文件", command=load_file)
load_btn.pack(side=tk.LEFT, padx=5)

analyze_btn = tk.Button(frame_btn, text="开始分析", command=analyze)
analyze_btn.pack(side=tk.LEFT, padx=5)

result_label = tk.Label(root, text="分析结果：")
result_label.pack(anchor='w', padx=10)

result_text = scrolledtext.ScrolledText(root, height=15, width=100)
result_text.pack(fill=tk.BOTH, expand=True, padx=10, pady=(0, 10))

root.mainloop()