Ubuntu
134 lines
4.1 KiB
Python
Executable File
134 lines
4.1 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
"""
|
|
小红书热点趋势工作流
|
|
基于热门话题创作并发布原创内容
|
|
"""
|
|
|
|
import os
|
|
import sys
|
|
import json
|
|
import re
|
|
import requests
|
|
from datetime import datetime
|
|
|
|
sys.path.insert(0, os.path.dirname(__file__))
|
|
from xiaohongshu_multi_image import XiaohongshuMultiImagePublisher
|
|
|
|
class XHSTrendingWorkflow:
|
|
"""小红书热点工作流"""
|
|
|
|
def __init__(self):
|
|
self.api_key = os.environ.get('MOONSHOT_API_KEY')
|
|
self.api_base = os.environ.get('MOONSHOT_API_BASE', 'https://api.moonshot.cn/v1')
|
|
|
|
def search_trending(self, keyword="AI科技", limit=5):
|
|
"""
|
|
搜索小红书热门笔记
|
|
注意：这里使用xhs CLI命令，需要在shell中执行
|
|
"""
|
|
import subprocess
|
|
result = subprocess.run(
|
|
['xhs', 'search', keyword],
|
|
capture_output=True, text=True
|
|
)
|
|
# 解析结果...（简化版）
|
|
return []
|
|
|
|
def read_note(self, note_id):
|
|
"""读取笔记详情"""
|
|
import subprocess
|
|
result = subprocess.run(
|
|
['xhs', 'read', note_id],
|
|
capture_output=True, text=True
|
|
)
|
|
return result.stdout
|
|
|
|
def generate_content(self, topic):
|
|
"""
|
|
基于话题AI创作原创内容
|
|
"""
|
|
messages = [
|
|
{"role": "system", "content": "你是科技内容创作专家，擅长撰写小红书上受欢迎的科技趋势分析文章。"},
|
|
{"role": "user", "content": f"""基于"{topic}"这个话题，创作一篇全新的小红书科技笔记。
|
|
|
|
创作要求：
|
|
1. 标题吸引人，20-30字，不用任何markdown符号
|
|
2. 正文600-800字，结构清晰
|
|
3. 使用-作为列表符号，不要用**加粗
|
|
4. 自然融入emoji
|
|
5. 包含独到的技术趋势洞察
|
|
6. 纯原创，不要复制任何现有文章
|
|
7. 结尾带5个话题标签，格式为#标签
|
|
|
|
输出格式：
|
|
标题：[你的标题]
|
|
|
|
正文：[你的正文]
|
|
|
|
标签：[标签1] [标签2] [标签3] [标签4] [标签5]"""}
|
|
]
|
|
|
|
response = requests.post(
|
|
f"{self.api_base}/chat/completions",
|
|
headers={
|
|
"Authorization": f"Bearer {self.api_key}",
|
|
"Content-Type": "application/json"
|
|
},
|
|
json={
|
|
"model": "kimi-k2.5",
|
|
"messages": messages,
|
|
"temperature": 1,
|
|
"max_tokens": 1500
|
|
},
|
|
timeout=60
|
|
)
|
|
|
|
return response.json()['choices'][0]['message']['content']
|
|
|
|
def publish(self, content_text):
|
|
"""
|
|
发布到小红书
|
|
"""
|
|
# 解析内容
|
|
title_match = re.search(r'标题：(.+)', content_text)
|
|
title = title_match.group(1) if title_match else "科技趋势分析"
|
|
|
|
# 清理内容
|
|
content = content_text
|
|
content = re.sub(r'标题：.+\n', '', content)
|
|
content = re.sub(r'正文：', '', content)
|
|
content = re.sub(r'标签：', '\n\n', content)
|
|
content = re.sub(r'\*\*(.+?)\*\*', r'\1', content)
|
|
content = re.sub(r'##?\s*', '', content)
|
|
|
|
# 发布
|
|
publisher = XiaohongshuMultiImagePublisher()
|
|
tags = ["科技趋势", "人工智能", "AI产业", "商业洞察", "科技创新"]
|
|
return publisher.publish_to_xiaohongshu(title, content, tags, "tech")
|
|
|
|
def run(self, topic="AI产业趋势"):
|
|
"""执行完整工作流"""
|
|
print(f"🚀 开始创作: {topic}")
|
|
|
|
# 1. AI创作内容
|
|
print("📝 AI创作原创内容...")
|
|
content = self.generate_content(topic)
|
|
|
|
# 2. 发布到小红书
|
|
print("📱 发布到小红书...")
|
|
success = self.publish(content)
|
|
|
|
if success:
|
|
print("✅ 发布成功！")
|
|
else:
|
|
print("❌ 发布失败")
|
|
|
|
return success
|
|
|
|
if __name__ == "__main__":
|
|
workflow = XHSTrendingWorkflow()
|
|
|
|
# 可以从命令行参数获取话题
|
|
topic = sys.argv[1] if len(sys.argv) > 1 else "AI产业趋势"
|
|
workflow.run(topic)
|