package prompt

import (
	"data_ai/ai"
	"data_ai/clean"
	"data_ai/ul"
	"fmt"
	qu "jygit.jydev.jianyu360.cn/data_processing/common_utils"
	"strings"
	"unicode/utf8"
)

// 获取是否为分包信息
func AcquireIsPackageInfo(detail string) bool {
	content := PromptIsPackageText(detail)
	zp := ai.PostZhiPuInfo(content)
	if qu.ObjToString(zp["分包"]) == "是" {
		return true
	}
	return false
}

// 获取标讯多包信息
func AcquireMultiplePackageInfo(detail string) map[string]interface{} {
	content := PromptMultiplePackageText(detail)
	zp := ai.PostZhiPuInfo(content)
	//后续在转格式...
	ai_pkg := map[string]interface{}{}
	s_winner, s_bidamount, s_pkg := "", 0.0, map[string]map[string]interface{}{}
	win_arr, win_temp := []string{}, map[string]string{}
	if score := qu.Float64All(zp["分包信息score"]); score >= 90.0 {
		pkginfo := ul.IsMarkInterfaceMap(zp["分包信息"])
		for k, v := range pkginfo { //
			key := fmt.Sprintf("%d", k+1)
			name := qu.ObjToString(v["标段/包号"])
			winner := clean.CleanWinner(qu.ObjToString(v["中标单位"]))
			bidamount := clean.CleanMoney((v["中标金额"]))
			//分包信息结构
			s_pkg[key] = map[string]interface{}{
				"name":      name,
				"winner":    winner,
				"bidamount": bidamount,
			}
			//去重计算单位与总金额
			s_bidamount += bidamount
			if win_temp[winner] == "" && winner != "" {
				win_arr = append(win_arr, winner)
				win_temp[winner] = winner
			}
		}
		s_winner = strings.Join(win_arr, ",")
		ai_pkg["s_winner"] = s_winner
		ai_pkg["s_bidamount"] = s_bidamount
		ai_pkg["s_pkg"] = s_pkg
	}
	return ai_pkg
}

// 分包判断-提问词
func PromptIsPackageText(detail string) string {
	if utf8.RuneCountInString(detail) > ul.MaxLen {
		detail = string([]rune(detail)[:ul.MaxLen])
	}
	content := `
你是一名"招标工程师"，拥有写标书及阅读理解公告的能力。请帮我判断下方公告中出现了几个项目：
注意：多个项目通常会以多个分包、多个标段的形式出现。业绩关联的项目要排除在外;
json形式回答，急着要结果，避免解释要干脆利落：
{
"分包":"（是/否）",
}
请回答我的问题，不要联想，不要无中生有，不要生成多余解释;
公告：` + detail + "\n结果JSON："
	return content
}

func PromptMultiplePackageText(detail string) string {
	if utf8.RuneCountInString(detail) > ul.MaxLen {
		detail = string([]rune(detail)[:ul.MaxLen])
	}
	content := `
请根据提供的招投标文件内容，精确提取并整理以下信息，形成一个便于统计的JSON：
{
"分包信息":[{
"标段/包号":（形式如下但不限于：一标段、包一、I包、I标段，不可以填写项目编号）,
"中标单位":（中标角色包括但不限于成交供应商（注：当入围供应商/中标人存在多个，选择第一位为中标单位）、中标人、中标方、承包方、中选单位、服务商、第一|1名中标候选人(忽略其他中标候选人)。当流标显示流标，废标时显示废标。联合体投标时，请列出所有单位名称使用","分割）,
"中标金额":（中标金额数值及单位，多个金额时请进行计算。非单价，如果是单价，则等于单价*数量）,
"投标折扣系数":"",
]
"包预算金额":（预算金额数值及单位）,
},
....
],
# 下方打分要求：根据抽取内容与要求匹配程度打分0-100，默认100分，1、当原文中存在，未抽取到扣分。2、原文中存在多个，选择有歧义扣分。3、原文中不存在，未抽取到不扣分
"分包信息score":,
}
请务必确保每一条记录都地囊括了各分包项目的详细信息，并按照指定的JSON格式排列数据， 对于尚未确定或未在文件中明确提供的信息，请在相应value位置填写“无”，输出结果JSON即可
请回答我的问题，不要联想，不要无中生有;
招投标文件内容：` + detail + "\n结果JSON："

	return content
}