edit : 실제 이미지 생성한 갯수 response 에 추가

rest/app/models.py

@@ -571,4 +571,31 @@ class ImageGenerateReq(BaseModel):
     ### [Request] image generate request
     """
     prompt : str = Field(description='프롬프트', example='검은색 안경')
-    downloadCount : int = Field(1, description='이미지 생성 갯수', example=1)
+    downloadCount : int = Field(1, description='이미지 생성 갯수', example=1)
+    
+
+#===============================================================================
+#===============================================================================
+#===============================================================================
+#===============================================================================
+
+class ImageGenerateRes(ResponseBase):
+    """
+    ### image generate response
+    """
+    imageLen : int = Field(0, description='실제 이미지 생성 갯수', example=1)
+    
+    @staticmethod
+    def set_error(error,img_len=0):
+        ImageGenerateRes.result = False
+        ImageGenerateRes.error = str(error)
+        ImageGenerateRes.imageLen = img_len
+        
+        return ImageGenerateRes
+    
+    @staticmethod
+    def set_message(img_len):
+        ImageGenerateRes.result = True
+        ImageGenerateRes.error = None
+        ImageGenerateRes.imageLen = img_len
+        return ImageGenerateRes