AI_Recommendations

.gitignore

@@ -3,7 +3,10 @@ target/
 !.mvn/wrapper/maven-wrapper.jar
 !**/src/main/**/target/
 !**/src/test/**/target/
+
+# Ignore env files
 .env
+.env.properties
 
 ### STS ###
 .apt_generated

compose.yaml

@@ -8,10 +8,14 @@ services:
       - 'POSTGRES_USER=myuser'
     ports:
       - '5432:5432'
+
   app:
+    build:
+      context: .
+      dockerfile: Dockerfile
+    image: 'catinder:latest'
     depends_on:
       - postgres
-    image: 'catinder:latest'
     env_file:
       - .env
     environment:

src/main/java/org/example/springboot25/controller/AiRecommendationRestController.java

@@ -0,0 +1,40 @@
+package org.example.springboot25.controller;
+
+import org.example.springboot25.service.AiRecommendationService;
+import org.springframework.http.HttpStatus;
+import org.springframework.http.ResponseEntity;
+import org.springframework.web.bind.annotation.*;
+
+import java.util.HashMap;
+import java.util.Map;
+
+@RestController
+@RequestMapping("/api/ai-recommendations")
+public class AiRecommendationRestController {
+
+    private final AiRecommendationService aiRecommendationService;
+
+    public AiRecommendationRestController(AiRecommendationService aiRecommendationService) {
+        this.aiRecommendationService = aiRecommendationService;
+    }
+
+    // Return recommendation based on cat breed with structured JSON and error handling
+    @GetMapping("/{breed}")
+    public ResponseEntity<?> getRecommendation(@PathVariable String breed) {
+        String recommendation = aiRecommendationService.getRecommendationForBreed(breed);
+
+        // If the breed is not recognized, return error message with 404
+        if (recommendation.contains("Unknown breed")) {
+            Map<String, String> errorResponse = new HashMap<>();
+            errorResponse.put("error", "Unknown cat breed");
+            errorResponse.put("message", recommendation);
+            return ResponseEntity.status(HttpStatus.NOT_FOUND).body(errorResponse);
+        }
+
+        // Success: return breed and recommendation
+        Map<String, String> successResponse = new HashMap<>();
+        successResponse.put("breed", breed);
+        successResponse.put("recommendation", recommendation);
+        return ResponseEntity.ok(successResponse);
+    }
+}

src/main/java/org/example/springboot25/controller/AiRecommendationViewController.java

@@ -0,0 +1,38 @@
+package org.example.springboot25.controller;
+
+import org.example.springboot25.service.AiRecommendationService;
+import org.springframework.stereotype.Controller;
+import org.springframework.ui.Model;
+import org.springframework.web.bind.annotation.*;
+
+@Controller
+@RequestMapping("/ai-recommendations")
+public class AiRecommendationViewController {
+
+    private final AiRecommendationService aiRecommendationService;
+
+    public AiRecommendationViewController(AiRecommendationService aiRecommendationService) {
+        this.aiRecommendationService = aiRecommendationService;
+    }
+
+    // Displays the form where the user can input the cat breed
+    @GetMapping
+    public String showForm() {
+        return "ai_recommendation_form";
+    }
+
+    @PostMapping
+    public String showResult(@RequestParam String breed, Model model) {
+        // Validate input: ensure the breed is not null or empty
+        if (breed == null || breed.trim().isEmpty()) {
+            model.addAttribute("error", "Please enter a breed name");
+            return "ai_recommendation_form";
+        }
+
+        breed = breed.trim();
+        String recommendation = aiRecommendationService.getRecommendationForBreed(breed);
+        model.addAttribute("breed", breed);
+        model.addAttribute("recommendation", recommendation);
+        return "ai_recommendation_result";
+    }
+}

src/main/java/org/example/springboot25/service/AiRecommendationService.java

