Update app.py

app.py

@@ -171,42 +171,27 @@ def apply_mask(img: Image.Image, mask_img: Image.Image, defringe: bool = True) -
     return result
 
 
-def adjust_size_to_multiple_of_8(width: int, height: int) -> tuple[int, int]:
-    """이미지 크기를 8의 배수로 조정하는 함수"""
-    new_width = ((width + 7) // 8) * 8
-    new_height = ((height + 7) // 8) * 8
-    return new_width, new_height
-
 def calculate_dimensions(aspect_ratio: str, base_size: int = 512) -> tuple[int, int]:
     """선택된 비율에 따라 이미지 크기 계산"""
+    # 기본 크기를 512로 고정하고, 비율에 따라 조정
     if aspect_ratio == "1:1":
-        width, height = base_size, base_size
+        width = height = 512
     elif aspect_ratio == "16:9":
-        width, height = base_size * 16 // 9, base_size
+        width, height = 512, 288  # 16:9 비율 유지하면서 높이 조정
     elif aspect_ratio == "9:16":
-        width, height = base_size, base_size * 16 // 9
+        width, height = 288, 512  # 9:16 비율 유지하면서 너비 조정
     elif aspect_ratio == "4:3":
-        width, height = base_size * 4 // 3, base_size
+        width, height = 512, 384  # 4:3 비율 유지
     else:
-        width, height = base_size, base_size
-        
-    # 8의 배수로 조정
-    return adjust_size_to_multiple_of_8(width, height)
+        width = height = 512
+    
+    return width, height
 
 def generate_background(prompt: str, aspect_ratio: str) -> Image.Image:
     try:
-        # 이미지 크기 계산 및 8의 배수로 조정
+        # 안전한 크기 계산
         width, height = calculate_dimensions(aspect_ratio)
         
-        # 최대 크기 제한 적용
-        max_size = 768
-        if width > max_size or height > max_size:
-            ratio = max_size / max(width, height)
-            width = int(width * ratio)
-            height = int(height * ratio)
-            # 다시 8의 배수로 조정
-            width, height = adjust_size_to_multiple_of_8(width, height)
-        
         print(f"Generating background with size: {width}x{height}")
         
         with timer("Background generation"):
@@ -217,17 +202,40 @@ def generate_background(prompt: str, aspect_ratio: str) -> Image.Image:
                         width=width,
                         height=height,
                         num_inference_steps=8,
-                        guidance_scale=4.0
+                        guidance_scale=4.0,
                     ).images[0]
+                return image
             except Exception as e:
                 print(f"Pipeline error: {str(e)}")
-                return Image.new('RGB', (width, height), 'white')
+                # 에러 발생 시 기본 크기로 재시도
+                try:
+                    with torch.inference_mode():
+                        image = pipe(
+                            prompt=prompt,
+                            width=512,
+                            height=512,
+                            num_inference_steps=8,
+                            guidance_scale=4.0,
+                        ).images[0]
+                        
+                        # 원하는 크기로 리사이즈
+                        if width != 512 or height != 512:
+                            image = image.resize((width, height), Image.LANCZOS)
+                        return image
+                except Exception as e2:
+                    print(f"Fallback generation error: {str(e2)}")
+                    return Image.new('RGB', (width, height), 'white')
                 
-        return image
     except Exception as e:
         print(f"Background generation error: {str(e)}")
         return Image.new('RGB', (512, 512), 'white')
 
+def adjust_size_to_multiple_of_8(width: int, height: int) -> tuple[int, int]:
+    """이미지 크기를 8의 배수로 조정"""
+    new_width = max(8, ((width + 7) // 8) * 8)  # 최소 8픽셀 보장
+    new_height = max(8, ((height + 7) // 8) * 8)  # 최소 8픽셀 보장
+    return new_width, new_height
+
 def create_position_grid():
     return """
     <div class="position-grid" style="display: grid; grid-template-columns: repeat(3, 1fr); gap: 10px; width: 150px; margin: auto;">