package l_ai_category

import (
	"context"
	"encoding/json"
	"fmt"
	"gitea.cdlsxd.cn/self-tools/l_request"
	"strconv"
	"strings"
)

func GetCategoryWithSelf(ctx context.Context, goodsName string, address string) (cateId int, confidence float64, err error) {
	cateId, name, confidence, err := getCategoryWithAi(ctx, goodsName, address)
	fmt.Printf("商品：%s,所属分类：%s,置信度：%f\n", goodsName, name, confidence)
	return
}

type CatRes struct {
	Predictions []struct {
		Category   string  `json:"category"`
		Confidence float64 `json:"confidence"`
	} `json:"predictions"`
	Status string `json:"status"`
}

func getCategoryWithAi(ctx context.Context, goodsName string, address string) (cateId int, cateName string, confidence float64, err error) {
	request := map[string]interface{}{
		"product_names": goodsName,
	}
	req := l_request.Request{
		Method: "POST",
		Url:    address,
		Json:   request,
	}
	res, err := req.Send()
	if err != nil {
		return
	}
	var resMap CatRes
	err = json.Unmarshal(res.Content, &resMap)
	if err != nil {
		return
	}
	cateInfo := strings.Split(resMap.Predictions[0].Category, "_")
	if len(cateInfo) != 2 {
		cateName = ""
		cateId = 0.000000
	} else {
		cateName = cateInfo[0]
		cateId, err = strconv.Atoi(cateInfo[1])
		if err != nil {
		}
	}

	return cateId, cateName, resMap.Predictions[0].Confidence, nil
}

type Category struct {
	Id    int         `json:"id"`
	Name  string      `json:"name"`
	Level int         `json:"level"`
	Pid   int         `json:"pid"`
	Pids  map[int]int `json:"pid"`
}

type CategoryDic struct {
	Category
}
type QuesStruct struct {
	Id   int    `json:"id"`
	Name string `json:"name"`
}

type Req struct {
	Text string   `json:"text"`
	Cate []string `json:"cate"`
}