Update app.py

app.py

@@ -66,7 +66,63 @@ def upload_image_to_temp_url(image):
         image.save(tmp_file.name, format='PNG')
         return tmp_file.name
 
-def create_white_background_shadow(image, shadow_intensity=0.3, shadow_offset=(10, 10)):
+def create_perfect_white_background(image, margin=100):
+    """완전한 흰색 배경 생성"""
+    if image.mode != 'RGBA':
+        image = image.convert('RGBA')
+    
+    # 원본 이미지 크기
+    width, height = image.size
+    
+    # 여백을 포함한 새로운 크기
+    new_width = width + margin * 2
+    new_height = height + margin * 2
+    
+    # 완전한 흰색 배경 생성
+    white_background = Image.new('RGB', (new_width, new_height), (255, 255, 255))
+    
+    # 원본 이미지를 중앙에 배치
+    paste_x = (new_width - width) // 2
+    paste_y = (new_height - height) // 2
+    
+    # 알파 채널이 있는 경우 투명 부분 처리
+    if image.mode == 'RGBA':
+        # 투명 배경을 흰색으로 변경
+        white_bg_temp = Image.new('RGBA', image.size, (255, 255, 255, 255))
+        alpha_composite = Image.alpha_composite(white_bg_temp, image)
+        white_background.paste(alpha_composite.convert('RGB'), (paste_x, paste_y))
+    else:
+        white_background.paste(image, (paste_x, paste_y))
+    
+    return white_background
+
+def ensure_white_background_85_percent(image):
+    """85% 흰색 배경 규칙을 확실히 만족시키기"""
+    # 현재 흰색 비율 확인
+    is_compliant, white_ratio = check_white_background_ratio(image)
+    
+    if not is_compliant:
+        # 85% 미달시 이미지를 축소하여 흰색 영역 늘리기
+        current_size = image.size
+        # 현재 흰색 비율에 따라 축소 비율 계산
+        target_ratio = 0.87  # 85%보다 약간 여유있게
+        if white_ratio > 0:
+            scale_factor = min(0.8, (white_ratio / target_ratio) * 0.9)
+        else:
+            scale_factor = 0.7
+        
+        new_size = (int(current_size[0] * scale_factor), int(current_size[1] * scale_factor))
+        resized_image = image.resize(new_size, Image.Resampling.LANCZOS)
+        
+        # 완전한 흰색 배경에 축소된 이미지를 중앙 배치
+        final_image = Image.new('RGB', current_size, (255, 255, 255))
+        paste_x = (current_size[0] - new_size[0]) // 2
+        paste_y = (current_size[1] - new_size[1]) // 2
+        final_image.paste(resized_image, (paste_x, paste_y))
+        
+        return final_image
+    
+    return image
     """흰 배경에 그림자 효과 추가"""
     if image.mode != 'RGBA':
         image = image.convert('RGBA')
@@ -207,7 +263,7 @@ def process_with_ai_enhancement(image_path, prompt_option):
     except Exception as e:
         return None, f"AI 처리 중 오류: {str(e)}"
 
-def create_coupang_thumbnail(image, auto_remove_bg, apply_shadow, apply_angle_change, angle_option, ai_enhancement, prompt_option):
+def create_coupang_thumbnail(image, auto_remove_bg, white_bg_mode, apply_angle_change, angle_option, ai_enhancement, prompt_option):
     """쿠팡 썸네일 생성 메인 함수"""
     if image is None:
         return None, "이미지를 업로드해주세요."
@@ -260,23 +316,30 @@ def create_coupang_thumbnail(image, auto_remove_bg, apply_shadow, apply_angle_ch
         result_image = enhance_image_quality(result_image)
         status_messages.append("✅ 화질 향상 완료")
         
-        # 4단계: 흰 배경 + 그림자 효과
-        if apply_shadow:
+        # 4단계: 배경 처리 (흰색 배경 모드 선택)
+        if white_bg_mode == "완전한 흰색":
+            result_image = create_perfect_white_background(result_image)
+            status_messages.append("✅ 완전한 흰색 배경 적용")
+        elif white_bg_mode == "그림자 효과":
             result_image = create_white_background_shadow(result_image)
-            status_messages.append("✅ 그림자 효과 적용")
-        else:
+            status_messages.append("✅ 그림자 효과가 있는 흰색 배경 적용")
+        else:  # "기본 흰색"
             # 그림자 없이 흰 배경만 적용
             if result_image.mode == 'RGBA':
                 white_bg = Image.new('RGB', result_image.size, (255, 255, 255))
                 white_bg.paste(result_image, mask=result_image.split()[-1] if result_image.mode == 'RGBA' else None)
                 result_image = white_bg
+            status_messages.append("✅ 기본 흰색 배경 적용")
+        
+        # 5단계: 85% 흰색 배경 규칙 강제 적용
+        result_image = ensure_white_background_85_percent(result_image)
         
         # 85% 흰색 배경 규칙 확인
         is_compliant, white_ratio = check_white_background_ratio(result_image)
         if is_compliant:
             status_messages.append(f"✅ 85% 흰색 배경 규칙 준수 ({white_ratio:.1%})")
         else:
-            status_messages.append(f"⚠️ 흰색 배경 비율 부족 ({white_ratio:.1%}) - 85% 미만")
+            status_messages.append(f"⚠️ 흰색 배경 비율: ({white_ratio:.1%}) - 추가 조정 필요")
         
         # 최종 이미지 크기 조정 (쿠팡 권장 사이즈)
         result_image = result_image.resize((1000, 1000), Image.Resampling.LANCZOS)
@@ -323,11 +386,12 @@ def create_interface():
                     )
                 
                 with gr.Group():
-                    gr.Markdown("**🎨 효과 옵션**")
-                    apply_shadow = gr.Checkbox(
-                        label="그림자 효과 적용", 
-                        value=True,
-                        info="자연스러운 입체감 연출"
+                    gr.Markdown("**🎨 배경 옵션**")
+                    white_bg_mode = gr.Radio(
+                        choices=["완전한 흰색", "그림자 효과", "기본 흰색"],
+                        value="완전한 흰색",
+                        label="흰색 배경 모드",
+                        info="완전한 흰색: 100% 순백색 / 그림자 효과: 입체감 연출 / 기본 흰색: 단순 배경"
                     )
                     apply_angle_change = gr.Checkbox(
                         label="상품 각도 변경 (아이템 위너 회피)", 
@@ -409,7 +473,7 @@ def create_interface():
             inputs=[
                 input_image, 
                 auto_remove_bg,
-                apply_shadow, 
+                white_bg_mode,
                 apply_angle_change, 
                 angle_option, 
                 ai_enhancement,