import anthropic

client = anthropic.Anthropic(
    api_key='your-derouter-key',
    base_url='https://api.derouter.ai/proxy/v1'
)

def generate_xiaohongshu(topic, style='种草'):
    prompt = f'''你是小红书内容创作专家，粉丝 50 万。

请为以下主题创作一篇小红书笔记：
主题：{topic}
风格：{style}

严格遵循以下结构：

【标题】（18-22字，必须包含）：
- 数字（如「3个」「7天」「100%」）
- emoji（2-3个）
- 痛点或好奇心钩子

【正文】（500-800字）：
- 第一段：痛点共鸣（「姐妹们是不是也遇到过...」）
- 中间段：干货内容（分点列举，每点带 emoji）
- 最后段：总结 + 互动引导

【话题标签】：5-8个相关标签

风格要求：
- 口语化，像在和闺蜜聊天
- 用「姐妹」「集美」「绝绝子」等小红书用语
- 适当使用 emoji 但不过度
- 有真实使用感受，不要广告味
- 节奏感强，多用短句和换行'''

    msg = client.messages.create(
        model='claude-sonnet-4-6',
        max_tokens=2048,
        messages=[{'role': 'user', 'content': prompt}]
    )
    return msg.content[0].text

# 批量生成
topics = [
    '油皮平价防晒推荐',
    '小个子穿搭显高技巧',
    '租房必备好物清单',
    '考研每日作息安排',
]

for topic in topics:
    note = generate_xiaohongshu(topic)
    print(f'--- {topic} ---')
    print(note)
    print()

def generate_title_variants(topic, count=5):
    prompt = f'''为小红书笔记生成 {count} 个标题变体。

主题：{topic}

每个标题用不同策略：
1. 数字 + 痛点型（「3 个让你少走弯路的 xxx」）
2. 对比反差型（「用了 xxx 之后，同事都问我...」）
3. 清单型（「xxx 必备/必看/必收藏」）
4. 提问型（「为什么 xxx？原因居然是...」）
5. 结果展示型（「坚持 xxx 一个月后...」）

每个标题 18-22 字，包含 2-3 个 emoji。'''

    msg = client.messages.create(
        model='claude-sonnet-4-6',
        max_tokens=512,
        messages=[{'role': 'user', 'content': prompt}]
    )
    return msg.content[0].text

def generate_video_script(topic, duration=60):
    prompt = f'''你是抖音短视频编导，多个百万粉账号操盘手。

主题：{topic}
时长：{duration}秒

请生成完整的短视频脚本：

【开头 Hook】（前 3 秒，决定是否划走）：
- 用悬念/冲突/反常识开头
- 例：「我花了 3 万块买的教训，免费告诉你」

【正文内容】（按时间轴）：
- 00:03-00:10：痛点描述
- 00:10-00:40：核心内容（3 个要点）
- 00:40-00:55：总结 + 金句
- 00:55-01:00：CTA（关注/收藏/评论）

【画面建议】：每段配什么画面/字幕
【BGM 风格】：推荐背景音乐类型
【字幕重点】：需要加粗/加大的关键句'''

    msg = client.messages.create(
        model='claude-sonnet-4-6',
        max_tokens=2048,
        messages=[{'role': 'user', 'content': prompt}]
    )
    return msg.content[0].text

def generate_wechat_article(topic, word_count=2000):
    prompt = f'''你是公众号头部创作者，擅长写深度长文。

主题：{topic}
字数：{word_count}字

文章结构：
1. 标题：引人点击但不标题党（20-30字）
2. 导语：100字内，点出核心论点
3. 正文：
   - 3-5 个小标题，每个标题下 300-500 字
   - 每个观点配案例/数据佐证
   - 适当用金句增加记忆点
4. 结尾：总结 + 引导转发

风格要求：
- 有观点有态度，不是百科式平铺
- 逻辑严密，每段之间有过渡
- 长短句交替，节奏感好
- 专业但不晦涩
- 引用要有具体来源'''

    msg = client.messages.create(
        model='claude-opus-4-6',  # 长文用 Opus 质量更好
        max_tokens=8000,
        messages=[{'role': 'user', 'content': prompt}]
    )
    return msg.content[0].text

def adapt_content(original_content, target_platform):
    '''将一篇内容适配到不同平台'''
    platform_rules = {
        '小红书': '500-800字，口语化，emoji丰富，话题标签',
        '抖音': '60秒口播脚本，前3秒要有hook',
        '公众号': '2000-3000字深度长文，有论据',
        '知乎': '1500字专业回答，有数据引用',
        '微博': '140字精华版 + 9宫格配图文案',
    }

    prompt = f'''将以下内容适配为{target_platform}格式。

平台规则：{platform_rules[target_platform]}

原始内容：
{original_content}

请输出适配后的完整内容，符合该平台的风格和用户习惯。'''

    msg = client.messages.create(
        model='claude-sonnet-4-6',
        max_tokens=4096,
        messages=[{'role': 'user', 'content': prompt}]
    )
    return msg.content[0].text

# 一篇内容 → 5个平台
original = '关于AI工具提升工作效率的深度分析...'
for platform in ['小红书', '抖音', '公众号', '知乎', '微博']:
    adapted = adapt_content(original, platform)
    print(f'=== {platform} ===')
    print(adapted[:200] + '...')  # 预览前200字

def generate_content_calendar(niche, days=7):
    '''生成一周内容日历'''
    prompt = f'''你是自媒体运营总监。

领域：{niche}
时间范围：未来 {days} 天

请规划内容日历，每天 2 篇内容（1 篇小红书 + 1 篇抖音/公众号交替）。

每篇内容包含：
- 平台
- 选题（一句话描述）
- 选题理由（为什么现在发这个）
- 预计流量评级（A/B/C）
- 关键词

考虑因素：
- 当前热点和季节性话题
- 内容类型多样化（干货/种草/故事/争议）
- 发布时间建议
- 系列内容的连贯性'''

    msg = client.messages.create(
        model='claude-sonnet-4-6',
        max_tokens=4096,
        messages=[{'role': 'user', 'content': prompt}]
    )
    return msg.content[0].text

calendar = generate_content_calendar('职场效率/AI工具', days=7)