package main

import (
	"fmt"
	"github.com/otiai10/gosseract/v2"
	"github.com/ssgo/log"
	"github.com/ssgo/u"
	"strings"
)

func checkImg(imgStr string) error {
	text := ocrImage(u.UnBase64(strings.Split(imgStr, "base64,")[1]))
	log.DefaultLogger.Info("checkImg", "imgStrLen", len(imgStr), "imgLen", len(u.UnBase64(strings.Split(imgStr, "base64,")[1])), "text", text)
	score, err := parseRiskScore(ask(buildImgDetectionPrompt(text)))
	if err != nil {
		log.DefaultLogger.Error("parseImgResp error", "err", err)
		return err
	}
	err = conn.WriteJSON(map[string]any{
		"type": "imageResult",
		"result": map[string]any{
			"score":        score,
			"isSuspicious": score >= 50,
		},
	})
	if err != nil {
		log.DefaultLogger.Error("sendImgScore Error", "err", err)
		return err
	}
	log.DefaultLogger.Info("sendImgScore")
	return nil
}

func ocrImage(image []byte) string {
	client := gosseract.NewClient()
	defer client.Close()
	err := client.SetImageFromBytes(image)
	if err != nil {
		fmt.Println(err)
		log.DefaultLogger.Error("ocr setimg error", "err", err)
		return ""
	}
	fmt.Println(u.Red("111"))
	text, err := client.Text()
	fmt.Println(text)
	if err != nil {
		fmt.Println(err)
		log.DefaultLogger.Error("ocr error", "err", err)
		return ""
	}
	return text
}

func buildImgDetectionPrompt(content string) string {
	return fmt.Sprintf(`你是一名反诈骗专家，专门分析聊天记录，判断其中是否涉及诈骗信息。请仔细阅读以下文字，并回答以下问题：

1. 这段内容是否包含诈骗信息？如果是，请给出具体的诈骗类型（如刷单、冒充客服、杀猪盘等）。
2. 请分析这段内容是否具有诈骗话术，例如使用了“限时优惠”“保证赚钱”“不透露给别人”等常见诈骗用语。
3. 诈骗信息通常会使用哪些伎俩欺骗受害者？请指出可能的欺诈手法。
4. 请用简洁的语言（不超过100字）总结你的分析结果，并给出风险等级（低、中、高）。
5. 如果有必要，请建议用户如何避免被骗。

请返回JSON格式：
{
    "risk_score": 0-100，
    "risk_reason": "主要风险点"
}

待分析的文本如下：%s`, content)
}