diff --git a/HalfFifty_BE/src/main/java/HalfFifty/HalfFifty_BE/translation/bean/FlaskSignLanguageBean.java b/HalfFifty_BE/src/main/java/HalfFifty/HalfFifty_BE/translation/bean/FlaskSignLanguageBean.java index 7394ab7..78e2813 100644 --- a/HalfFifty_BE/src/main/java/HalfFifty/HalfFifty_BE/translation/bean/FlaskSignLanguageBean.java +++ b/HalfFifty_BE/src/main/java/HalfFifty/HalfFifty_BE/translation/bean/FlaskSignLanguageBean.java @@ -1,6 +1,8 @@ package HalfFifty.HalfFifty_BE.translation.bean; import HalfFifty.HalfFifty_BE.translation.domain.DTO.RequestSignLanguageDTO; +import org.springframework.beans.factory.annotation.Value; +import org.springframework.core.ParameterizedTypeReference; import org.springframework.http.*; import org.springframework.stereotype.Component; import org.springframework.web.client.RestTemplate; @@ -11,7 +13,9 @@ @Component public class FlaskSignLanguageBean { private final RestTemplate restTemplate; - private final String aiServerUrl = "http://3.39.24.155/predict"; // Flask 서버 URL + + @Value("${ai.server.url}") + private String aiServerUrl; // 환경변수로 관리 public FlaskSignLanguageBean() { this.restTemplate = new RestTemplate(); @@ -28,7 +32,12 @@ public Map exec(RequestSignLanguageDTO requestSignLanguageDTO) { HttpEntity> request = new HttpEntity<>(requestBody, headers); try { - ResponseEntity response = restTemplate.postForEntity(aiServerUrl, request, Map.class); + ResponseEntity> response = restTemplate.exchange( + aiServerUrl, + HttpMethod.POST, + request, + new ParameterizedTypeReference>() {} + ); if (response.getStatusCode() == HttpStatus.OK) { return response.getBody(); diff --git a/HalfFifty_BE/src/main/resources/application.properties b/HalfFifty_BE/src/main/resources/application.properties index 22cb0b0..5984c36 100644 --- a/HalfFifty_BE/src/main/resources/application.properties +++ b/HalfFifty_BE/src/main/resources/application.properties @@ -16,4 +16,7 @@ spring.datasource.username=${DB_USERNAME} spring.datasource.password=${DB_PASSWORD} # lambda -aws.lambda.functionName = myLambda \ No newline at end of file +aws.lambda.functionName = myLambda + +# AI server +ai.server.url = ${AI_URL} \ No newline at end of file