feature: artikel AI (#82)

feature: artikel AI
capstone-kelompok-7 · Dec 4, 2023 · b813c2c · b813c2c
2 parents 432c5ed + ae2ff8e
commit b813c2c
Show file tree

Hide file tree

Showing 5 changed files with 63 additions and 0 deletions.
diff --git a/module/feature/chatbot/dto/request.go b/module/feature/chatbot/dto/request.go
@@ -14,3 +14,6 @@ type ChatRequest struct {
 	Text      string `json:"text" form:"text"`
 	CreatedAt time.Time
 }
+type GenerateArtikelAiRequest struct {
+	Text string `json:"text" form:"text"`
+}
diff --git a/module/feature/chatbot/handler/handler.go b/module/feature/chatbot/handler/handler.go
@@ -3,6 +3,7 @@ package handler
 import (
 	"github.com/capstone-kelompok-7/backend-disappear/module/entities"
 	"github.com/capstone-kelompok-7/backend-disappear/module/feature/chatbot"
+	"github.com/capstone-kelompok-7/backend-disappear/module/feature/chatbot/dto"
 	"github.com/capstone-kelompok-7/backend-disappear/utils/response"
 	"github.com/labstack/echo/v4"
 )
@@ -58,3 +59,19 @@ func (h *ChatbotHandler) GetChatByIdUser() echo.HandlerFunc {
 		return response.SendSuccessResponse(c, "Berhasil mendapatkan chat by id User", chat)
 	}
 }
+
+func (h *ChatbotHandler) GenerateArtikelAi() echo.HandlerFunc {
+	return func(c echo.Context) error {
+		judulRequest := new(dto.GenerateArtikelAiRequest)
+		if err := c.Bind(judulRequest); err != nil {
+			return response.SendBadRequestResponse(c, "Format input yang Anda masukkan tidak sesuai.")
+		}
+
+		chat, err := h.service.GenerateArtikelAi(judulRequest.Text)
+		if err != nil {
+			return response.SendStatusInternalServerResponse(c, "Gagal generate artikel: "+err.Error())
+		}
+
+		return response.SendSuccessResponse(c, "Berhasil mendapatkan jawaban", chat)
+	}
+}
diff --git a/module/feature/chatbot/interface.go b/module/feature/chatbot/interface.go
@@ -19,10 +19,12 @@ type ServicChatbotInterface interface {
 	CreateQuestion(chat entities.ChatModel) error
 	CreateAnswer(chat entities.ChatModel) (string, error)
 	GetAnswerFromAi(chat []openai.ChatCompletionMessage, ctx context.Context) (openai.ChatCompletionResponse, error)
+	GenerateArtikelAi(judul string) (string, error)
 }
 
 type HandlerChatbotInterface interface {
 	GetChatByIdUser() echo.HandlerFunc
 	CreateQuestion() echo.HandlerFunc
 	CreateAnswer() echo.HandlerFunc
+	GenerateArtikelAi() echo.HandlerFunc
 }
diff --git a/module/feature/chatbot/service/service.go b/module/feature/chatbot/service/service.go
@@ -112,3 +112,43 @@ func (s *ChatbotService) GetChatByIdUser(id string) ([]entities.ChatModel, error
 	}
 	return res, nil
 }
+
+func (s *ChatbotService) GenerateArtikelAi(judul string) (string, error) {
+	ctx := context.Background()
+	chat := []openai.ChatCompletionMessage{
+		{
+			Role:    openai.ChatMessageRoleUser,
+			Content: "Kamu Adalah Chatbot Bertema Lingkungan",
+		},
+		{
+			Role:    openai.ChatMessageRoleUser,
+			Content: "Hi Bisa Bantu Saya Menjawab Pertanyaan Tentang Lingkungan",
+		},
+	}
+
+	if judul != "" {
+		chat = append(chat, openai.ChatCompletionMessage{
+			Role:    openai.ChatMessageRoleUser,
+			Content: fmt.Sprintf("Buatlah Artikel Dengan Judul %s", judul),
+		})
+	}
+
+	resp, err := s.GetAnswerFromAi(chat, ctx)
+	if err != nil {
+		logrus.Error("Can't Get Answer From Ai: ", err.Error())
+	}
+
+	if s.debug {
+		fmt.Printf(
+			"ID: %s. Created: %d. Model: %s. Choices: %v.\n",
+			resp.ID, resp.Created, resp.Model, resp.Choices,
+		)
+	}
+
+	answer := openai.ChatCompletionMessage{
+		Role:    resp.Choices[0].Message.Role,
+		Content: resp.Choices[0].Message.Content,
+	}
+
+	return answer.Content, nil
+}
diff --git a/routes/routes.go b/routes/routes.go
@@ -148,6 +148,7 @@ func RouteChatbot(e *echo.Echo, h chatbot.HandlerChatbotInterface, jwtService ut
 	chatbotGroup.POST("/question", h.CreateQuestion(), middlewares.AuthMiddleware(jwtService, userService))
 	chatbotGroup.POST("/answer", h.CreateAnswer(), middlewares.AuthMiddleware(jwtService, userService))
 	chatbotGroup.GET("/:id", h.GetChatByIdUser(), middlewares.AuthMiddleware(jwtService, userService))
+	chatbotGroup.POST("/generate", h.GenerateArtikelAi(), middlewares.AuthMiddleware(jwtService, userService))
 }
 
 func RouteDashboard(e *echo.Echo, h dashboard.HandlerDashboardInterface, jwtService utils.JWTInterface, userService users.ServiceUserInterface) {