package ai

import (
	"bytes"
	"data_ai/ul"
	"encoding/json"
	log "github.com/donnie4w/go-logger/logger"
	"io/ioutil"
	qu "jygit.jydev.jianyu360.cn/data_processing/common_utils"
	"net/http"
	"strings"
)

// 智谱清言
func PostZhiPuAI(content string) map[string]interface{} {
	// API的URL
	apiURL := "https://open.bigmodel.cn/api/paas/v4/chat/completions"
	// 构造请求数据
	messages := []map[string]interface{}{}
	messages = append(messages, map[string]interface{}{
		"role":    "user",
		"content": content,
	})
	//glm-4-air	glm-4-0520  glm-4-flash
	requestData := map[string]interface{}{
		"model":       "glm-4-flash",
		"messages":    messages,
		"temperature": 0.1,
		"max_tokens":  4096,
	}
	jsonData, _ := json.Marshal(requestData)
	// 创建HTTP请求
	req, err := http.NewRequest("POST", apiURL, bytes.NewBuffer(jsonData))
	if err != nil {
		log.Debug("Error: %s", err)
		return map[string]interface{}{}
	}
	// 设置请求头
	req.Header.Set("Content-Type", "application/json")
	// 如果API需要认证，可以在这里设置认证信息
	req.Header.Set("Authorization", "Bearer 3d84d30b7ab4c94dbf71853cb7e44719.hLLS4CA2MqVQs6kR")
	// 发起请求 14543f0d69d6987c8782fd846e164f26.DXaoS1axLaMP892a
	client := &http.Client{}
	resp, err := client.Do(req)
	if err != nil {
		log.Debug("Error: %s", err)
		return map[string]interface{}{}
	}
	defer resp.Body.Close()

	// 解析响应
	body, _ := ioutil.ReadAll(resp.Body)
	res := make(map[string]interface{})
	json.Unmarshal(body, &res)
	if res != nil {
		if choices := ul.IsMarkInterfaceMap(res["choices"]); len(choices) > 0 {
			if message := qu.ObjToMap(choices[0]["message"]); message != nil {
				result := qu.ObjToString((*message)["content"])
				result = strings.ReplaceAll(result, "\n", "")
				result = strings.ReplaceAll(result, "json", "")
				result = strings.ReplaceAll(result, "`", "")
				if new_result := ul.SaveResultReg.FindString(result); new_result != "" {
					result = new_result
				}
				dict := make(map[string]interface{})
				json.Unmarshal([]byte(result), &dict)
				return dict
			}
		}
	}
	return map[string]interface{}{}
}

// 智谱清言-分类字段
func PostClassZhiPuAI(content string) map[string]interface{} {
	// API的URL
	apiURL := "https://open.bigmodel.cn/api/paas/v4/chat/completions"
	// 构造请求数据
	messages := []map[string]interface{}{}
	messages = append(messages, map[string]interface{}{
		"role":    "user",
		"content": content,
	})
	//glm-4-air	glm-4-0520  glm-4-flash
	requestData := map[string]interface{}{
		"model":       "glm-4-flash",
		"messages":    messages,
		"temperature": 0.2,
		"top_p":       0.7,
		//"max_tokens":  4096,
	}
	jsonData, _ := json.Marshal(requestData)
	// 创建HTTP请求
	req, err := http.NewRequest("POST", apiURL, bytes.NewBuffer(jsonData))
	if err != nil {
		log.Debug("Error: %s", err)
		return map[string]interface{}{}
	}
	// 设置请求头
	req.Header.Set("Content-Type", "application/json")
	// 如果API需要认证，可以在这里设置认证信息
	req.Header.Set("Authorization", "Bearer 3d84d30b7ab4c94dbf71853cb7e44719.hLLS4CA2MqVQs6kR")
	// 发起请求 14543f0d69d6987c8782fd846e164f26.DXaoS1axLaMP892a
	client := &http.Client{}
	resp, err := client.Do(req)
	if err != nil {
		log.Debug("Error: %s", err)
		return map[string]interface{}{}
	}
	defer resp.Body.Close()

	// 解析响应
	body, _ := ioutil.ReadAll(resp.Body)
	res := make(map[string]interface{})
	json.Unmarshal(body, &res)
	if res != nil {
		if choices := ul.IsMarkInterfaceMap(res["choices"]); len(choices) > 0 {
			if message := qu.ObjToMap(choices[0]["message"]); message != nil {
				result := qu.ObjToString((*message)["content"])
				result = strings.ReplaceAll(result, "\n", "")
				result = strings.ReplaceAll(result, "json", "")
				result = strings.ReplaceAll(result, "`", "")
				if new_result := ul.SaveResultReg.FindString(result); new_result != "" {
					result = new_result
				}
				dict := make(map[string]interface{})
				json.Unmarshal([]byte(result), &dict)
				return dict
			}
		}
	}
	return map[string]interface{}{}
}