updates

controller/openai.go

@@ -2,30 +2,83 @@ package controller
 
 import (
 	"context"
+	"fmt"
 	"net/http"
+	"os"
 
 	"github.com/gin-gonic/gin"
+	"github.com/joho/godotenv"
 	openai "github.com/sashabaranov/go-openai"
 )
 
-var client = openai.NewClient("")
+var (
+	// Declare global variables
+	apiKey string
+
+	config *configStruct
+)
+
+type configStruct struct {
+	apiKey string `json : "API_KEY"`
+}
+
+func init() {
+	// Load values for global variables from environment variables
+	err := godotenv.Load(".env")
+
+	if err != nil {
+		fmt.Println(err.Error())
+		//return err
+	}
+	apiKey = os.Getenv("API_KEY")
+
+}
 
 // GeneralOpenAI godoc
 //
-//	@Summary		ping example
-//	@Description	do ping
+//	@Summary		Gerneral ChatGPT
+//	@Description	Ask ChatGPT a general question
 //	@Tags			openai
 //	@Accept			json
 //	@Produce		plain
 //	@Param			message	query		string		true	"Ask ChatGPT a general question"
 //	@Success		200	{string}	string	"response"
-//	@Failure		400	{string}	string	"ok"
-//	@Failure		404	{string}	string	"ok"
-//	@Failure		500	{string}	string	"ok"
-//	@Router			/openai/GeneralOpenAI [get]
+//	@Router			/openai/general [get]
 func (c *Controller) GeneralOpenAI(ctx *gin.Context) {
 	message := ctx.Query("message")
 
+	result, err := createChatCompletion(message)
+	if err != nil {
+		ctx.AbortWithError(http.StatusInternalServerError, err)
+	}
+
+	ctx.String(http.StatusOK, result)
+}
+
+// TravelAgentOpenAI godoc
+//
+//	@Summary		Travel Agent ChatGPT
+//	@Description	Ask ChatGPT for suggestions as if it was a travel agent
+//	@Tags			openai
+//	@Accept			json
+//	@Produce		plain
+//	@Param			message	query		string		true	"Ask ChatGPT for suggestions as a travel agent"
+//	@Success		200	{string}	string	"response"
+//	@Router			/openai/travelagent [get]
+func (c *Controller) TravelAgentOpenAI(ctx *gin.Context) {
+	message := "want you to act as a travel guide. I will give you my location and you will give me suggestions " + ctx.Query("message")
+
+	result, err := createChatCompletion(message)
+	if err != nil {
+		ctx.AbortWithError(http.StatusInternalServerError, err)
+	}
+
+	ctx.String(http.StatusOK, result)
+}
+
+func createChatCompletion(message string) (string, error) {
+
+	var client = openai.NewClient(apiKey)
 	resp, err := client.CreateChatCompletion(
 		context.Background(),
 		openai.ChatCompletionRequest{
@@ -39,8 +92,8 @@ func (c *Controller) GeneralOpenAI(ctx *gin.Context) {
 		},
 	)
 	if err != nil {
-		ctx.AbortWithError(http.StatusInternalServerError, err)
+		return "", err
 	}
 
-	ctx.String(http.StatusOK, resp.Choices[0].Message.Content)
+	return resp.Choices[0].Message.Content, nil
 }