Update app.py

app.py

@@ -178,69 +178,21 @@ def convert_to_jpg(pil_image):
     jpg_image = Image.open(jpg_buffer)
     return jpg_image
 
-def generate_with_images(prompt, images, variation_index=0):
-    # 기존 함수 유지
-    try:
-        api_key = os.environ.get("GEMINI_API_KEY")
-        if not api_key:
-            return None, "API 키가 설정되지 않았습니다. 환경변수를 확인해주세요."
-
-        client = genai.Client(api_key=api_key)
-        logger.info(f"Gemini API 요청 시작 - 프롬프트: {prompt}, 변형 인덱스: {variation_index}")
-
-        variation_suffixes = [
-            " Create this as the first variation. Do not add any text, watermarks, or labels to the image.",
-            " Create this as the second variation with more vivid colors. Do not add any text, watermarks, or labels to the image.",
-            " Create this as the third variation with a more creative style. Do not add any text, watermarks, or labels to the image.",
-            " Create this as the fourth variation with enhanced details. Do not add any text, watermarks, or labels to the image."
-        ]
+def save_as_jpg(img):
+    """이미지를 강제로 JPG로 저장하고 그 경로를 반환합니다."""
+    if img is None:
+        return None
         
-        if variation_index < len(variation_suffixes):
-            prompt = prompt + variation_suffixes[variation_index]
-        else:
-            prompt = prompt + " Do not add any text, watermarks, or labels to the image."
-
-        contents = [prompt]
-        for idx, img in enumerate(images, 1):
-            if img is not None:
-                contents.append(img)
-                logger.info(f"이미지 #{idx} 추가됨")
-
-        response = client.models.generate_content(
-            model="gemini-2.0-flash-exp-image-generation",
-            contents=contents,
-            config=types.GenerateContentConfig(
-                response_modalities=['Text', 'Image'],
-                temperature=1,
-                top_p=0.95,
-                top_k=40,
-                max_output_tokens=8192
-            )
-        )
-
-        with tempfile.NamedTemporaryFile(suffix=".png", delete=False) as tmp:
-            temp_path = tmp.name
-            result_text = ""
-            image_found = False
-            for part in response.candidates[0].content.parts:
-                if hasattr(part, 'text') and part.text:
-                    result_text += part.text
-                    logger.info(f"응답 텍스트: {part.text}")
-                elif hasattr(part, 'inline_data') and part.inline_data:
-                    save_binary_file(temp_path, part.inline_data.data)
-                    image_found = True
-                    logger.info("응답에서 이미지 추출 성공")
-            if not image_found:
-                return None, f"API에서 이미지를 생성하지 못했습니다. 응답 텍스트: {result_text}"
-            
-            # 이미지를 JPG로 변환
-            result_img = Image.open(temp_path)
-            result_img = convert_to_jpg(result_img)
-            
-            return result_img, f"이미지가 성공적으로 생성되었습니다. {result_text}"
-    except Exception as e:
-        logger.exception("이미지 생성 중 오류 발생:")
-        return None, f"오류 발생: {str(e)}"
+    # RGB로 변환 (JPG는 알파 채널을 지원하지 않음)
+    if img.mode == "RGBA":
+        img = img.convert("RGB")
+        
+    # 임시 파일로 JPG 저장
+    temp_jpg = tempfile.NamedTemporaryFile(suffix=".jpg", delete=False)
+    img.save(temp_jpg.name, format="JPEG", quality=95)
+    temp_jpg.close()
+    
+    return temp_jpg.name
 
 def process_images_with_prompt(image1, image2, image3, prompt, variation_index=0, max_retries=3):
     # 기존 함수 내용 유지
@@ -966,11 +918,11 @@ with gr.Blocks(css=custom_css) as demo:
                         gr.HTML('<div class="section-title"><i class="fas fa-images"></i> 생성된 이미지</div>')
                         with gr.Row():
                             with gr.Column():
-                                output_image1 = gr.Image(label="이미지 #1", elem_classes="image-container", height=400, format="jpg")
-                                output_image3 = gr.Image(label="이미지 #3", elem_classes="image-container", height=400, format="jpg")
+                                output_image1 = gr.Image(label="이미지 #1", elem_classes="image-container", height=400, type="filepath")
+                                output_image3 = gr.Image(label="이미지 #3", elem_classes="image-container", height=400, type="filepath")
                             with gr.Column():
-                                output_image2 = gr.Image(label="이미지 #2", elem_classes="image-container", height=400, format="jpg")
-                                output_image4 = gr.Image(label="이미지 #4", elem_classes="image-container", height=400, format="jpg")
+                                output_image2 = gr.Image(label="이미지 #2", elem_classes="image-container", height=400, type="filepath")
+                                output_image4 = gr.Image(label="이미지 #4", elem_classes="image-container", height=400, type="filepath")
                         
                         gr.HTML('<div class="section-title"><i class="fas fa-info-circle"></i> 결과 정보</div>')
                         output_text = gr.Textbox(label="상태 메시지", lines=2, elem_classes="custom-input")