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_task_json(content):
    # 设置Authorization和URL
    authorization = "46a5674a-e978-491b-a810-5d54605f2c36"  # 请替换为真实的Authorization token
    url = 'http://127.0.0.1:8080/v1/chat/completions'

    prompt = "你是一个益智百科问答大师，可以随时提出百科类的问题，问题需要有一定的难度，回答完毕之后用户能有所收获，并且对问题进行打分，同时根据问题难度告知答对之后给多少分（1-10）请只返回JSON格式的内容：格式要求如下：{\"question\": \"哪个国家最早将玫瑰与爱情联系起来？\", \"score\":\"1\",  \"answer\": \"波斯\",\"description\":\"描述问题答案的原因\"}"
    # 设置请求的payload
    # prompt ="你是一个益智百科问答大师，可以根据用户回答的答案进行判断，并且对问题答案进行打分,打分时请参考最高分要求，告知用户能获得多少分，请只返回JSON格式的内容：格式要求如下：{\"question\": \"哪个国家最早将玫瑰与爱情联系起来？\", \"score\":\"1\",  \"answer\": \"波斯\",\"description\":\"描述问题答案的原因\"}"

    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)


if __name__ == '__main__':
    # print(message_task_json('question:哪个国家的节日与裸体狂欢有关？,answer:古罗马,top_score:3'))
    print(message_task_json('请出题！'))