Test

.gitignore

@@ -40,4 +40,5 @@ out/
 /src/main/resources/database/**
 /src/main/resources/oauth2/**
 /src/main/resources/s3/**
+/src/main/resources/chatgpt/**
 

build.gradle

@@ -41,6 +41,9 @@ dependencies {
     runtimeOnly group: 'io.jsonwebtoken', name: 'jjwt-impl', version: '0.11.5'
     runtimeOnly group: 'io.jsonwebtoken', name: 'jjwt-jackson', version: '0.11.5'
 
+    // GPT
+    implementation 'io.github.flashvayne:chatgpt-spring-boot-starter:1.0.4'
+
     // Amazon S3
     implementation platform('com.amazonaws:aws-java-sdk-bom:1.11.1000')
     implementation 'com.amazonaws:aws-java-sdk-s3'

src/main/java/com/example/ai_tutor/AiTutorApplication.java

@@ -11,6 +11,7 @@
 @PropertySource(value = { "classpath:oauth2/application-oauth2.yml" }, factory = YamlPropertySourceFactory.class)
 @PropertySource(value = { "classpath:database/application-database.yml" }, factory = YamlPropertySourceFactory.class)
 @PropertySource(value = { "classpath:s3/application-s3.yml" }, factory = YamlPropertySourceFactory.class)
+@PropertySource(value = { "classpath:chatgpt/application-chatgpt.yml" }, factory = YamlPropertySourceFactory.class)
 public class AiTutorApplication {
 
     public static void main(String[] args) {

src/main/java/com/example/ai_tutor/domain/chatgpt/application/GptService.java

@@ -0,0 +1,101 @@
+package com.example.ai_tutor.domain.chatgpt.application;
+
+import com.example.ai_tutor.global.config.ChatGptConfig;
+import com.fasterxml.jackson.core.JsonProcessingException;
+import com.fasterxml.jackson.databind.JsonNode;
+import com.fasterxml.jackson.databind.ObjectMapper;
+import lombok.RequiredArgsConstructor;
+import org.springframework.beans.factory.annotation.Value;
+import org.springframework.http.*;
+import org.springframework.stereotype.Service;
+import org.springframework.web.client.RestTemplate;
+
+import java.util.*;
+
+@Service
+@RequiredArgsConstructor
+public class GptService {
+
+    private final Map<Long, List<Map<String, String>>> noteChatHistories = new HashMap<>();
+    private final Map<Long, String> noteOriginalTexts = new HashMap<>();
+    private final Map<Long, Integer> messageCounts = new HashMap<>(); // 메시지 카운트
+
+    @Value("${chatgpt.api-key}")
+    private String apiKey;
+
+    // 대화 시작 시 원문을 포함시키는 메소드
+    public void startConversation(Long noteId, String originalText) {
+        noteOriginalTexts.put(noteId, originalText);
+        clearChatHistory(noteId); // 대화 내역 초기화
+        // 원문 설명 메시지 추가
+        addOriginalTextToChatHistory(noteId);
+    }
+
+    private void addOriginalTextToChatHistory(Long noteId) {
+        String originalText = noteOriginalTexts.get(noteId);
+        if (originalText != null) {
+            List<Map<String, String>> chatHistory = noteChatHistories.computeIfAbsent(noteId, k -> new ArrayList<>());
+            Map<String, String> originalTextIntro = new HashMap<>();
+            originalTextIntro.put("role", "system");
+            originalTextIntro.put("content", "다음은 원문의 내용입니다:\n" + originalText);
+            chatHistory.add(originalTextIntro);
+        }
+    }
+
+    public JsonNode callChatGpt(Long noteId, String userMsg) throws JsonProcessingException {
+        HttpHeaders headers = new HttpHeaders();
+        headers.setAccept(Collections.singletonList(MediaType.APPLICATION_JSON));
+        headers.setContentType(MediaType.APPLICATION_JSON);
+        headers.setBearerAuth(apiKey);
+
+        ObjectMapper objectMapper = new ObjectMapper();
+
+        Map<String, Object> bodyMap = new HashMap<>();
+        bodyMap.put("model", ChatGptConfig.MODEL);
+        bodyMap.put("max_tokens", ChatGptConfig.MAX_TOKEN);
+
+        // 사용자 메시지와 컨텍스트 추가
+        List<Map<String, String>> chatHistory = noteChatHistories.computeIfAbsent(noteId, k -> new ArrayList<>());
+        Map<String, String> userMessage = new HashMap<>();
+        userMessage.put("role", "user");
+        userMessage.put("content", ChatGptConfig.CONTENT + userMsg);
+        chatHistory.add(userMessage);
+
+        bodyMap.put("messages", chatHistory);
+
+        String body = objectMapper.writeValueAsString(bodyMap);
+
+        HttpEntity<String> request = new HttpEntity<>(body, headers);
+
+        RestTemplate restTemplate = new RestTemplate();
+        ResponseEntity<String> response = restTemplate.exchange(ChatGptConfig.URL, HttpMethod.POST, request, String.class);
+
+        JsonNode responseBody = objectMapper.readTree(response.getBody());
+
+        // 대화 내역 업데이트
+        Map<String, String> assistantResponse = new HashMap<>();
+        assistantResponse.put("role", "assistant");
+        assistantResponse.put("content", responseBody.path("choices").get(0).path("message").path("content").asText());
+        chatHistory.add(assistantResponse);
+
+        // 메세지 카운트 (15회마다 원문 주입)
+        int count = messageCounts.getOrDefault(noteId, 0) + 1;
+        messageCounts.put(noteId, count);
+
+        if (count % 15 == 0) {
+            addOriginalTextToChatHistory(noteId);
+        }
+
+        return responseBody;
+    }
+
+    public String getAssistantMsg(Long noteId, String userMsg) throws JsonProcessingException {
+        JsonNode jsonNode = callChatGpt(noteId, userMsg);
+        return jsonNode.path("choices").get(0).path("message").path("content").asText();
+    }
+
+    // 대화 내역 초기화
+    public void clearChatHistory(Long noteId) {
+        noteChatHistories.remove(noteId);
+    }
+}

src/main/java/com/example/ai_tutor/domain/practice/application/PracticeService.java

@@ -103,6 +103,8 @@ public ResponseEntity<?> getQuestionsAndAnswers(UserPrincipal userPrincipal, Lon
                 .sorted(Comparator.comparing(PracticeResultsRes::getSequence))
                 .collect(Collectors.toList());
 
+        // test
+
         ApiResponse apiResponse = ApiResponse.builder()
                 .check(true)
                 .information(practiceResultsRes)

src/main/java/com/example/ai_tutor/domain/practice/dto/response/TutorRecordRes.java

@@ -1,11 +1,13 @@
 package com.example.ai_tutor.domain.practice.dto.response;
 
+import io.swagger.v3.oas.annotations.media.Schema;
 import lombok.Builder;
 import lombok.Getter;
 
 @Getter
 @Builder
 public class TutorRecordRes {
 
+    @Schema( type = "String", example ="(추후 추가)", description="특정 답변에 대한 튜터의 tts 녹음파일 주소입니다.")
     private String tutorRecordUrl;
 }

src/main/java/com/example/ai_tutor/domain/tutor/application/TutorService.java

@@ -0,0 +1,110 @@
+package com.example.ai_tutor.domain.tutor.application;
+
+import com.example.ai_tutor.domain.chatgpt.application.GptService;
+import com.example.ai_tutor.domain.note.domain.Note;
+import com.example.ai_tutor.domain.note.domain.repository.NoteRepository;
+import com.example.ai_tutor.domain.tutor.domain.Tutor;
+import com.example.ai_tutor.domain.tutor.domain.repository.TutorRepository;
+import com.example.ai_tutor.domain.tutor.dto.request.QuestionReq;
+import com.example.ai_tutor.domain.tutor.dto.response.AnswerRes;
+import com.example.ai_tutor.domain.tutor.dto.response.TutorRes;
+import com.example.ai_tutor.domain.user.domain.User;
+import com.example.ai_tutor.domain.user.domain.repository.UserRepository;
+import com.example.ai_tutor.global.DefaultAssert;
+import com.example.ai_tutor.global.config.security.token.UserPrincipal;
+import com.example.ai_tutor.global.payload.ApiResponse;
+import com.example.ai_tutor.global.payload.Message;
+import com.fasterxml.jackson.core.JsonProcessingException;
+import lombok.RequiredArgsConstructor;
+import org.springframework.http.ResponseEntity;
+import org.springframework.stereotype.Service;
+import org.springframework.transaction.annotation.Transactional;
+
+import java.util.List;
+import java.util.Objects;
+import java.util.Optional;
+
+@Service
+@RequiredArgsConstructor
+public class TutorService {
+
+    private final UserRepository userRepository;
+    private final NoteRepository noteRepository;
+    private final TutorRepository tutorRepository;
+    private final GptService gptService;
+
+    // 최초 호출 시 원문 내용 주입
+    public ResponseEntity<?> initializeChatting(UserPrincipal userPrincipal, Long noteId) {
+        Optional<Note> noteOptional = noteRepository.findById(noteId);
+        DefaultAssert.isTrue(noteOptional.isPresent(), "해당 노트가 존재하지 않습니다.");
+        Note note = noteOptional.get();
+
+        gptService.startConversation(noteId, note.getOriginalText());
+
+        ApiResponse apiResponse = ApiResponse.builder()
+                .check(true)
+                .information(Message.builder().message("원문이 주입되었습니다.").build())
+                .build();
+        return ResponseEntity.ok(apiResponse);
+    }
+
+    @Transactional
+    public ResponseEntity<?> questionToTutor(UserPrincipal userPrincipal, Long noteId, QuestionReq questionReq) throws JsonProcessingException {
+        Optional<User> userOptional = userRepository.findById(userPrincipal.getId());
+        DefaultAssert.isTrue(userOptional.isPresent(), "해당 사용자가 존재하지 않습니다.");
+        User user = userOptional.get();
+
+        Optional<Note> noteOptional = noteRepository.findById(noteId);
+        DefaultAssert.isTrue(noteOptional.isPresent(), "해당 노트가 존재하지 않습니다.");
+        Note note = noteOptional.get();
+
+        DefaultAssert.isTrue(Objects.equals(note.getUser().getUserId(), userPrincipal.getId()), "잘못된 접근입니다.");
+        // 챗지피티 호출
+        String answer = gptService.getAssistantMsg(noteId, questionReq.getQuestion());
+        // 질문 저장
+        // 답변 저장
+        Tutor tutor = Tutor.builder()
+                .question(questionReq.getQuestion())
+                .answer(answer)   // 튜터의 답변 저장
+                .folder(note.getFolder())
+                .note(note)
+                .user(user).build();
+        tutorRepository.save(tutor);
+        // response 생성
+        AnswerRes answerRes = AnswerRes.builder()
+                .answer(tutor.getAnswer())
+                .build();
+
+        ApiResponse apiResponse = ApiResponse.builder()
+                .check(true)
+                .information(answerRes)
+                .build();
+        return ResponseEntity.ok(apiResponse);
+    }
+
+    // 기존 학습 질문 및 답변 조회
+    public ResponseEntity<?> getQuestionAndAnswer(UserPrincipal userPrincipal, Long noteId) {
+        Optional<Note> noteOptional = noteRepository.findById(noteId);
+        DefaultAssert.isTrue(noteOptional.isPresent(), "해당 노트가 존재하지 않습니다.");
+        Note note = noteOptional.get();
+        // 사용자 검증
+        DefaultAssert.isTrue(Objects.equals(note.getUser().getUserId(), userPrincipal.getId()), "잘못된 접근입니다.");
+
+        List<Tutor> tutors = tutorRepository.findByNoteOrderByCreatedAt(note);
+
+        List<TutorRes> tutorResList = tutors.stream()
+                .map(tutor -> TutorRes.builder()
+                        .tutorId(tutor.getTutorId())
+                        .question(tutor.getQuestion())
+                        .answer(tutor.getAnswer())
+                        .build())
+                .toList();
+
+        ApiResponse apiResponse = ApiResponse.builder()
+                .check(true)
+                .information(tutorResList)
+                .build();
+
+        return ResponseEntity.ok(apiResponse);
+    }
+}

src/main/java/com/example/ai_tutor/domain/tutor/domain/Tutor.java

@@ -33,7 +33,7 @@ public class Tutor extends BaseEntity {
     @JoinColumn(name="folder_id")
     private Folder folder;
 
-    @OneToOne(fetch = FetchType.LAZY)
+    @ManyToOne(fetch = FetchType.LAZY)
     @JoinColumn(name="note_id")
     private Note note;
 

src/main/java/com/example/ai_tutor/domain/tutor/domain/repository/TutorRepository.java

@@ -0,0 +1,13 @@
+package com.example.ai_tutor.domain.tutor.domain.repository;
+
+import com.example.ai_tutor.domain.note.domain.Note;
+import com.example.ai_tutor.domain.tutor.domain.Tutor;
+import org.springframework.data.jpa.repository.JpaRepository;
+import org.springframework.stereotype.Repository;
+
+import java.util.List;
+
+@Repository
+public interface TutorRepository extends JpaRepository<Tutor, Long> {
+    List<Tutor> findByNoteOrderByCreatedAt(Note note);
+}

src/main/java/com/example/ai_tutor/domain/tutor/dto/request/QuestionReq.java

@@ -0,0 +1,11 @@
+package com.example.ai_tutor.domain.tutor.dto.request;
+
+import io.swagger.v3.oas.annotations.media.Schema;
+import lombok.Getter;
+
+@Getter
+public class QuestionReq {
+
+    @Schema( type = "String", example ="기말고사 잘 보는 방법 알려줘", description="질문 내용입니다.")
+    public String question;
+}

src/main/java/com/example/ai_tutor/domain/tutor/dto/response/AnswerRes.java

@@ -0,0 +1,13 @@
+package com.example.ai_tutor.domain.tutor.dto.response;
+
+import io.swagger.v3.oas.annotations.media.Schema;
+import lombok.Builder;
+import lombok.Getter;
+
+@Getter
+@Builder
+public class AnswerRes {
+
+    @Schema( type = "String", example ="계획적으로 공부하고 충분한 휴식을 취하세요.", description="질문에 대한 답변입니다.")
+    public String answer;
+}

src/main/java/com/example/ai_tutor/domain/tutor/dto/response/TutorRes.java

@@ -0,0 +1,19 @@
+package com.example.ai_tutor.domain.tutor.dto.response;
+
+import io.swagger.v3.oas.annotations.media.Schema;
+import lombok.Builder;
+import lombok.Getter;
+
+@Getter
+@Builder
+public class TutorRes {
+
+    @Schema( type = "Long", example ="1", description="챗봇 질문/답변의 id입니다.")
+    public Long tutorId;
+
+    @Schema( type = "String", example ="기말고사 잘 보는 방법 알려줘", description="질문 내용입니다.")
+    public String question;
+
+    @Schema( type = "String", example ="계획적으로 공부하고 충분한 휴식을 취하세요.", description="질문에 대한 답변입니다.")
+    public String answer;
+}