edit : clip-vit 모델 추가

custom_apps/imagen/custom_imagen.py

@@ -4,10 +4,12 @@ import vertexai
 import os
 from vertexai.preview.vision_models import ImageGenerationModel
 
-from const import REMOTE_FOLDER, TEMP_FOLDER
+from const import TEMP_FOLDER
 from main_rest.app.utils.parsing_utils import prompt_to_filenames
 from main_rest.app.utils.date_utils import D
 from utils.custom_sftp import sftp_client
+from config import rest_config
+
 
 class ImagenConst:
     project_id = "glasses-imagen"
@@ -22,7 +24,7 @@ def imagen_generate_image(prompt,download_count=1):
 
     _file_name = prompt_to_filenames(prompt)
     _folder = os.path.join(TEMP_FOLDER)
-    _remote_folder = os.path.join(REMOTE_FOLDER,"imagen")
+    _remote_folder = os.path.join(rest_config.remote_folder,"imagen")
     _datetime = D.date_file_name()
     
     # if not os.path.isdir(_folder):
@@ -88,6 +90,19 @@ def imagen_generate_image_path(image_prompt):
        
     return os.path.join(folder_name,f"query.png")
 
+def imagen_generate_temp_image_path(image_prompt):
+    
+    create_time = D.date_file_name()
+    
+    if not os.path.exists(TEMP_FOLDER):
+        os.makedirs(TEMP_FOLDER)
+    
+    generate_img = imagen_generate_image_data(image_prompt)
+    
+    generate_img.save(os.path.join(TEMP_FOLDER,f"query_{create_time}.png"))
+       
+    return os.path.join(TEMP_FOLDER,f"query_{create_time}.png")
+
 if __name__ == '__main__':
     pass
     # imagen_generate_image_data("cat")