119 lines
3.4 KiB
Python
119 lines
3.4 KiB
Python
"""Use DeepSeek to summarize top news into 500-word Chinese briefs."""
|
||
import json, os, requests
|
||
from datetime import datetime, timezone, timedelta
|
||
|
||
API_KEY = "sk-bbca4a0380d549389f0d27cdea0b5228"
|
||
API_URL = "https://api.deepseek.com/v1/chat/completions"
|
||
MODEL = "deepseek-v4-flash"
|
||
|
||
NEWS_FILE = "/var/www/nav/data/news.json"
|
||
OUTPUT = "/var/www/nav/data/news_summary.json"
|
||
tz = timezone(timedelta(hours=8))
|
||
|
||
|
||
def load_raw_news():
|
||
if not os.path.exists(NEWS_FILE):
|
||
return []
|
||
with open(NEWS_FILE) as f:
|
||
data = json.load(f)
|
||
return data.get("items", [])
|
||
|
||
|
||
def summarize(items):
|
||
"""Send raw news to DeepSeek and get summarized articles."""
|
||
# Build a condensed input for the AI (titles + summaries)
|
||
news_text = ""
|
||
for i, item in enumerate(items[:20], 1):
|
||
title = item.get("title", "")
|
||
summary = item.get("summary", "")[:150].replace("<", "").replace(">", "")
|
||
news_text += f"{i}. [{item['source']}] {title}\n {summary}\n\n"
|
||
|
||
prompt = f"""你是一个新闻整理助手。以下是来自BBC世界新闻、BBC中文、Hacker News的最新资讯。
|
||
|
||
请从以上内容中精选出最重要的10条新闻，每条用中文写成一段300-500字的精炼摘要。
|
||
要求：
|
||
1. 选择真正重要、有信息量的新闻（优先世界大事、科技突破、财经动向）
|
||
2. 每条摘要独立、完整，包含核心事实和背景
|
||
3. 语言简洁有力，适合快速阅读
|
||
4. 按重要性排序
|
||
5. 每条附上来源标签
|
||
|
||
输出格式为JSON数组：
|
||
[
|
||
{{
|
||
"rank": 1,
|
||
"title": "精炼标题（20字以内）",
|
||
"source": "BBC World",
|
||
"summary": "300-500字中文摘要..."
|
||
}},
|
||
...
|
||
]
|
||
|
||
只输出JSON，不要其他内容。
|
||
|
||
今日要闻原始数据：
|
||
{news_text}"""
|
||
|
||
resp = requests.post(
|
||
API_URL,
|
||
headers={"Authorization": f"Bearer {API_KEY}", "Content-Type": "application/json"},
|
||
json={
|
||
"model": MODEL,
|
||
"messages": [{"role": "user", "content": prompt}],
|
||
"temperature": 0.5,
|
||
"max_tokens": 6000,
|
||
},
|
||
timeout=120,
|
||
)
|
||
|
||
if resp.status_code != 200:
|
||
print(f"API error: {resp.status_code} {resp.text[:200]}")
|
||
return None
|
||
|
||
result = resp.json()
|
||
content = result["choices"][0]["message"]["content"]
|
||
|
||
# Parse JSON from response
|
||
content = content.strip()
|
||
if content.startswith("```"):
|
||
content = content.split("\n", 1)[1]
|
||
content = content.rsplit("\n", 1)[0]
|
||
if content.endswith("```"):
|
||
content = content[:-3]
|
||
content = content.strip()
|
||
|
||
try:
|
||
articles = json.loads(content)
|
||
except:
|
||
print(f"JSON parse failed, content: {content[:300]}")
|
||
return None
|
||
|
||
return articles
|
||
|
||
|
||
if __name__ == "__main__":
|
||
now = datetime.now(tz).strftime("%Y-%m-%d %H:%M")
|
||
items = load_raw_news()
|
||
if not items:
|
||
print("No news data")
|
||
exit(0)
|
||
|
||
print(f"Loaded {len(items)} raw news items. Summarizing...")
|
||
articles = summarize(items)
|
||
|
||
if not articles:
|
||
print("Summarization failed")
|
||
exit(1)
|
||
|
||
result = {
|
||
"updated_at": now,
|
||
"total": len(articles),
|
||
"articles": articles,
|
||
}
|
||
|
||
os.makedirs(os.path.dirname(OUTPUT), exist_ok=True)
|
||
with open(OUTPUT, "w", encoding="utf-8") as f:
|
||
json.dump(result, f, ensure_ascii=False, indent=2)
|
||
|
||
print(f"Written {len(articles)} summarized articles to {OUTPUT}")
|