import requests
import json


# 解析JSON
def extract_content(data_string):
    try:
        data = json.loads(data_string)
        # 提取content字段
        content = data["choices"][0]["message"].get("content", "")
        return content
    except json.JSONDecodeError:
        print("Invalid JSON")
        return None


def message_summary(content):
    # 设置Authorization和URL
    authorization = "46a5674a-e978-491b-a810-5d54605f2c36"  # 请替换为真实的Authorization token
    url = 'http://127.0.0.1:8080/v1/chat/completions'

    prompt = ('你是一个聊天记录总结助手，请使用如下格式进行聊天记录整理（如果存在多个主题则每个主题分开显示相关的参与者时间点，内容摘要）：'
              '[聊天主题] ☆☆☆'
              '参与者：[列出参与者名字]'
              '时间点：[开始时间] - [结束时间]'
              '内容摘要： [简要总结讨论的内容，包括讨论的主题、观点、信息等。]'
              '☑ 点评： [对讨论内容的简要评价，结合实际情况分析其意义或影响。]'
              '· 趣味互动'
              '[列出群成员之间互动的有趣部分，例如个人经验、笑点或特别的观点分享等。]'
              '· 待跟进事项'
              '[列出具体的行动项或下一步的讨论计划，包括需要落实的具体任务。]'
              '· 结语'
              '[总结今日讨论的主题和氛围，简要表达对未来讨论的期待。鼓励群成员继续参与。]'
              '注： [对本次讨论的特殊情况进行说明，例如讨论话题较简单或较复杂等。]')
    # 设置请求的payload
    data = {
        # "stream": True,
        "model": "windsurf/gpt4o",
        "messages": [
            {
                "role": "system",
                "content": f"{prompt}"
            },
            {
                "role": "user",
                "content": f"{content}"
            }

        ]
    }

    # 设置请求头
    headers = {
        "Content-Type": "application/json; charset=utf-8",
        "Authorization": authorization
    }

    # 发送POST请求
    response = requests.post(url, headers=headers, data=json.dumps(data), )
    response.encoding = 'utf-8'

    # 输出响应内容
    print(response.status_code)
    print(response.text)
    return extract_content(response.text)