package prompt

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

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

var pmt_pkg_1 = `
你是一名’招标工程师’，拥有写标书及阅读理解公告的能力，根据要求抽取所需的内容，抽取内容要实事求是，不会无中生有。
公告start:
{` + `%s` + `}
公告end

任务如下：
第一步、公告预处理：
将上述公告中信息拆分为两部分
标的信息：标的名称、标的编号、规格型号、数量等相关信息
其他信息：去除上述"标的信息"剩余文本内容，执行第二步、第三步、第四步
不输出内容，仅做思考，完成后输出："已拆分完成"即可

第二步、请根据第一步拆解分后的"其他信息"，精确提取并整理以下信息，形成一个便于统计的JSON：
注意：入围供应商、入围中标人、中标候选人存在多个时，选择第一名为中标单位，忽略其他排名。请务必确保每一条记录都来源于"公告内容"， 对于尚未确定或未在文中明确提供的信息，请在相应value位置填写“无”
{
"分包信息":[
{
"标段名称":（标包名称，通常包含地理区域、专业类别、标包内容描述等内容，不可以填写"标的名称"）,
"标段/包号":（标包编号，可以来自标段名称，如果不存在写"无"，比如：一标段、二标段、包一、I包、I标段、标包一、标包编号等，不可以填写"项目编号"或"标的编号"）,
"中标单位":（中标单位名称，要求：中标角色包括但不限于成交供应商（注：当入围供应商/中标人存在多个，选择第一位为中标单位）、中标人、中标方、承包方、中选单位、服务商、第一|1名中标候选人(忽略其他中标候选人)。当流标显示流标，废标时显示废标。联合体投标时，请列出所有单位名称使用","分割）,
"中标金额":（中标金额数值及单位，要求：不能使用预算金额。多个金额时请进行计算，非单价，如果是单价，则等于单价*数量。）,
"预算金额":（预算金额数值及单位，要求：不可以用中标金额来填充）,
"投标折扣系数":""
},
....
],
}

第三步、审查第一步输出的结果，
0.检查"标段名称"及"标段/包号"是否真的在原文中出现，
1.检查"标段名称"及"标段/包号"是否有明确提及，如果只有标的名称明确提及，请删除
2.检查中标金额是否在原文中明确提及，否则删除第一步的中标金额结果
3.检查中标金额与预算金额中的数值与单位是否正确
4.修正答案

第四步、根据第一、二、三步结果总结得到最终正确的结果

答：
第一步结果：

第二步结果：

第三步验证：

最终正确的结果：`

var pmt_pkg_2 = `
你是一名’招标工程师’，拥有写标书及阅读理解公告的能力，根据要求抽取所需的内容，抽取内容要实事求是，不会无中生有。
公告start:
{` + `%s` + `}
公告end

任务如下：
第一步、请根据公告内容，精确提取并整理以下信息，形成一个便于统计的JSON：
注意：入围供应商、入围中标人、中标候选人存在多个时，选择第一名为中标单位，忽略其他排名。请务必确保每一条记录都来源于"公告内容"， 对于尚未确定或未在文中明确提供的信息，请在相应value位置填写“无”
{
"分包信息":[
{
"标段名称":（标包名称，通常包含地理区域、专业类别、标包内容描述等内容，不可以填写"标的名称"）,
"标段/包号":（标包编号，可以来自标段名称，如果不存在写"无"，比如：一标段、二标段、包一、I包、I标段、标包一、标包编号等，不可以填写"项目编号"或"标的编号"）,
"中标单位":（中标单位名称，要求：中标角色包括但不限于成交供应商（注：当入围供应商/中标人存在多个，选择第一位为中标单位）、中标人、中标方、承包方、中选单位、服务商、第一|1名中标候选人(忽略其他中标候选人)。当流标显示流标，废标时显示废标。联合体投标时，请列出所有单位名称使用","分割）,
"中标金额":（中标金额数值及单位，如果公告中没有明确说明，输出"无"，不能使用预算金额。）,
"投标折扣系数":"",
"包预算金额":（预算金额数值及单位，要求：不可以用中标金额来填充）
},
....
],
}

第二步、审查第一步输出的结果，
0.检查"标段名称"及"标段/包号"是否真的在原文中出现，
1.检查"标段名称"及"标段/包号"是否有明确提及，如果只有标的名称明确提及，请删除
2.检查"中标金额"是否在原文中明确提及，否则删除第一步的中标金额结果
3.检查中标金额与预算金额中的数值与单位是否正确
4.修正答案

第三步、根据第一、二步结果总结得到最终正确的结果

答：
第一步结果：

第二步结果：

最终正确的结果：`

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

// 获取标讯多包信息
func AcquireMultiplePackageInfo(detail string) map[string]interface{} {
	content := PromptMultiplePackageText(detail)
	zp := ai.PostZhiPuPackageInfo(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{}
	pkginfo := ul.IsMarkInterfaceMap(zp["分包信息"])
	for k, v := range pkginfo { //
		key := fmt.Sprintf("%d", k+1)
		name := qu.ObjToString(v["标段名称"])
		code := qu.ObjToString(v["标段/包号"])
		winner := clean.CleanWinner(qu.ObjToString(v["中标单位"]))
		bidamount := clean.CleanMoney([]interface{}{v["中标金额"], ""})
		//分包信息结构
		s_pkg[key] = map[string]interface{}{
			"name":      name,
			"code":      code,
			"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 {
	content := fmt.Sprintf(pmt_pkg_0, detail)
	l := len(content)
	if l > ul.MaxByte {
		detail = truncatedText(detail, len(detail)-(l-ul.MaxByte))
		content = fmt.Sprintf(pmt_pkg_0, detail)
	}
	return content
}

// 分包信息提取
func PromptMultiplePackageText(detail string) string {
	pmt_pkg := ""
	if strings.Contains(detail, "标的名称") || strings.Contains(detail, "标的编号") {
		pmt_pkg = pmt_pkg_1
	} else {
		pmt_pkg = pmt_pkg_2
	}
	content := fmt.Sprintf(pmt_pkg, detail)
	l := len(content)
	if l > ul.MaxByte {
		detail = truncatedText(detail, len(detail)-(l-ul.MaxByte))
		content = fmt.Sprintf(pmt_pkg, detail)
	}
	return content
}