import requests
import json
import time
from datetime import datetime

# 配置
PERPLEXITY_API_KEY = "your_perplexity_api_key"
MONITORED_TERMS = ["你的品牌名", "你的产品名", "你的核心域名"] # 要监控的实体
QUESTIONS = [
    "What is the best tool for [你的行业]?",
    "How to solve [你的领域痛点]?",
    "Compare [你的产品] with [竞品]"
]

def query_perplexity(question):
    """向Perplexity API发送问题并返回回答文本"""
    url = "https://api.perplexity.ai/chat/completions"
    headers = {
        "Authorization": f"Bearer {PERPLEXITY_API_KEY}",
        "Content-Type": "application/json"
    }
    payload = {
        "model": "sonar-pro",
        "messages": [
            {"role": "system", "content": "你是一个帮助监控的助手，请完整回答用户问题。"},
            {"role": "user", "content": question}
        ]
    }
    try:
        response = requests.post(url, headers=headers, json=payload, timeout=30)
        response.raise_for_status()
        return response.json()["choices"][0]["message"]["content"]
    except Exception as e:
        print(f"Error querying Perplexity: {e}")
        return None

def check_references(text, terms):
    """检查文本中是否包含监控词条"""
    found = []
    if text:
        for term in terms:
            if term.lower() in text.lower():
                found.append(term)
    return found

def log_reference(question, answer, found_terms):
    """将结果记录到日志文件或数据库（此处用JSON Lines格式）"""
    log_entry = {
        "timestamp": datetime.now().isoformat(),
        "question": question,
        "answer_preview": answer[:200] if answer else "N/A",
        "found_terms": found_terms,
        "engine": "Perplexity"
    }
    with open("reference_log.jsonl", "a") as f:
        f.write(json.dumps(log_entry) + "\n")

# 主循环
if __name__ == "__main__":
    for question in QUESTIONS:
        print(f"Querying: {question}")
        answer = query_perplexity(question)
        if answer:
            found = check_references(answer, MONITORED_TERMS)
            if found:
                print(f"  -> Found references: {found}")
            else:
                print("  -> No references found.")
            log_reference(question, answer, found)
        time.sleep(5) # 避免API限流

from flask import Flask, render_template, jsonify
import json
from collections import defaultdict, Counter

app = Flask(__name__)

def load_logs():
    """从JSONL文件加载日志"""
    logs = []
    try:
        with open("reference_log.jsonl", "r") as f:
            for line in f:
                logs.append(json.loads(line))
    except FileNotFoundError:
        pass
    return logs

@app.route('/')
def dashboard():
    logs = load_logs()
    # 计算统计
    total_queries = len(logs)
    total_references = sum(1 for log in logs if log['found_terms'])
    term_counts = Counter()
    for log in logs:
        for term in log['found_terms']:
            term_counts[term] += 1
    return render_template('dashboard.html', 
                           total_queries=total_queries,
                           total_references=total_references,
                           term_counts=term_counts.most_common(10))

if __name__ == '__main__':
    app.run(debug=True, port=5000)