edit: ict센터입고세팅을 기본값으로 설정, ict 설정시 sftp비활성화, l14_336 사용 못하던것 수정

custom_apps/gemini/main.py

@@ -16,25 +16,30 @@ def gemini_image(prompt, folder=None):
     image_path = ''
     client = genai.Client(api_key="AIzaSyB7tu67y9gOkJkpQtvI5OAYSzUzwv9qwnE")
 
-    response = client.models.generate_content(
-        model="gemini-2.0-flash-preview-image-generation",
-        contents=prompt,
-        config=types.GenerateContentConfig(
-          response_modalities=['TEXT', 'IMAGE']
+    for i in range(3):
+        response = client.models.generate_content(
+            model="gemini-2.0-flash-preview-image-generation",
+            contents=prompt,
+            config=types.GenerateContentConfig(
+            response_modalities=['TEXT', 'IMAGE']
+            )
         )
-    )
-    if folder == None:
-        folder = TEMP_FOLDER
+        if folder == None:
+            folder = TEMP_FOLDER
+        
+        if not os.path.exists(folder):
+            os.makedirs(folder)
+        
+        for part in response.candidates[0].content.parts:
+            if part.inline_data is not None:
+                image = Image.open(BytesIO((part.inline_data.data)))
+                image_path = os.path.join(folder,f"gemini_{D.date_file_name()}_query.png")
+                image.save(image_path)
+        
+        LOG.info(f"image generate : {image_path}")
+        time.sleep(2)
+        
+        if os.path.exists(image_path):
+            return image_path
     
-    if not os.path.exists(folder):
-        os.makedirs(folder)
-    
-    for part in response.candidates[0].content.parts:
-        if part.inline_data is not None:
-            image = Image.open(BytesIO((part.inline_data.data)))
-            image_path = os.path.join(folder,f"gemini_{D.date_file_name()}_query.png")
-            image.save(image_path)
-    
-    LOG.info(f"image generate : {image_path}")
-    time.sleep(2)
     return image_path