package web

import (
	"catface/app/global/consts"
	"catface/app/global/errcode"
	"catface/app/global/variable"
	"catface/app/model_es"
	"catface/app/service/nlp"
	"catface/app/utils/response"

	"github.com/gin-gonic/gin"
	"go.uber.org/zap"
)

type Rag struct {
}

func (r *Rag) Chat(context *gin.Context) {
	// 1. query embedding
	query := context.GetString(consts.ValidatorPrefix + "query")
	embedding, ok := nlp.GetEmbedding(query)
	if !ok {
		code := errcode.ErrPythonService
		response.Fail(context, code, errcode.ErrMsg[code], "")
		return
	}

	// 2. ES TopK
	docs, err := model_es.CreateDocESFactory().TopK(embedding, 1)
	if err != nil || len(docs) == 0 {
		variable.ZapLog.Error("ES TopK error", zap.Error(err))

		code := errcode.ErrNoDocFound
		response.Fail(context, code, errcode.ErrMsg[code], errcode.ErrMsgForUser[code])
	}

	// 3. LLM answer
	if answer, err := nlp.ChatKnoledgeRAG(docs[0].Content, query); err == nil {
		response.Success(context, consts.CurdStatusOkMsg, gin.H{
			"answer": answer,
		})
	} else {
		response.Fail(context, consts.CurdStatusOkCode, consts.CurdStatusOkMsg, "")
	}
}

func (r *Rag) HelpDetectCat(context *gin.Context) {
	// TODO
}