edit : dalle, imagen 추가

rest/app/routes/services.py

@@ -19,8 +19,57 @@ from rest.app import models as M
 from rest.app.utils.date_utils import D
 from custom_logger.custom_log import custom_logger as LOG
 
+from custom_apps.dalle3.custom_dalle import DallEArgument,dalle3_generate_image
+from custom_apps.imagen.custom_imagen import imagen_generate_image
 
 
 router = APIRouter(prefix="/services")
 
 
+@router.post("/imageGenerate/dalle3", summary="이미지 생성(AI) - DALL-E 3", response_model=M.ResponseBase)
+async def dalle3(request: Request, request_body_info: M.ImageGenerateReq):
+    """
+    ## 이미지 생성(AI) - DALL-E 3
+    > DALL-E 3 AI를 이용하여 이미지 생성
+    
+    ### Requriements
+    > - 쿠키 정보 설정(https://github.com/acheong08/BingImageCreator) 추후 set api 추가 예정 -> 현재 고정값
+    > - const.py 에 지정한 OUTPUT_FOLDER 하위에 dalle 폴더가 있어야함.
+    
+    """
+    response = M.ResponseBase()
+    try:
+        args = DallEArgument(prompt=request_body_info.prompt
+                    #   , download_count=request_body_info.downloadCount
+                    )
+        
+        dalle3_generate_image(args)
+        
+        return response.set_message()
+    
+    except Exception as e:
+        LOG.error(traceback.format_exc())
+        return response.set_error(e)
+    
+
+@router.post("/imageGenerate/imagen", summary="이미지 생성(AI) - imagen", response_model=M.ResponseBase)
+async def imagen(request: Request, request_body_info: M.ImageGenerateReq):
+    """
+    ## 이미지 생성(AI) - imagen
+    > imagen AI를 이용하여 이미지 생성
+    
+    ### Requriements
+    > - googlecli 설치(https://cloud.google.com/sdk/docs/install?hl=ko#linux)
+    > - const.py 에 지정한 OUTPUT_FOLDER 하위에 imagen 폴더가 있어야함.
+    
+    """
+    response = M.ResponseBase()
+    try:
+        
+        imagen_generate_image(request_body_info.prompt)
+        
+        return response.set_message()
+    
+    except Exception as e:
+        LOG.error(traceback.format_exc())
+        return response.set_error(e)