Update app.py

app.py

@@ -196,8 +196,7 @@ def generate_with_images(prompt, images, variation_index=0):
             )
         )
 
-        # .jpg 확장자로 변경
-        with tempfile.NamedTemporaryFile(suffix=".jpg", delete=False) as tmp:
+        with tempfile.NamedTemporaryFile(suffix=".png", delete=False) as tmp:
             temp_path = tmp.name
             result_text = ""
             image_found = False
@@ -211,18 +210,9 @@ def generate_with_images(prompt, images, variation_index=0):
                     logger.info("응답에서 이미지 추출 성공")
             if not image_found:
                 return None, f"API에서 이미지를 생성하지 못했습니다. 응답 텍스트: {result_text}"
-            
-            # 이미지를 열고 JPG 형식으로 저장
             result_img = Image.open(temp_path)
             if result_img.mode == "RGBA":
-                # RGBA 모드는 JPG에서 지원하지 않으므로 RGB로 변환
                 result_img = result_img.convert("RGB")
-                
-            # 이미지를 JPG로 저장하고 다시 로드
-            jpg_path = temp_path + ".jpg"
-            result_img.save(jpg_path, "JPEG", quality=95)
-            result_img = Image.open(jpg_path)
-            
             return result_img, f"이미지가 성공적으로 생성되었습니다. {result_text}"
     except Exception as e:
         logger.exception("이미지 생성 중 오류 발생:")