Update app.py

app.py

@@ -13,7 +13,30 @@ REPLICATE_API_TOKEN = os.getenv("REPLICATE_API_TOKEN")
 if REPLICATE_API_TOKEN:
     os.environ["REPLICATE_API_TOKEN"] = REPLICATE_API_TOKEN
 
-def upload_image_to_temp_url(image):
+def remove_background_with_ai(image_path):
+    """AI를 이용한 배경 제거"""
+    if not REPLICATE_API_TOKEN:
+        return None, "Replicate API 토큰이 설정되지 않았습니다."
+    
+    try:
+        # Replicate Background Remover API 호출
+        output = replicate.run(
+            "851-labs/background-remover:a029dff38972b5fda4ec5d75d7d1cd25aeff621d2cf4946a41055d7db66b80bc",
+            input={
+                "image": open(image_path, "rb")
+            }
+        )
+        
+        # 결과 이미지 다운로드
+        response = requests.get(output)
+        if response.status_code == 200:
+            bg_removed_image = Image.open(io.BytesIO(response.content))
+            return bg_removed_image, "배경 제거 완료"
+        else:
+            return None, "배경 제거 실패"
+            
+    except Exception as e:
+        return None, f"배경 제거 중 오류: {str(e)}"
     """이미지를 임시 URL로 업로드하는 함수 (실제로는 base64 인코딩)"""
     if image is None:
         return None
@@ -169,7 +192,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, apply_shadow, apply_angle_change, angle_option, ai_enhancement, prompt_option):
+def create_coupang_thumbnail(image, auto_remove_bg, apply_shadow, apply_angle_change, angle_option, ai_enhancement, prompt_option):
     """쿠팡 썸네일 생성 메인 함수"""
     if image is None:
         return None, "이미지를 업로드해주세요."
@@ -178,6 +201,20 @@ def create_coupang_thumbnail(image, apply_shadow, apply_angle_change, angle_opti
         result_image = image.copy()
         status_messages = []
         
+        # 0단계: 자동 배경 제거 (선택사항)
+        if auto_remove_bg and REPLICATE_API_TOKEN:
+            temp_path = upload_image_to_temp_url(result_image)
+            if temp_path:
+                bg_removed_img, bg_msg = remove_background_with_ai(temp_path)
+                if bg_removed_img:
+                    result_image = bg_removed_img
+                    status_messages.append(f"✅ 자동 배경 제거: {bg_msg}")
+                else:
+                    status_messages.append(f"❌ 배경 제거: {bg_msg}")
+                os.unlink(temp_path)  # 임시 파일 삭제
+        elif auto_remove_bg and not REPLICATE_API_TOKEN:
+            status_messages.append("⚠️ 배경 제거에는 Replicate API 토큰이 필요합니다")
+        
         # 1단계: 상품 각도 변경 (아이템 위너 회피)
         if apply_angle_change and REPLICATE_API_TOKEN:
             temp_path = upload_image_to_temp_url(result_image)
@@ -239,11 +276,12 @@ def create_coupang_thumbnail(image, apply_shadow, apply_angle_change, angle_opti
 def create_interface():
     with gr.Blocks(title="🛒 쿠팡 썸네일 생성기", theme=gr.themes.Soft()) as iface:
         gr.Markdown("""
-        # 🛒 쿠팡 썸네일 생성기
+        # 쿠팡 썸네일 생성기 - Gradio 5.0+
         
-        배경이 제거된 상품 이미지를 업로드하면 쿠팡 규정에 맞는 썸네일로 자동 변환해드립니다!
+        일반 상품 이미지를 업로드하면 **자동 배경 제거 → 쿠팡 규정 준수 썸네일**로 한 번에 변환해드립니다!
         
-        ## 💡 적용되는 쿠팡 규정 대응 전략:
+        ## 💡 완전 자동화된 쿠팡 규정 대응:
+        - 🎯 **자동 배경 제거** (AI 기반)
         - ✅ **흰 배경 + 85% 규칙** 완벽 준수
         - ✅ **AI 마법 프롬프트**로 화질 향상 
         - ✅ **상품 각도 변경**으로 아이템 위너(중복) 회피
@@ -254,13 +292,21 @@ def create_interface():
             with gr.Column(scale=1):
                 gr.Markdown("### 📤 이미지 업로드")
                 input_image = gr.Image(
-                    label="배경 제거된 상품 이미지", 
+                    label="상품 이미지 (배경 있어도 OK!)", 
                     type="pil",
                     height=300
                 )
                 
                 gr.Markdown("### ⚙️ 옵션 설정")
                 
+                with gr.Group():
+                    gr.Markdown("**🔧 전처리 옵션**")
+                    auto_remove_bg = gr.Checkbox(
+                        label="자동 배경 제거", 
+                        value=True,
+                        info="AI로 배경을 자동 제거 (Replicate API 필요)"
+                    )
+                
                 with gr.Group():
                     gr.Markdown("**🎨 효과 옵션**")
                     apply_shadow = gr.Checkbox(
@@ -312,11 +358,15 @@ def create_interface():
         gr.Markdown("""
         ---
         ## 📋 사용 방법
-        1. **배경이 제거된 상품 이미지**를 업로드하세요
-        2. **상품 각도 변경 옵션**을 선택하여 아이템 위너(중복)를 회피하세요
-        3. 원하는 **효과 옵션**을 선택하세요
-        4. **AI 향상**을 사용하려면 Replicate API 토큰을 환경변수로 설정하세요
-        5. **'쿠팡 썸네일 생성'** 버튼을 클릭하세요
+        1. **일반 상품 이미지**를 업로드하세요 (배경 있어도 OK!)
+        2. **자동 배경 제거** 옵션을 켜두세요 (기본값)
+        3. **상품 각도 변경 옵션**을 선택하여 아이템 위너(중복)를 회피하세요
+        4. 원하는 **효과 옵션**을 선택하세요
+        5. **Replicate API 토큰**을 환경변수로 설정하세요 (필수)
+        6. **'쿠팡 썸네일 생성'** 버튼을 클릭하세요
+        
+        ## 🔄 완전 자동화 워크플로우
+        **원본 이미지** → **배경 제거** → **각도 변경** → **품질 향상** → **쿠팡 규정 썸네일**
         
         ## 🎯 상품 각도 변경 옵션 설명
         - **왼쪽 45도**: 상품을 왼쪽에서 45도 각도로 촬영한 느낌
@@ -343,6 +393,7 @@ def create_interface():
             create_coupang_thumbnail,
             inputs=[
                 input_image, 
+                auto_remove_bg,
                 apply_shadow, 
                 apply_angle_change, 
                 angle_option,