@@ -0,0 +1,63 @@
+package org.example.springboot25.service;
+
+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.HashMap;
+import java.util.Map;
+
+@Service
+public class AiRecommendationService {
+
+    @Value("${openai.api.key}")
+    private String apiKey;
+
+    private final RestTemplate restTemplate = new RestTemplate();
+    private static final String OPENAI_API_URL = "https://api.openai.com/v1/chat/completions";
+
+    public String getRecommendationForBreed(String breed) {
+        if (breed == null || breed.trim().isEmpty()) {
+            return "Unknown breed. Please enter a valid cat breed.";
+        }
+
+        breed = breed.trim();
+
+        try {
+            // Prepare the OpenAI API request
+            HttpHeaders headers = new HttpHeaders();
+            headers.setContentType(MediaType.APPLICATION_JSON);
+            headers.setBearerAuth(apiKey);
+
+            Map<String, Object> message = new HashMap<>();
+            message.put("role", "user");
+            message.put("content", "Give me a short but useful care tip for the cat breed: " + breed);
+
+            Map<String, Object> body = new HashMap<>();
+            body.put("model", "gpt-3.5-turbo");
+            body.put("messages", new Object[]{message});
+            body.put("max_tokens", 50);
+
+            HttpEntity<Map<String, Object>> entity = new HttpEntity<>(body, headers);
+
+            ResponseEntity<Map> response = restTemplate.exchange(
+                    OPENAI_API_URL,
+                    HttpMethod.POST,
+                    entity,
+                    Map.class
+            );
+
+            if (response.getStatusCode() == HttpStatus.OK && response.getBody() != null) {
+                Map<String, Object> choice = ((java.util.List<Map<String, Object>>) response.getBody().get("choices")).get(0);
+                Map<String, Object> messageContent = (Map<String, Object>) choice.get("message");
+                return (String) messageContent.get("content");
+            } else {
+                return "Something went wrong while getting recommendation. Try again later.";
+            }
+
+        } catch (Exception e) {
+            return "Error calling AI service: " + e.getMessage();
+        }
+    }
+}

src/main/resources/templates/ai_recommendation/ai_recommendation_form.html

@@ -0,0 +1,14 @@
+<!DOCTYPE html>
+<html xmlns:th="http://www.thymeleaf.org">
+<head>
+    <title>AI Recommendation</title>
+</head>
+<body>
+<h1>Cat AI Recommendation</h1>
+<form action="#" th:action="@{/ai-recommendations}" method="post">
+    <label for="breed">Cat Breed:</label>
+    <input type="text" id="breed" name="breed" required>
+    <button type="submit">Get Recommendation</button>
+</form>
+</body>
+</html>

src/main/resources/templates/ai_recommendation/ai_recommendation_result.html

@@ -0,0 +1,11 @@
+<!DOCTYPE html>
+<html xmlns:th="http://www.thymeleaf.org">
+<head>
+    <title>Recommendation Result</title>
+</head>
+<body>
+<h2>Recommendation for <span th:text="${breed}"></span>:</h2>
+<p th:text="${recommendation}"></p>
+<a href="/ai-recommendations">Try another</a>
+</body>
+</html>

src/test/java/org/example/springboot25/service/AiRecommendationServiceTest.java

@@ -0,0 +1,58 @@
+package org.example.springboot25.service;
+
+import org.junit.jupiter.api.BeforeEach;
+import org.junit.jupiter.api.Test;
+import static org.assertj.core.api.Assertions.*;
+
+public class AiRecommendationServiceTest {
+
+    private AiRecommendationService aiRecommendationService;
+
+    @BeforeEach
+    void setUp() {
+        aiRecommendationService = new AiRecommendationService();
+    }
+
+    @Test
+    void shouldReturnCorrectRecommendationForValidBreed() {
+        String result = aiRecommendationService.getRecommendationForBreed("PERSIAN");
+        assertThat(result).contains("Groom daily");
+    }
+
+    @Test
+    void shouldReturnCorrectRecommendationForBreedWithSpaces() {
+        String result = aiRecommendationService.getRecommendationForBreed("american shorthair");
+        assertThat(result).contains("Easy-going");
+    }
+
+    @Test
+    void shouldReturnCorrectRecommendationForLowercaseInput() {
+        String result = aiRecommendationService.getRecommendationForBreed("siamese");
+        assertThat(result).contains("Talkative");
+    }
+
+    @Test
+    void shouldHandleUnknownBreedGracefully() {
+        String result = aiRecommendationService.getRecommendationForBreed("fluffybean");
+        assertThat(result).contains("Unknown breed");
+    }
+
+    @Test
+    void shouldHandleEmptyStringInput() {
+        String result = aiRecommendationService.getRecommendationForBreed("");
+        assertThat(result).contains("Unknown breed");
+    }
+
+    @Test
+    void shouldHandleWhitespaceOnlyInput() {
+        String result = aiRecommendationService.getRecommendationForBreed("   ");
+        assertThat(result).contains("Unknown breed");
+    }
+
+    @Test
+    void shouldHandleNullInput() {
+        String result = aiRecommendationService.getRecommendationForBreed(null);
+        assertThat(result).contains("Unknown breed");
+    }
+}
+