修改 imageToSketch

src/main/java/com/ai/da/python/PythonService.java

@@ -63,8 +63,6 @@ public class PythonService {
     private String accessPythonIp;
     @Value("${access.python.port:''}")
     private String accessPythonPort;
-    @Value("${access.python.address}")
-    private String fastApiPythonAddress;
     @Value("${minio.bucketName.gradient}")
     private String gradientBucketName;
     @Value("${access.python.generate_sr_port}")
@@ -3449,7 +3447,7 @@ public class PythonService {
         throw new BusinessException("relightImage.interface.exception");
     }
 
-    public String imageToSketch(String imagePath, String bucket, String objectName){
+    public String imageToSketch(String imagePath, String bucket, String objectName, String styleCode, String styleImageUrl){
         OkHttpClient client = new OkHttpClient().newBuilder()
                 .connectTimeout(30, TimeUnit.SECONDS)
                 .pingInterval(5, TimeUnit.SECONDS)//websocket轮训间隔(单位:秒)
@@ -3460,6 +3458,8 @@ public class PythonService {
         //关闭FastJson的引用检测 防止出现$ref 现象
         Map<String, Object> map = new HashMap<>();
         map.put("image_url", imagePath);
+        map.put("style_image_url", styleImageUrl);
+        map.put("default_style", styleCode);
         map.put("sketch_bucket", bucket);
         map.put("sketch_name", objectName);