批量克隆"AI算命"生意！Python+DeepSeek自动生成星座运势，日更100篇小红书

import random
import requests
import json
from datetime import datetime, timedelta

# 配置DeepSeek API
DEEPSEEK_API_KEY = "你的API_KEY"
API_URL = "https://api.deepseek.com/v1/chat/completions"

# 星座列表
ZODIAC_SIGNS = [
    "白羊座", "金牛座", "双子座", "巨蟹座", 
    "狮子座", "处女座", "天秤座", "天蝎座",
    "射手座", "摩羯座", "水瓶座", "双鱼座"
]

# 运势关键词库
KEYWORDS = {
    "love": ["桃花运", "邂逅", "复合", "甜蜜", "冷战", "理解", "包容", "浪漫"],
    "career": ["升职", "加薪", "跳槽", "合作", "项目", "贵人", "挑战", "机遇"],
    "wealth": ["偏财", "正财", "投资", "消费", "储蓄", "理财", "债务", "收益"],
    "health": ["精力", "睡眠", "饮食", "运动", "压力", "免疫力", "情绪", "平衡"]
}

def generate_horoscope(sign):
    """生成单个星座的运势内容"""
    # 随机选择本周运势基调
    overall_luck = random.choice(["极佳", "不错", "平稳", "需谨慎", "挑战多"])

    # 构建提示词
    prompt = f"""
    你是一位专业的星座运势撰写人，请为{sign}生成一篇本周运势分析，要求：
    1. 开头用一句吸引眼球的话概括本周运势，整体运势为{overall_luck}
    2. 分爱情、事业、财运、健康四个方面分析，每方面2-3句话
    3. 使用生动具体的描述，避免笼统
    4. 结尾加上鼓励或提醒的话语
    5. 适当使用emoji表情

    参考关键词：
    {KEYWORDS}
    """

    headers = {
        "Content-Type": "application/json",
        "Authorization": f"Bearer {DEEPSEEK_API_KEY}"
    }

    payload = {
        "model": "deepseek-chat",
        "messages": [{"role": "user", "content": prompt}],
        "temperature": 0.7,
        "max_tokens": 800
    }

    response = requests.post(API_URL, headers=headers, data=json.dumps(payload))
    result = response.json()

    return result["choices"][0]["message"]["content"]

def batch_generate_horoscopes():
    """批量生成所有星座运势"""
    horoscopes = {}
    for sign in ZODIAC_SIGNS:
        content = generate_horoscope(sign)
        horoscopes[sign] = content
        print(f"已生成 {sign} 运势")
    return horoscopes

def save_to_markdown(horoscopes):
    """保存为Markdown格式，方便发布到小红书"""
    today = datetime.now().strftime("%Y年%m月%d日")
    filename = f"星座运势_{today}.md"

    with open(filename, "w", encoding="utf-8") as f:
        f.write(f"# ⭐ 十二星座本周运势({today})\n\n")
        for sign, content in horoscopes.items():
            f.write(f"## ♈ {sign}\n\n")
            f.write(content + "\n\n")
            f.write("---\n\n")
    print(f"运势已保存到 {filename}")

if __name__ == "__main__":
    print("开始生成星座运势...")
    horoscopes = batch_generate_horoscopes()
    save_to_markdown(horoscopes)