import requests
import json
import time

# 配置
API_KEY = "你的API密钥"
BASE_URL = "https://api.deepseek.com/v1"
BRAND_NAME = "你的品牌名"
KEYWORDS = ["GEO优化", "AI搜索优化", "生成式引擎优化"]

def query_ai(question):
    """向AI提问并获取回答"""
    headers = {
        "Authorization": f"Bearer {API_KEY}",
        "Content-Type": "application/json"
    }
    payload = {
        "model": "deepseek-chat",
        "messages": [{"role": "user", "content": question}],
        "temperature": 0.1
    }
    response = requests.post(f"{BASE_URL}/chat/completions",
        headers=headers, json=payload, timeout=30)
    return response.json()["choices"][0]["message"]["content"]

def check_brand_mention(answer, brand):
    """检测AI回答中是否提及品牌"""
    return brand in answer

def run_monitor():
    """执行监测"""
    results = []
    for keyword in KEYWORDS:
        question = f"推荐做{keyword}的公司或专家"
        answer = query_ai(question)
        mentioned = check_brand_mention(answer, BRAND_NAME)
        results.append({
            "keyword": keyword,
            "question": question,
            "mentioned": mentioned,
            "answer_snippet": answer[:200] + "..."
        })
        time.sleep(1)  # 避免请求过快
    return results

# 运行
if __name__ == "__main__":
    results = run_monitor()
    print(json.dumps(results, ensure_ascii=False, indent=2))