Update app.py

app.py

@@ -156,6 +156,28 @@ def preprocess_prompt(prompt, image1, image2, image3):
     prompt += " 이미지를 생성해주세요. 이미지에 텍스트나 글자를 포함하지 마세요."
     return prompt
 
+def convert_to_jpg(pil_image):
+    """이미지를 JPG 형식으로 변환합니다."""
+    import tempfile
+    from PIL import Image
+    import io
+    
+    if pil_image is None:
+        return None
+        
+    # RGBA 이미지를 RGB로 변환 (JPG는 알파 채널을 지원하지 않음)
+    if pil_image.mode == "RGBA":
+        pil_image = pil_image.convert("RGB")
+    
+    # 이미지를 JPG로 변환
+    jpg_buffer = io.BytesIO()
+    pil_image.save(jpg_buffer, format="JPEG", quality=95)
+    jpg_buffer.seek(0)
+    
+    # JPG 이미지 다시 로드
+    jpg_image = Image.open(jpg_buffer)
+    return jpg_image
+
 def generate_with_images(prompt, images, variation_index=0):
     # 기존 함수 유지
     try:
@@ -211,17 +233,9 @@ def generate_with_images(prompt, images, variation_index=0):
             if not image_found:
                 return None, f"API에서 이미지를 생성하지 못했습니다. 응답 텍스트: {result_text}"
             
-            # 여기서 이미지를 JPG로 변환합니다
+            # 이미지를 JPG로 변환
             result_img = Image.open(temp_path)
-            if result_img.mode == "RGBA":
-                result_img = result_img.convert("RGB")  # RGBA를 RGB로 변환 (JPG는 알파 채널을 지원하지 않음)
-            
-            # 이미지를 JPG로 저장
-            jpg_temp = tempfile.NamedTemporaryFile(suffix=".jpg", delete=False).name
-            result_img.save(jpg_temp, format="JPEG", quality=95)
-            
-            # JPG 이미지를 다시 로드
-            result_img = Image.open(jpg_temp)
+            result_img = convert_to_jpg(result_img)
             
             return result_img, f"이미지가 성공적으로 생성되었습니다. {result_text}"
     except Exception as e:
@@ -229,7 +243,7 @@ def generate_with_images(prompt, images, variation_index=0):
         return None, f"오류 발생: {str(e)}"
 
 def process_images_with_prompt(image1, image2, image3, prompt, variation_index=0, max_retries=3):
-    # 기존 함수 유지
+    # 기존 함수 내용 유지
     retry_count = 0
     last_error = None
     
@@ -259,6 +273,8 @@ def process_images_with_prompt(image1, image2, image3, prompt, variation_index=0
 
             result_img, status = generate_with_images(final_prompt, valid_images, variation_index)
             if result_img is not None:
+                # 이미지를 JPG로 변환하여 반환
+                result_img = convert_to_jpg(result_img)
                 return result_img, status, final_prompt
             else:
                 last_error = status
@@ -274,7 +290,7 @@ def process_images_with_prompt(image1, image2, image3, prompt, variation_index=0
     return None, f"최대 재시도 횟수({max_retries}회) 초과 후 실패: {last_error}", prompt
 
 def generate_multiple_images(image1, image2, image3, prompt, progress=gr.Progress()):
-    # 기존 함수 유지
+    # 결과 이미지들이 JPG로 변환되도록 수정
     results = []
     statuses = []
     prompts = []
@@ -289,6 +305,8 @@ def generate_multiple_images(image1, image2, image3, prompt, progress=gr.Progres
         result_img, status, final_prompt = process_images_with_prompt(image1, image2, image3, prompt, i, max_retries)
         
         if result_img is not None:
+            # 이미지를 JPG로 변환
+            result_img = convert_to_jpg(result_img)
             results.append(result_img)
             statuses.append(f"이미지 #{i+1}: {status}")
             prompts.append(f"이미지 #{i+1}: {final_prompt}")
@@ -309,6 +327,7 @@ def generate_multiple_images(image1, image2, image3, prompt, progress=gr.Progres
     
     return results[0], results[1], results[2], results[3], combined_status, combined_prompts
 
+    
 # GFPGAN 관련 코드 유지
 import sys
 from torchvision.transforms import functional