Update app.py

app.py

@@ -81,9 +81,44 @@ def enhance_image_quality(image):
     
     return image
 
-def rotate_image_slightly(image, angle=5):
-    """이미지를 약간 회전시켜 아이템 위너 회피"""
-    return image.rotate(angle, expand=True, fillcolor=(255, 255, 255))
+def change_product_angle_with_ai(image_path, angle_option):
+    """AI를 이용한 상품 각도/방향 변경으로 아이템 위너 회피"""
+    if not REPLICATE_API_TOKEN:
+        return None, "Replicate API 토큰이 설정되지 않았습니다."
+    
+    try:
+        # 각도 변경 프롬프트 옵션
+        angle_prompts = {
+            "왼쪽 45도": "Change the product angle to show it from 45-degree left side view, professional product photography, white background",
+            "오른쪽 45도": "Change the product angle to show it from 45-degree right side view, professional product photography, white background", 
+            "상단 앵글": "Change to top-down angle view of the product, bird's eye view, professional product photography, white background",
+            "3/4 뷰": "Change to 3/4 angle view of the product, three-quarter perspective, professional product photography, white background",
+            "정면에서 측면": "Change from front view to side view of the product, profile angle, professional product photography, white background",
+            "측면에서 정면": "Change from side view to front view of the product, frontal angle, professional product photography, white background"
+        }
+        
+        selected_prompt = angle_prompts.get(angle_option, angle_prompts["3/4 뷰"])
+        
+        # Replicate API 호출
+        output = replicate.run(
+            "black-forest-labs/flux-kontext-pro",
+            input={
+                "prompt": selected_prompt,
+                "input_image": open(image_path, "rb"),
+                "output_format": "jpg"
+            }
+        )
+        
+        # 결과 이미지 다운로드
+        response = requests.get(output)
+        if response.status_code == 200:
+            angled_image = Image.open(io.BytesIO(response.content))
+            return angled_image, f"상품 각도 변경 완료: {angle_option}"
+        else:
+            return None, "각도 변경 실패"
+            
+    except Exception as e:
+        return None, f"각도 변경 중 오류: {str(e)}"
 
 def check_white_background_ratio(image, threshold=0.85):
     """흰색 배경 비율 확인 (85% 규칙)"""
@@ -134,7 +169,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_rotation, rotation_angle, ai_enhancement, prompt_option):
+def create_coupang_thumbnail(image, apply_shadow, apply_angle_change, angle_option, ai_enhancement, prompt_option):
     """쿠팡 썸네일 생성 메인 함수"""
     if image is None:
         return None, "이미지를 업로드해주세요."
