71 lines
2.1 KiB
Go
71 lines
2.1 KiB
Go
|
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)
|
|||
|
|
}
|