用 Python + Ollama 搭一个本地博客文章改写器：一篇文章变多个平台版本

# 安装 Ollama（如果还没装）
curl -fsSL https://ollama.com/install.sh | sh

# 拉取中文能力强的模型
ollama pull qwen2.5:7b

pip install ollama

import ollama
import sys
import json
from pathlib import Path


def rewrite_article(text: str, platform: str) -> str:
    """根据平台类型改写文章"""
    
    prompts = {
        "wechat_moments": """你是一个内容改写助手。把下面这篇文章改写成适合微信朋友圈发布的短文案。

要求：
- 3-5 行，每行不超过 20 字
- 口语化，像在跟朋友聊天
- 可以适当加 emoji
- 开头要有吸引力，让人想点全文
- 结尾引导点击原文链接

只输出改写后的文案，不要解释。

文章：
{text}""",
        
        "weibo": """你是一个内容改写助手。把下面这篇文章改写成适合微博发布的短文案。

要求：
- 280 字以内
- 开头第一句要抓眼球
- 保留文章最核心的 1-2 个观点
- 口语化，有态度
- 可以加 1-2 个相关话题标签

只输出改写后的文案，不要解释。

文章：
{text}""",
        
        "linkedin": """你是一个内容改写助手。把下面这篇文章改写成适合领英发布的帖子。

要求：
- 专业但不生硬的语气
- 1-2 段，每段 3-5 句话
- 强调这篇文章对职场人的实用价值
- 可以用"分享一个心得"开头
- 结尾加一个开放式问题引发讨论

只输出改写后的文案，不要解释。

文章：
{text}""",
        
        "email": """你是一个内容改写助手。把下面这篇文章改写成适合邮件列表发送的预览内容。

要求：
- 标题行：简洁有力，激发点击欲
- 开头：一句话概括今天内容的核心价值
- 中间：3-5 个要点，用短句列出
- 结尾：引导点击阅读完整文章
- 整体语气：友好、专业、像老朋友来信

只输出改写后的内容，格式如下：

标题：[标题]
预览：[一句话预览]
正文：[要点列表]
引导：[点击阅读引导语]

文章：
{text}"""
    }
    
    prompt = prompts.get(platform)
    if not prompt:
        available = ", ".join(prompts.keys())
        raise ValueError(f"未知平台类型。可用类型：{available}")
    
    response = ollama.chat(
        model="qwen2.5:7b",
        messages=[{"role": "user", "content": prompt.format(text=text)}]
    )
    return response["message"]["content"]


def main():
    if len(sys.argv) < 2:
        print("用法: python article_rewriter.py <文章文件路径> [平台]")
        print("")
        print("平台类型：")
        print("  wechat_moments  - 微信朋友圈")
        print("  weibo           - 微博")
        print("  linkedin        - 领英")
        print("  email           - 邮件列表")
        print("")
        print("示例：")
        print("  python article_rewriter.py my-article.md")
        print("  python article_rewriter.py my-article.md weibo")
        sys.exit(1)
    
    input_file = sys.argv[1]
    platform = sys.argv[2] if len(sys.argv) > 2 else None
    
    # 读取文章
    text = Path(input_file).read_text(encoding="utf-8")
    
    # 决定要改写的平台
    if platform:
        platforms = [platform]
    else:
        platforms = ["wechat_moments", "weibo", "linkedin", "email"]
    
    # 逐个平台改写
    results = {}
    for p in platforms:
        print(f"\n正在改写为 {p} 风格...")
        try:
            results[p] = rewrite_article(text, p)
            print(f"✓ {p} 完成")
        except Exception as e:
            print(f"✗ {p} 失败: {e}")
            results[p] = f"（生成失败）"
    
    # 保存结果
    output_dir = Path(input_file).parent / "rewritten"
    output_dir.mkdir(exist_ok=True)
    
    output_file = output_dir / f"{Path(input_file).stem}_versions.json"
    with open(output_file, "w", encoding="utf-8") as f:
        json.dump(results, f, ensure_ascii=False, indent=2)
    
    print(f"\n结果已保存到: {output_file}")
    
    # 同时输出到单独文件，方便复制
    for p, content in results.items():
        single_file = output_dir / f"{p}.md"
        single_file.write_text(content, encoding="utf-8")


if __name__ == "__main__":
    main()

# 改写全部平台
python article_rewriter.py my-article.md

# 只改写微博版本
python article_rewriter.py my-article.md weibo

# 改写朋友圈版本
python article_rewriter.py my-article.md wechat_moments

# 修改 prompts["wechat_moments"] 中的要求部分
# 把 "口语化" 改成 "像发朋友圈一样随意"
# 加一句 "不要用太书面化的词"