@@ -143,7 +178,21 @@ def create_coupang_thumbnail(image, apply_shadow, apply_rotation, rotation_angle
         result_image = image.copy()
         status_messages = []
         
-        # 1단계: AI 향상 (선택사항)
+        # 1단계: 상품 각도 변경 (아이템 위너 회피)
+        if apply_angle_change and REPLICATE_API_TOKEN:
+            temp_path = upload_image_to_temp_url(result_image)
+            if temp_path:
+                angled_img, angle_msg = change_product_angle_with_ai(temp_path, angle_option)
+                if angled_img:
+                    result_image = angled_img
+                    status_messages.append(f"✅ 상품 각도 변경: {angle_msg}")
+                else:
+                    status_messages.append(f"❌ 각도 변경: {angle_msg}")
+                os.unlink(temp_path)  # 임시 파일 삭제
+        elif apply_angle_change and not REPLICATE_API_TOKEN:
+            status_messages.append("⚠️ 각도 변경에는 Replicate API 토큰이 필요합니다")
+        
+        # 2단계: AI 품질 향상 (선택사항)
         if ai_enhancement and REPLICATE_API_TOKEN:
             temp_path = upload_image_to_temp_url(result_image)
             if temp_path:
@@ -155,15 +204,10 @@ def create_coupang_thumbnail(image, apply_shadow, apply_rotation, rotation_angle
                     status_messages.append(f"❌ AI 향상: {ai_msg}")
                 os.unlink(temp_path)  # 임시 파일 삭제
         
-        # 2단계: 화질 향상
+        # 3단계: 화질 향상
         result_image = enhance_image_quality(result_image)
         status_messages.append("✅ 화질 향상 완료")
         
-        # 3단계: 각도 변경 (아이템 위너 회피)
-        if apply_rotation:
-            result_image = rotate_image_slightly(result_image, rotation_angle)
-            status_messages.append(f"✅ 각도 변경: {rotation_angle}도 회전")
-        
         # 4단계: 흰 배경 + 그림자 효과
         if apply_shadow:
             result_image = create_white_background_shadow(result_image)
@@ -202,7 +246,7 @@ def create_interface():
         ## 💡 적용되는 쿠팡 규정 대응 전략:
         - ✅ **흰 배경 + 85% 규칙** 완벽 준수
         - ✅ **AI 마법 프롬프트**로 화질 향상 
-        - ✅ **각도 변경**으로 아이템 위너 회피
+        - ✅ **상품 각도 변경**으로 아이템 위너(중복) 회피
         - ✅ **그림자 효과**로 자연스러운 입체감 연출
         """)
         
@@ -224,17 +268,16 @@ def create_interface():
                         value=True,
                         info="자연스러운 입체감 연출"
                     )
-                    apply_rotation = gr.Checkbox(
-                        label="각도 변경 적용", 
+                    apply_angle_change = gr.Checkbox(
+                        label="상품 각도 변경 (아이템 위너 회피)", 
                         value=True,
-                        info="아이템 위너 회피"
+                        info="AI로 상품의 각도/방향을 변경하여 중복 회피"
                     )
-                    rotation_angle = gr.Slider(
-                        minimum=-15, 
-                        maximum=15, 
-                        value=5, 
-                        step=1,
-                        label="회전 각도",
+                    angle_option = gr.Dropdown(
+                        choices=["왼쪽 45도", "오른쪽 45도", "상단 앵글", "3/4 뷰", "정면에서 측면", "측면에서 정면"],
+                        value="3/4 뷰",
+                        label="변경할 각도",
+                        info="상품을 어떤 각도로 변경할지 선택",
                         visible=True
                     )
                 
@@ -270,9 +313,18 @@ def create_interface():
         ---
         ## 📋 사용 방법
         1. **배경이 제거된 상품 이미지**를 업로드하세요
-        2. 원하는 **효과 옵션**을 선택하세요
-        3. **AI 향상**을 사용하려면 Replicate API 토큰을 환경변수로 설정하세요
-        4. **'쿠팡 썸네일 생성'** 버튼을 클릭하세요
+        2. **상품 각도 변경 옵션**을 선택하여 아이템 위너(중복)를 회피하세요
+        3. 원하는 **효과 옵션**을 선택하세요
+        4. **AI 향상**을 사용하려면 Replicate API 토큰을 환경변수로 설정하세요
+        5. **'쿠팡 썸네일 생성'** 버튼을 클릭하세요
+        
+        ## 🎯 상품 각도 변경 옵션 설명
+        - **왼쪽 45도**: 상품을 왼쪽에서 45도 각도로 촬영한 느낌
+        - **오른쪽 45도**: 상품을 오른쪽에서 45도 각도로 촬영한 느낌  
+        - **상단 앵글**: 위에서 내려다본 탑뷰 각도
+        - **3/4 뷰**: 가장 일반적인 상품 촬영 각도 (추천)
+        - **정면에서 측면**: 정면 이미지를 측면 각도로 변경
+        - **측면에서 정면**: 측면 이미지를 정면 각도로 변경
         
         ## 🔑 API 설정 (선택사항)
         AI 향상 기능을 사용하려면 환경변수를 설정하세요:
@@ -282,18 +334,18 @@ def create_interface():
         """)
         
         # 이벤트 처리
-        def update_rotation_visibility(apply_rot):
-            return gr.update(visible=apply_rot)
+        def update_angle_visibility(apply_angle):
+            return gr.update(visible=apply_angle)
         
-        apply_rotation.change(update_rotation_visibility, apply_rotation, rotation_angle)
+        apply_angle_change.change(update_angle_visibility, apply_angle_change, angle_option)
         
         process_btn.click(
             create_coupang_thumbnail,
             inputs=[
                 input_image, 
                 apply_shadow, 
-                apply_rotation, 
-                rotation_angle, 
+                apply_angle_change, 
+                angle_option, 
                 ai_enhancement, 
                 prompt_option
             ],