diff --git a/src/main/java/com/ai/da/python/PythonService.java b/src/main/java/com/ai/da/python/PythonService.java index e281a5b3..dbb005ff 100644 --- a/src/main/java/com/ai/da/python/PythonService.java +++ b/src/main/java/com/ai/da/python/PythonService.java @@ -3994,11 +3994,12 @@ public class PythonService { // } //**********************end*********************************** //临时补丁 - if (category.split("_")[0].equals("male")&&style.toLowerCase().equals("lolita")){ - style=null; - } - if (category.split("_")[0].equals("male")&&style.toLowerCase().equalsIgnoreCase("romantic")){ - style=null; + if (category != null && style != null) { + String categoryPrefix = category.split("_")[0]; + if ("male".equals(categoryPrefix) && + (style.equalsIgnoreCase("lolita") || style.equalsIgnoreCase("romantic"))) { + style = null; + } } AuthPrincipalVo userHolder = UserContext.getUserHolder();