import requests
import json
import time

# DeepSeek API配置
DEEPSEEK_API_URL = "https://api.deepseek.com/v1/chat/completions"
API_KEY = "sk-9d1469d346994d87ada1df0bcae28bc4"  # 您提供的API密钥

# 定义不同的回复风格
REPLY_STYLES = {
    "positive": "积极热情",
    "humorous": "幽默风趣",
    "polite": "礼貌得体",
    "empathetic": "共情理解",
    "professional": "专业理性"
}

def ask_deepseek(prompt, style="positive"):
    """发送提示到DeepSeek API并获取回复"""
    headers = {
        "Content-Type": "application/json",
        "Authorization": f"Bearer {API_KEY}"
    }
    
    data = {
        "model": "deepseek-chat",
        "messages": [
            {
                "role": "system",
                "content": f"""你是一个高情商聊天助手，擅长提供{REPLY_STYLES[style]}的回复。
                请分析用户输入的消息，并提供一个适合微信聊天场景的高情商回复。
                回复应简洁明了，符合中文表达习惯，并且能够体现出良好的沟通技巧和情感智慧。"""
            },
            {
                "role": "user",
                "content": prompt
            }
        ],
        "temperature": 0.7,
        "max_tokens": 256
    }
    
    try:
        response = requests.post(DEEPSEEK_API_URL, headers=headers, json=data)
        response.raise_for_status()  # 检查请求是否成功
        
        result = response.json()
        return result["choices"][0]["message"]["content"].strip()
    except Exception as e:
        return f"发生错误: {str(e)}"

def generate_multi_style_replies(message, styles=None):
    """为一条消息生成多种风格的回复"""
    if styles is None:
        styles = list(REPLY_STYLES.keys())
    
    replies = {}
    for style in styles:
        print(f"正在生成{REPLY_STYLES[style]}风格的回复...")
        reply = ask_deepseek(message, style)
        replies[style] = reply
        time.sleep(1)  # 避免API请求过于频繁
    
    return replies

def main():
    print("=" * 50)
    print("DeepSeek-R1高情商AI微信聊天回复助手")
    print("=" * 50)
    
    while True:
        print("\n1. 输入对方消息获取高情商回复")
        print("2. 退出")
        
        choice = input("\n请选择操作 [1-2]: ")
        
        if choice == '1':
            message = input("\n请输入对方的微信消息 (输入'q'返回): ")
            if message.lower() == 'q':
                continue
            
            if not message.strip():
                print("消息不能为空")
                continue
            
            # 生成多种风格的回复
            replies = generate_multi_style_replies(message)
            
            # 显示回复
            print("\n" + "=" * 40)
            print("高情商回复建议:")
            print("=" * 40)
            for style, reply in replies.items():
                print(f"\n[{REPLY_STYLES[style]}]")
                print(reply)
        
        elif choice == '2':
            print("\n感谢使用DeepSeek-R1高情商AI微信聊天回复助手，再见!")
            break
        
        else:
            print("\n无效的选择，请重新输入")

if __name__ == "__main__":
    main()