package com.tzld.piaoquan.longarticle.utils;

import cn.hutool.core.convert.ConvertException;
import cn.hutool.http.HttpException;
import cn.hutool.http.HttpRequest;
import com.alibaba.fastjson.JSONArray;
import com.alibaba.fastjson.JSONObject;

import java.util.*;

public class OpenAIAPI {

    /**
     * 聊天端点
     */
    static String chatEndpoint = "https://api.openai.com/v1/chat/completions";
    /**
     * api密匙
     */
    static String apiKey = "Bearer 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";

    /**
     * 发送消息
     *
     * @param txt 内容
     * @return {@link String}
     */
    public static String chat(String txt) {
        Map<String, Object> paramMap = new HashMap<>();
        paramMap.put("model", "gpt-3.5-turbo");
        List<Map<String, String>> dataList = new ArrayList<>();
        dataList.add(new HashMap<String, String>() {{
            put("role", "user");
            put("content", txt);
        }});
        paramMap.put("messages", dataList);
        JSONObject message = null;
        try {
            String body = HttpRequest.post(chatEndpoint)
                    .header("Authorization", apiKey)
                    .header("Content-Type", "application/json")
                    .body(JSONObject.toJSONString(paramMap))
                    .execute()
                    .body();
            JSONObject jsonObject = JSONObject.parseObject(body);
            JSONArray choices = jsonObject.getJSONArray("choices");
            JSONObject result = choices.getJSONObject(0);
            message = result.getJSONObject("message");
        } catch (HttpException e) {
            return "出现了异常";
        } catch (ConvertException e) {
            return "出现了异常";
        }
        return message.getString("content");
    }

    public static String promptGenerate(String promptType) {
        if (Objects.equals(promptType, "title")) {
            return " 请将以上标题改写成适合小程序点击和传播的小程序标题，小程序标题的写作规范如下，请学习后进行小程序标题的编写。直接输出最终的小程序标题\n" +
                    "                小程序标题写作规范：\n" +
                    "                1.要点前置：将最重要的信息放在标题的最前面，以快速吸引读者的注意力。例如，“5月一辈子同学，三辈子亲，送给我的老同学，听哭无数人！”中的“5月”和“一辈子同学，三辈子亲”都是重要的信息点。\n" +
                    "                2.激发情绪：使用能够触动人心的语言，激发读者的情感共鸣。如“只剩两人同学聚会，看后感动落泪。”使用“感动落泪”激发读者的同情和怀旧情绪。\n" +
                    "                3.使用数字和特殊符号：数字可以提供具体性，而特殊符号如“\uD83D\uDD34”、“\uD83D\uDE04”、“\uD83D\uDD25”等可以吸引视觉注意力，增加点击率。\n" +
                    "                4.悬念和好奇心：创建悬念或提出问题，激发读者的好奇心。例如，“太神奇了!长江水位下降，重庆出现惊奇一幕!”中的“惊奇一幕”就是一个悬念。\n" +
                    "                5.名人效应：如果内容与知名人士相关，提及他们的名字可以增加标题的吸引力。\n" +
                    "                6.社会价值观：触及读者的文化和社会价值观，如家庭、友情、国家荣誉等。\n" +
                    "                7.标点符号的运用：使用感叹号、问号等标点来增强语气和情感表达。\n" +
                    "                8.直接的语言：使用直白、口语化的语言，易于理解，如“狗屁股，笑死我了!”。\n" +
                    "                9.热点人物或事件：提及当前的热点人物或事件，利用热点效应吸引读者。\n" +
                    "                10.字数适中：保持标题在10-20个字之间，既不过长也不过短，确保信息的完整性和吸引力。\n" +
                    "                11.适当的紧迫感：使用“最新”、“首次”、“紧急”等词汇，创造一种紧迫感，促使读者立即行动。\n" +
                    "                12.情感或价值诉求：使用如“感动”、“泪目”、“经典”等词汇，直接与读者的情感或价值观产生共鸣。\n" +
                    "                避免误导：确保标题准确反映内容，避免夸大或误导读者。";

        } else if (Objects.equals(promptType, "text")) {
            return "请从我给你的文章中挖掘出以下信息并且返回如下结果。\n" +
                    "                你返回的结果是一个 json, 格式如下:\n" +
                    "                {\n" +
                    "                    \"content_keys\": [] # 同时提供三个与文章内容高度相关的关键词，这些关键词将用于网络上搜索相关视频内容,\n" +
                    "                    \"content_title\": 一个总结性的标题，该标题应简洁并能够反映文章的主要内容\n" +
                    "                }\n" +
                    "                你需要处理的文本是：";
        } else {
            return "用中文帮我回答以下问题";
        }
    }

}