package main

import (
	"context"
	"github.com/gin-gonic/gin"
	"github.com/sashabaranov/go-openai"
	"os"
)

func main() {
	// get token
	token := os.Getenv("OPEN_AI_KEY")
	r := gin.Default()
	// /getAns?msg=foobar
	r.GET("/getAns", func(c *gin.Context) {
		msg := c.Query("msg")
		ans := GetChatGPTAns(token, msg)
		c.JSON(200, gin.H{
			"answer": ans,
		})
	})
	err := r.Run()
	if err != nil {
		return
	}
}

func GetChatGPTAns(token string, msg string) string {
	c := openai.NewClient(token)
	resp, err := c.CreateChatCompletion(
		context.Background(),
		openai.ChatCompletionRequest{
			Model: openai.GPT3Dot5Turbo,
			Messages: []openai.ChatCompletionMessage{
				{
					Role:    openai.ChatMessageRoleUser,
					Content: msg,
				},
			},
		},
	)
	if err != nil {
		return ""
	}
	return resp.Choices[0].Message.Content
}