import anthropic

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

# 选题 brainstorm
msg = client.messages.create(
    model='claude-sonnet-4-6',
    max_tokens=4096,
    messages=[{
        'role': 'user',
        'content': '''我是市场营销专业本科生，帮我 brainstorm 毕业论文选题。

要求：
1. 2026 年有研究价值和热度的方向
2. 数据可获取（不需要大规模调研）
3. 难度适中（本科水平）
4. 有创新点但不用太前沿

请给出 10 个选题，每个包含：
- 题目
- 一句话说明研究什么
- 可能的研究方法
- 数据来源建议
- 创新点在哪'''
    }]
)

topic = '直播电商中AI虚拟主播对消费者购买意愿的影响研究'

prompt = f'''帮我撰写毕业论文开题报告的框架。

题目：{topic}
专业：市场营销
学位：本科

请按以下结构输出：

1. 研究背景（300字）
   - 行业现状
   - 存在的问题
   - 研究意义

2. 国内外研究现状（500字）
   - 国外研究（3-5个方向）
   - 国内研究（3-5个方向）
   - 研究空白/不足

3. 研究内容与方法
   - 研究问题（3个）
   - 研究方法（问卷/实验/案例分析）
   - 技术路线

4. 论文大纲（章节结构）

5. 研究进度安排

注意：这是框架，不是完整开题报告。每个部分给出关键论点和方向，我会自己展开。'''

msg = client.messages.create(
    model='claude-opus-4-6',  # 开题报告用 Opus 更有深度
    max_tokens=4096,
    messages=[{'role': 'user', 'content': prompt}]
)

prompt = '''我正在研究「AI虚拟主播对消费者购买意愿的影响」，请帮我梳理相关文献的理论脉络。

请从以下维度整理：
1. 核心理论框架（技术接受模型TAM、感知价值理论、拟社会关系理论等）
2. 每个理论的核心观点和代表学者
3. 这些理论如何应用到我的研究中
4. 理论之间的联系

输出格式：表格 + 文字说明'''

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

paper_abstract = '''（粘贴论文摘要或关键段落）'''

prompt = f'''请总结以下学术论文，提取对我的研究有用的信息：

论文内容：
{paper_abstract}

请输出：
1. 核心研究问题
2. 使用的理论/模型
3. 研究方法
4. 关键发现
5. 对我研究的启示（我研究AI虚拟主播对消费者购买意愿的影响）
6. 可引用的关键论点

格式：简洁的要点列表'''

msg = client.messages.create(
    model='claude-sonnet-4-6',  # 文献总结用 Sonnet 够了
    max_tokens=2048,
    messages=[{'role': 'user', 'content': prompt}]
)

prompt = '''帮我设计一份调查问卷，用于研究AI虚拟主播对消费者购买意愿的影响。

理论基础：技术接受模型(TAM) + 感知价值理论
变量：
- 自变量：感知有用性、感知易用性、感知趣味性、信任感
- 中介变量：感知价值
- 因变量：购买意愿
- 控制变量：年龄、性别、直播购物频率

要求：
1. 每个变量 3-5 个测量题项
2. 使用 Likert 5 级量表
3. 题项参考已有成熟量表，标注参考来源
4. 包含注意力检验题
5. 总题目不超过 35 个'''

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

prompt = '''我用 SPSS 做问卷数据分析，请告诉我完整的分析步骤和操作方法。

问卷信息：
- 变量：感知有用性、感知易用性、感知趣味性、信任感、感知价值、购买意愿
- 样本量：280份
- 量表：Likert 5级

请依次说明：
1. 数据清洗（异常值、缺失值处理）
2. 描述性统计
3. 信度分析（Cronbach's α）
4. 效度分析（KMO、Bartlett、因子分析）
5. 相关分析
6. 回归分析
7. 中介效应检验（Bootstrap）

每一步给出：SPSS 操作路径 + 结果判断标准 + 如何写到论文里'''

msg = client.messages.create(
    model='claude-opus-4-6',
    max_tokens=8000,
    messages=[{'role': 'user', 'content': prompt}]
)

def polish_paragraph(text):
    '''润色论文段落'''
    prompt = f'''请润色以下论文段落，要求：
1. 保持学术论文风格
2. 修正语法和逻辑错误
3. 提升句子流畅度
4. 不改变原文观点和意思
5. 不添加 AI 常用的套话
6. 控制修改幅度（不要大改）

原文：
{text}

请输出润色后的文本，并在最后列出主要修改点。'''

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

def reduce_similarity(text):
    '''降低论文重复率'''
    prompt = f'''以下论文段落重复率较高，请帮我降重。

降重策略（按优先级）：
1. 同义词替换（不改变专业术语）
2. 调整句子结构（主动↔被动、长句拆短句）
3. 换一种表达方式说同样的意思
4. 适当增加你的理解和分析

注意：
- 保持学术风格
- 不改变原意
- 数据和引用不要改
- 不要降到读不通

原文：
{text}'''

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