AI 自动把公众号文章转成小红书/微博帖

pip install readability-lxml requests
curl -fsSL https://ollama.com/install.sh | sh
ollama pull qwen2.5:7b

#!/usr/bin/env python3
"""
AI 内容改写器：公众号文章 → 小红书 + 微博
"""
import sys
import json
import requests
import subprocess
from readability import Document
from html import unescape

OLLAMA_URL = "http://localhost:11434/api/generate"
MODEL = "qwen2.5:7b"

def fetch_article(url: str) -> str:
    """从 URL 提取正文"""
    resp = requests.get(url, headers={
        "User-Agent": "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) "
                      "AppleWebKit/537.36 Chrome/120.0.0.0 Safari/537.36"
    }, timeout=15)
    doc = Document(resp.text)
    title = unescape(doc.title())
    summary = unescape(doc.summary().strip())
    return f"标题：{title}\n正文：{summary}"

def call_ollama(prompt: str, system: str = "") -> str:
    """调用本地 Ollama 模型"""
    payload = {
        "model": MODEL,
        "prompt": prompt,
        "system": system,
        "stream": False,
        "options": {"temperature": 0.7, "num_predict": 2000}
    }
    resp = requests.post(OLLAMA_URL, json=payload, timeout=120)
    return resp.json().get("response", "")

def rewrite_to_xiaohongshu(article_text: str) -> str:
    """改写为小红书风格"""
    system = """你是一个小红书爆款文案写手。
特点：标题有吸引力、多用 emoji、分段清晰、结尾加标签。"""
    prompt = f"""请把以下文章改写成小红书笔记：

{article_text}

要求：
1. 标题 15 字以内，要吸睛
2. 正文分 3-5 段，每段 2-4 句
3. 适当使用 emoji（每段开头一个）
4. 结尾加 5-8 个相关标签（#开头）
5. 语气口语化、有感染力

只输出改写后的完整笔记，不要解释。"""
    return call_ollama(prompt, system)

def rewrite_to_weibo(article_text: str) -> str:
    """改写为微博风格"""
    system = "你是一个微博运营专家，擅长写短平快的热门文案。"
    prompt = f"""请把以下文章改写成微博文案：

{article_text}

要求：
1. 字数 140 字以内
2. 开头一句抓人眼球
3. 内容精炼，有 1-2 个核心观点
4. 结尾加 2-3 个话题标签（#话题# 格式）
5. 适合转发和评论

只输出改写后的文案，不要解释。"""
    return call_ollama(prompt, system)

def main():
    if len(sys.argv) < 2:
        print("用法: python3 content_rewriter.py <公众号文章URL>")
        print("或者: cat article.txt | python3 content_rewriter.py")
        sys.exit(1)

    url = sys.argv[1]

    # 判断是 URL 还是文件
    if url.startswith("http"):
        print(f"正在抓取文章: {url}")
        article_text = fetch_article(url)
    else:
        with open(url, "r", encoding="utf-8") as f:
            article_text = f.read()

    print("\n" + "="*50)
    print("📕 小红书笔记")
    print("="*50)
    xhs = rewrite_to_xiaohongshu(article_text)
    print(xhs)

    print("\n" + "="*50)
    print("📱 微博文案")
    print("="*50)
    wb = rewrite_to_weibo(article_text)
    print(wb)

    # 保存到文件
    import os
    os.makedirs("output", exist_ok=True)
    with open("output/xiaohongshu.md", "w", encoding="utf-8") as f:
        f.write(xhs)
    with open("output/weibo.txt", "w", encoding="utf-8") as f:
        f.write(wb)
    print("\n✅ 已保存到 output/ 目录")

if __name__ == "__main__":
    main()

# 直接传 URL
python3 content_rewriter.py "https://mp.weixin.qq.com/s/xxxxxx"

# 或者传本地文本文件
python3 content_rewriter.py ./article.txt

#!/usr/bin/env python3
"""批量改写多篇文章"""
import subprocess
import sys

urls = [
    "https://mp.weixin.qq.com/s/xxxx1",
    "https://mp.weixin.qq.com/s/xxxx2",
    "https://mp.weixin.qq.com/s/xxxx3",
]

for i, url in enumerate(urls, 1):
    print(f"\n[{i}/{len(urls)}] 处理: {url}")
    subprocess.run(["python3", "content_rewriter.py", url], check=True)
    print(f"✅ {i}/{len(urls)} 完成")