package main

import (
	"data_ai/ai"
	"data_ai/extract"
	"data_ai/tool"
	"data_ai/udp"
	"data_ai/ul"
	"fmt"
	log "github.com/donnie4w/go-logger/logger"
	"net/http"
)

func init() {
	ul.IsLocal = false   //是否本地
	ul.IsService = false //是否服务
	ul.IsTool = true     //是否工具
	ul.IsFull = false    //是否全量
	ul.IsFlash = true    //是否flash-线上增量开关
	if ul.IsTool {
		ul.InitToolVar()
	} else {
		if ul.IsService {
			tool.InitCleanDeepSeekService()
			go http.ListenAndServe(":12321", nil)
			return
		}
		ul.InitGlobalVar()
		if !ul.IsFull {
			if ul.IsFlash {
				udp.InitProcessVar()
			} else {
				if ul.ModelType == "deepseek" {
					go extract.RunDeepSeek()
				}
			}
		}
	}

	log.Debug("测试一下转换MD方法···")
	detail := `<table border=\"1\"><tbody><tr><th>采购人名称</th><td>云南中医药大学</td></tr><tr><th>中标（成交）供应商名称</th><td>云南智蓝云鸽信息科技有限公司</td></tr><tr><th>合同金额</th><td>224.99万元 人民币</td></tr><tr><th>合同期限</th><td>年</td></tr><tr><th>合同签署时间</th><td>2023-12-29 00:00:00</td></tr></tbody></table>`
	new_detail := ul.HttpConvertToMarkdown(detail)
	log.Debug(new_detail)
}

func main() {
	if ul.IsTool {
		tool.StartToolInfo()
		return
	}
	if ul.IsFull {
		extract.ExtractFullInfo("6776b8000000000000000000")
		return
	}

	lock := make(chan bool)
	<-lock
}

func ttt() {

	var pmt = `
你是一个文本处理模型，根据下述信息做"项目分类"：
"项目分类"参考资料：
	1、水-第三方综合服务：[地表水精细化管理服务、河流水质保障第三方服务];
	2、水-监测、检测及运维服务：[排口监测、水质监测站运维、农村污水设施运维、水源地水质监测、断面监测];
	3、水-设备销售：[排口规范化建设、水质监测设备];
	4、水-智慧环保平台：[排污口监管系统、管网系统、水质监测系统];
	5、水-工程建设：[水源地整治工程、管网建设、农村污水治理工程、水环境工程、水生态工程、雨水管网工程、排口整治];
	6、水-专项咨询服务：[排口排查溯源、水源综合整治、农村污水治理、美丽河湖方案、污水排水综合规划、水生态技术服务、地下水污染防治技术服务、一河一策方案编制、排污许可];
	7、大气-第三方综合服务：[驻场服务、精细化、大气污染第三方服务、空气质量达标管控服务、抑尘服务];
	8、大气-监测、检测及运维服务：[网格监测、普通监测、VOCs检测、移动排放监测、机动车排放监测、监测设备运维、异味监测、空气自动监测站运维、尾气监测、加油站油气排放监测、餐饮油烟监测];
	9、大气-设备销售：[监测能力建设、监测能力提升];
	10、大气-智慧环保平台：[监控系统平台、大气管理系统];
	11、大气-专项咨询服务：[污染源评估、污染源溯源、排放清单、重污染天气应急预案、大气污染技术咨询、秸秆焚烧治理、温室气体融合清单、异味治理、大气污染源解析、一点一策];
	12、固废-监测、检测及运维服务：[垃圾填埋场监测];
	13、固废-智慧环保平台：[固废相关系统平台];
	14、固废-专项咨询服务：[矿渣治理、无废城市建设评价、固废污染源整治设计、固废技术审查、固废治理效果评估];
	15、土壤-监测、检测及运维服务：[土壤监测、地块环境监测];
	16、土壤-设备销售：[土壤监测设备];
	17、土壤-智慧环保平台：[土壤监测相关系统平台];
	18、土壤-专项咨询服务：[土壤污染状况风险评估、土壤修复评估、土壤金属污染综合治理、土壤重金属污染溯源];
	19、噪声-监测、检测及运维服务：[声环境监测、噪声监测、交通环境噪声监测];
	20、噪声-设备销售：[噪声监测设备];
	21、噪声-智慧环保平台：[噪声监测平台];
	22、噪声-专项咨询服务：[声环境功能区调整划分方案编制、噪声现状分析研究];
	23、其他-第三方综合服务：[生态环境局第三方驻场服务、园区环保管家、园区污染第三方治理服务];
	24、其他-监测、检测及运维服务：[生物多样性监测、生态环境质量监测、农村环境监测];
	25、其他-设备销售：[智慧园区、环境治理系统];
	26、其他-智慧环保平台：[园区环保系统平台];
	27、其他-专项咨询服务：[农村环境整治、工业园区环境整治、清洁生产审核、两山基地实施方案、环评、双碳];
要求：根据正文内容，仅从"项目分类"中选取一个合适的分类。在识别过程中出现多个结果时，选取一个最贴切的，并将项目分类识别结果严格按照下述JSON格式输出。
JSON格式:
{
"项目分类":"项目分类",
}
按照以上要求输出，不要联想，不要无中生有，不要生成解释，对于尚未确定或未明确的信息请在JSON对应的值填写：无
正文内容：` + `%s` + "\n" + `
结果JSON：`
	title := "剑鱼园区的噪声监测系统"
	content := fmt.Sprintf(pmt, title)
	data := ai.PostZhiPuAI(content)
	log.Debug(data)
}