feat relight 增加打光方向

fix

app/service/generate_image/service_generate_relight_image.py

@@ -7,16 +7,15 @@
 @Date    ：2023/7/26 12:01:05
 @detail  ：
 """
-import io
 import json
 import logging
 import time
+
 import cv2
+import numpy as np
 import redis
 import tritonclient.grpc as grpcclient
-import numpy as np
-from PIL import Image, ImageOps
-from minio import Minio
+from PIL import Image
 from tritonclient.utils import np_to_triton_dtype
 
 from app.core.config import *
@@ -40,7 +39,7 @@ class GenerateRelightImage:
         self.prompt = request_data.prompt
         self.seed = "1"
         self.negative_prompt = 'lowres, bad anatomy, bad hands, cropped, worst quality'
-        self.direction = "Right Light"
+        self.direction = request_data.direction
         self.image_url = request_data.image_url
         self.image = oss_get_image(bucket=self.image_url.split('/')[0], object_name=self.image_url[self.image_url.find('/') + 1:], data_type="cv2")
         self.tasks_id = request_data.tasks_id