package com.ssyf.test;

import cn.hutool.http.HttpUtil;
import cn.hutool.http.HttpRequest;
import cn.hutool.http.HttpResponse;
import cn.hutool.json.JSONUtil;
import cn.hutool.core.map.MapUtil;

import java.util.*;

public class Test {

    public static void main(String[] args) {
        // 设置API密钥
        String apiKey = "af98de1f-81a0-4d21-8300-962107415e36"; // 从环境变量获取API密钥

        // 构建请求数据
        Map<String, Object> requestData = new HashMap<>();
        requestData.put("model", "doubao-seed-1-6-flash-250828");

        List<Map<String, Object>> inputList = new ArrayList<>();
        Map<String, Object> inputItem = new HashMap<>();
        inputItem.put("role", "user");

        List<Map<String, Object>> contentList = new ArrayList<>();

        // 文本内容
        Map<String, Object> textContent = new HashMap<>();
        textContent.put("type", "input_text");
        textContent.put("text", "你将扮演名为“机器猫001”的真实用户在直播间内，首先必须严格解析给定的直播间互动信息（标准格式为“用户(昵称)：发言内容”），\n" +
                "从中精准定位你自己（即“用户(机器猫001)”）的历史发言时间与数量；\n" +
                "基于此进行严格的频次检测，若判断你的发言频率过高，请直接返回“频控：不发送”，并直接终止后续流程；\n" +
                "若允许发送，请直播间互动信息的内容、情绪、话题、趋势，生成一条不超过15个字的互动信息。\n" +
                "你的发言必须低调、不抢戏、不带节奏，风格需与其他网友高度同频、接地气且真实，\n" +
                "可适度使用网络梗，但严禁包含炫富、怼人、黄赌毒内容及任何涉及你自身昵称、个性、家庭或现实时空的信息，\n" +
                "确保内容紧贴上下文且无人设崩塌（OOC）风险，绝对避免被识别为机器人。\n" +
                "【互动信息】\n" +
                "主播(赵葫芦)：可能新学的还不熟，我们再熟练一下，辛苦了谢谢你。\n" +
                "用户(好运)：这个人唱歌可以。\n" +
                "用户(青岛吴彦祖)：这小子唱得好。\n" +
                "用户(不吃鱼的猫)：好多地方没有唱对。\n" +
                "主播(赵葫芦)：喜欢小a点点关注啊。\n" +
                "主播(赵葫芦)：还有没有人了，没人四哥控制一个位置，我摇一个。\n" +
                "主播(赵葫芦)：点点赞往200万万点点赞，你们难不成想听我唱歌吗？都不点赞。\n" +
                "用户(速度速度速哥哥)：速度速度速哥哥。\n" +
                "主播(赵葫芦)：100，100就行，那你们既然这样的情况之下，那我要展示了哦。\n" +
                "主播(赵葫芦)：来纯妍你重新登录一下，你有点卡，快快快快去快去。\n" +
                "用户(速度速度)：速度速度。\n" +
                "主播(赵葫芦)：点到100就行啊兄弟们。\n" +
                "主播(赵葫芦)：什么情况，泥石流来了吗？发洪水啦。\n" +
                "用户(都市游侠)：泥石流来了吗。");
        contentList.add(textContent);

        inputItem.put("content", contentList);
        inputList.add(inputItem);
        requestData.put("input", inputList);

        try {
            // 发送POST请求
            HttpResponse response = HttpRequest.post("https://ark.cn-beijing.volces.com/api/v3/responses")
                    .header("Authorization", "Bearer " + apiKey)
                    .header("Content-Type", "application/json")
                    .body(JSONUtil.toJsonStr(requestData))
                    .timeout(30000)
                    .execute();

            // 输出响应结果
            System.out.println("状态码: " + response.getStatus());
            System.out.println("响应内容: " + response.body());

        } catch (Exception e) {
            System.err.println("请求失败: " + e.getMessage());
            e.printStackTrace();
        }
    }
}
