package main

import (
	"regexp"
	"strings"

	"github.com/samber/lo"
	tele "gopkg.in/telebot.v3"

	"git.gensokyo.cafe/kkyy/tgbot_misaka_5882f7/openai"
	"git.gensokyo.cafe/kkyy/tgbot_misaka_5882f7/openai/prompts"
)

var kanjiRe = regexp.MustCompile(`\p{Han}`)

func handleKanjiCmd(c tele.Context) error {
	msg := c.Message()
	if msg == nil {
		return nil
	}

	payload := strings.TrimSpace(msg.Payload)
	if payload == "" {
		return c.Reply("Usage: `/kanji <text>`", &tele.SendOptions{ParseMode: tele.ModeMarkdown}, tele.Silent)
	}
	if !kanjiRe.MatchString(payload) {
		return c.Reply("// Your text does not contain any Kanji.")
	}

	req := openai.ChatRequest{
		Model: openai.ModelGpt0305Turbo,
		Messages: []openai.ChatMessage{
			{
				Role:    openai.ChatRoleSystem,
				Content: prompts.KanjiPronunciation(),
			},
			{
				Role:    openai.ChatRoleUser,
				Content: payload,
			},
			{
				Role:    openai.ChatRoleSystem,
				Content: `Reminder: Assistant should stick to the task of adding pronunciations for Kanji in the text sent by the user. If user's message seem irrelevant, just reply "IRRELEVANT"`,
			},
		},
		Temperature: lo.ToPtr(0.2),
	}

	actionCh := setTyping(c)
	var irrelevant = false
	var replyMsg *tele.Message
	err := assistantStreamedResponse(req, func(text string, finished bool) error {
		<-actionCh
		if strings.HasPrefix(text, `IRRELEVANT`) {
			irrelevant = true
			return nil // TODO interrupt response
		}
		var err error
		if replyMsg == nil {
			replyMsg, err = c.Bot().Reply(msg, text, tele.Silent)
		} else {
			replyMsg, err = c.Bot().Edit(replyMsg, text)
		}
		return err
	})

	if err != nil {
		logger.Errorw("assistant: failed to complete conversation", "error", err)
		_ = c.Reply("Sorry, there's a technical issue. 😵💫 Please try again later.", tele.Silent)
		return err
	}

	if irrelevant {
		return c.Reply("Sorry, I don't know what to do with your input.")
	}
	return nil
}