import requests
import json
import os
from dotenv import load_dotenv

# 配置你的 API 信息（就像拨号上网需要账号密码）
load_dotenv()  # 加载 .env 文件
API_KEY = os.getenv("DEEPSEEK_API_KEY") # 从环境变量里读取
URL = "https://api.deepseek.com/chat/completions"

def ask_ai(prompt):
    # 1. 设置请求头：告诉服务器你是谁，你要发什么格式的数据
    headers = {
        "Authorization": f"Bearer {API_KEY}",
        "Content-Type": "application/json"
    }

    # 2. 设置请求体：你要问 AI 的问题
    # 这就是你之前练过的“字典”结构！
    data = {
        "model": "deepseek-chat",
        "messages": [
            {"role": "system", "content": "你是一个专业的文本总结助手。"},
            {"role": "user", "content": prompt}
        ],
        "temperature": 0.7  # 控制随机性，0.7 比较平衡
    }

    try:
        # 3. 发送 POST 请求
        response = requests.post(URL, headers=headers, data=json.dumps(data))
        
        # 4. 解析结果
        if response.status_code == 200:
            result = response.json()
            # 这里的提取逻辑就是你之前练过的“剥洋葱”！
            return result["choices"][0]["message"]["content"]
        else:
            return f"请求失败，状态码：{response.status_code}, 错误：{response.text}"
            
    except Exception as e:
        return f"网络异常：{e}"

# --- 主程序逻辑 ---
if __name__ == "__main__":
    # 读取你之前的 task.txt 内容
    with open("./python/temp/task.txt", "r", encoding="utf-8") as f:
        user_input = f.read()
    
    print("正在请求 AI 总结，请稍候...")
    summary = ask_ai(user_input)
    print(f"\nAI 的总结结果：\n{summary}")