added generate multiple flashcards for word functionality

Author: jakegodsall

src/main/java/com/jakegodsall/services/flashcard/FlashcardService.java

@@ -32,6 +32,18 @@ public interface FlashcardService {
      */
     Flashcard generateFlashcard(String targetWord, FlashcardType flashcardType, Language language, Options options);
 
+    /**
+     * Generates multiple flashcards for a single target word in one API call.
+     *
+     * @param targetWord the word for which multiple flashcards are generated
+     * @param flashcardType the type of the flashcards (e.g., multiple choice, fill-in-the-blank)
+     * @param language the language of the flashcards
+     * @param options additional options to customize flashcard generation
+     * @param count the number of flashcards to generate
+     * @return a list of generated flashcards
+     */
+    List<Flashcard> generateMultipleFlashcardsForWord(String targetWord, FlashcardType flashcardType, Language language, Options options, int count);
+
     /**
      * Generates flashcards interactively by prompting the user for words until they exit.
      *

src/main/java/com/jakegodsall/services/flashcard/impl/FlashcardServiceGPTImpl.java

@@ -72,6 +72,27 @@ public Flashcard generateFlashcard(String targetWord, FlashcardType flashcardTyp
         return null;
     }
 
+    @Override
+    public List<Flashcard> generateMultipleFlashcardsForWord(String targetWord, FlashcardType flashcardType, Language language, Options options, int count) {
+        try {
+            // Generate the prompt for multiple flashcards
+            String prompt = promptGenerator.generatePromptForMultipleFlashcards(targetWord, flashcardType, language, options, count);
+            // Generate HTTP POST request body
+            String requestBody = promptGenerator.generateRequestBody(prompt);
+            // Send the POST request to the GPT API
+            HttpResponse response = httpClientService.sendPostRequest(API_CHAT_URL, requestBody);
+            // Get the result
+            HttpEntity responseEntity = response.getEntity();
+            String result = EntityUtils.toString(responseEntity);
+            // Parse multiple flashcards from the result
+            return jsonParseService.parseMultipleFlashcards(result, flashcardType);
+        } catch (IOException ex) {
+            logger.log(Level.SEVERE, ex.getMessage(), ex);
+            System.err.println(ex.getMessage());
+            return List.of();
+        }
+    }
+
     @Override
     public List<Flashcard> generateFlashcardsInteractively(FlashcardType flashcardType, Language language, Options options) throws IOException {
         List<Flashcard> flashcards = new ArrayList<>();

src/main/java/com/jakegodsall/services/json/JsonParseService.java

@@ -32,4 +32,13 @@ public interface JsonParseService {
      * @return a parsed {@link Flashcard} object.
      */
     Flashcard parseFlashcard(String responseBody, FlashcardType flashcardType);
+
+    /**
+     * Parses multiple flashcards from a single API response.
+     *
+     * @param responseBody the raw JSON response from the API containing multiple flashcards.
+     * @param flashcardType the type of flashcards to create, defined by the {@link FlashcardType} enum.
+     * @return a list of parsed {@link Flashcard} objects.
+     */
+    List<Flashcard> parseMultipleFlashcards(String responseBody, FlashcardType flashcardType);
 }

src/main/java/com/jakegodsall/services/json/impl/JsonParseServiceGPTImpl.java

@@ -50,7 +50,7 @@ public Flashcard parseFlashcard(String responseBody, FlashcardType flashcardType
         try {
             String content = parseContentFromResponse(responseBody);
 
-            // Parse the JSON int oa JsonNode tree
+            // Parse the JSON into a JsonNode tree
             JsonNode rootNode = objectMapper.readTree(content);
 
             // Delegate to specific parsing logic based on FlashcardType
@@ -65,6 +65,34 @@ public Flashcard parseFlashcard(String responseBody, FlashcardType flashcardType
         }
     }
 
+    @Override
+    public List<Flashcard> parseMultipleFlashcards(String responseBody, FlashcardType flashcardType) {
+        try {
+            String content = parseContentFromResponse(responseBody);
+            JsonNode rootNode = objectMapper.readTree(content);
+
+            List<Flashcard> flashcards = new ArrayList<>();
+
+            if (!rootNode.isArray()) {
+                throw new NoSuchElementException("Expected an array of flashcards in the response");
+            }
+
+            for (JsonNode flashcardNode : rootNode) {
+                Flashcard flashcard = switch (flashcardType) {
+                    case WORD -> parseWordFlashcardFromJson(flashcardNode);
+                    case SENTENCE -> parseSentenceFlashcardFromJson(flashcardNode);
+                    default -> throw new IllegalArgumentException("Unsupported FlashcardType: " + flashcardType);
+                };
+                flashcards.add(flashcard);
+            }
+
+            return flashcards;
+        } catch (JsonProcessingException ex) {
+            System.err.println(ex.getMessage());
+            return List.of();
+        }
+    }
+
     /**
      * Parses a WordFlashcard from the given JsonNode.
      */

src/main/java/com/jakegodsall/services/prompt/PromptService.java

@@ -35,6 +35,18 @@ public interface PromptService {
      */
     String generatePrompt(String targetWord, FlashcardType flashcardType, Language language, Options options);
 
+    /**
+     * Generates a prompt for multiple flashcards based on a single target word.
+     *
+     * @param targetWord the word in the target language that will be used to generate multiple flashcards.
+     * @param flashcardType the type of flashcard to generate.
+     * @param language the language in which the flashcard content will be generated.
+     * @param options additional options that influence the generation of the prompt.
+     * @param count the number of flashcards to generate.
+     * @return the generated prompt as a {@code String}, which will be sent to the API.
+     */
+    String generatePromptForMultipleFlashcards(String targetWord, FlashcardType flashcardType, Language language, Options options, int count);
+
     /**
      * Creates a simple prompt for the next word in a sequence.
      *

src/main/java/com/jakegodsall/services/prompt/impl/PromptServiceGPTImpl.java

@@ -40,6 +40,16 @@ public String generatePrompt(String targetWord, FlashcardType flashcardType, Lan
         return generateBasePrompt() + getDescriptionOfContent(flashcardType) + getFlashcardStructure(flashcardType) + "The word is " + targetWord + " and the target language is " + language.getName() + ".\n";
     }
 
+    @Override
+    public String generatePromptForMultipleFlashcards(String targetWord, FlashcardType flashcardType, Language language, Options options, int count) {
+        return "Generate " + count + " different flashcards as a JSON array. For each flashcard in the array:\n" +
+               generateBasePrompt() + getDescriptionOfContent(flashcardType) + 
+               "The structure for each flashcard in the array should be:\n" + 
+               getFlashcardStructure(flashcardType) + 
+               "The word is " + targetWord + " and the target language is " + language.getName() + ".\n" +
+               "Each flashcard should use different example sentences and translations while maintaining accuracy.\n";
+    }
+
     @Override
     public String generatePromptForSubsequentWord(String targetWord) {
         return "The next word is " + targetWord;