Update app.py

app.py

@@ -62,44 +62,46 @@ def upscale_with_clarity(image_path):
     except Exception as e:
         return None, f"화질 개선 오류: {str(e)}"
 
-def force_white_background(image, threshold=240):
-    """생성된 이미지의 배경을 강제로 완전한 흰색으로 변경"""
+def change_product_angle(image_path):
+    """상품 각도를 틀어서 변경"""
+    if not REPLICATE_API_TOKEN:
+        return None, "Replicate API 토큰이 필요합니다."
+    
     try:
-        # RGB 모드로 변환
-        if image.mode != 'RGB':
-            image = image.convert('RGB')
-        
-        # 이미지를 numpy 배열로 변환
-        img_array = np.array(image)
-        
-        # 배경으로 판단되는 영역 찾기 (밝은 색상 영역)
-        # 각 픽셀의 RGB 값이 모두 threshold 이상인 경우 배경으로 판단
-        background_mask = np.all(img_array >= threshold, axis=-1)
-        
-        # 배경 영역을 완전한 흰색(255, 255, 255)으로 변경
-        img_array[background_mask] = [255, 255, 255]
-        
-        # numpy 배열을 다시 PIL 이미지로 변환
-        result_image = Image.fromarray(img_array)
+        # 각도 변경 프롬프트 (배경은 신경쓰지 않고 각도만 집중)
+        angle_prompt = "Change product angle to dramatic 3/4 view rotated 45 degrees, dynamic perspective, tilted composition, angled product placement, professional product photography"
         
-        return result_image, "완전한 흰색 배경 강제 적용 완료"
+        output = replicate.run(
+            "black-forest-labs/flux-kontext-pro",
+            input={
+                "prompt": angle_prompt,
+                "input_image": open(image_path, "rb"),
+                "output_format": "jpg"
+            }
+        )
         
+        response = requests.get(output)
+        if response.status_code == 200:
+            return Image.open(io.BytesIO(response.content)), "각도 변경 완료"
+        else:
+            return None, "각도 변경 실패"
+            
     except Exception as e:
-        return image, f"배경 흰색 변경 실패: {str(e)}"
+        return None, f"각도 변경 오류: {str(e)}"
 
-def create_coupang_thumbnail_fixed(image_path):
-    """고정된 설정으로 쿠팡 썸네일 생성"""
+def add_shadow_only(image_path):
+    """배경 제거된 상품에 그림자만 추가 (완전한 흰색 배경)"""
     if not REPLICATE_API_TOKEN:
         return None, "Replicate API 토큰이 필요합니다."
     
     try:
-        # 고정된 프롬프트: 완전한 흰색배경 + 그림자 효과, 85%상품이미지, 각도를 더 틀어서
-        fixed_prompt = "Professional product photography, dramatic 3/4 angle view rotated 45 degrees, dynamic perspective, pure white background #FFFFFF, natural soft shadows on white background, clean white backdrop, high quality e-commerce photo, solid white background, product takes 85% of image space, tilted composition, angled product placement, subtle drop shadow on white surface"
+        # 그림자 추가 프롬프트 (완전한 흰색 배경 + 그림자 효과만)
+        shadow_prompt = "Add natural soft drop shadow to the product on pure white background #FFFFFF, clean white backdrop, subtle shadow on white surface, professional e-commerce photo, solid white background, product takes 85% of image space"
         
         output = replicate.run(
             "black-forest-labs/flux-kontext-pro",
             input={
-                "prompt": fixed_prompt,
+                "prompt": shadow_prompt,
                 "input_image": open(image_path, "rb"),
                 "output_format": "jpg"
             }
@@ -107,17 +109,12 @@ def create_coupang_thumbnail_fixed(image_path):
         
         response = requests.get(output)
         if response.status_code == 200:
-            generated_image = Image.open(io.BytesIO(response.content))
-            
-            # 생성된 이미지의 배경을 강제로 완전한 흰색으로 변경
-            final_image, white_bg_msg = force_white_background(generated_image)
-            
-            return final_image, f"썸네일 생성 완료 + {white_bg_msg}"
+            return Image.open(io.BytesIO(response.content)), "그림자 추가 완료"
         else:
-            return None, "썸네일 생성 실패"
+            return None, "그림자 추가 실패"
             
     except Exception as e:
-        return None, f"썸네일 생성 오류: {str(e)}"
+        return None, f"그림자 추가 오류: {str(e)}"
 
 def process_image(image, remove_bg, upscale_quality):
     """메인 처리 함수"""
@@ -128,7 +125,18 @@ def process_image(image, remove_bg, upscale_quality):
         result_image = image.copy()
         status_messages = []
         
-        # 1단계: 배경 제거 (이미지 생성 전)
+        # 1단계: 상품 각도 변경
+        temp_path = upload_image_to_temp_url(result_image)
+        if temp_path:
+            angled_img, angle_msg = change_product_angle(temp_path)
+            if angled_img:
+                result_image = angled_img
+                status_messages.append(f"✅ {angle_msg}")
+            else:
+                status_messages.append(f"❌ {angle_msg}")
+            os.unlink(temp_path)
+        
+        # 2단계: 배경 제거 (누끼)
         if remove_bg:
             temp_path = upload_image_to_temp_url(result_image)
             if temp_path:
@@ -140,19 +148,18 @@ def process_image(image, remove_bg, upscale_quality):
                     status_messages.append(f"❌ {bg_msg}")
                 os.unlink(temp_path)
         
-        # 2단계: 쿠팡 썸네일 생성 (고정 설정)
+        # 3단계: 그림자만 추가 (완전한 흰색 배경)
         temp_path = upload_image_to_temp_url(result_image)
         if temp_path:
-            final_img, final_msg = create_coupang_thumbnail_fixed(temp_path)
-            if final_img:
-                result_image = final_img
-                status_messages.append(f"✅ {final_msg}")
+            shadow_img, shadow_msg = add_shadow_only(temp_path)
+            if shadow_img:
+                result_image = shadow_img
+                status_messages.append(f"✅ {shadow_msg}")
             else:
-                status_messages.append(f"❌ {final_msg}")
-                return None, "\n".join(status_messages)
+                status_messages.append(f"❌ {shadow_msg}")
             os.unlink(temp_path)
         
-        # 3단계: 화질 개선 (이미지 생성 후)
+        # 4단계: 화질 개선 (이미지 생성 후)
         if upscale_quality:
             temp_path = upload_image_to_temp_url(result_image)
             if temp_path:
@@ -164,7 +171,7 @@ def process_image(image, remove_bg, upscale_quality):
                     status_messages.append(f"❌ {upscale_msg}")
                 os.unlink(temp_path)
         
-        # 4단계: 최종 크기 조정
+        # 5단계: 최종 크기 조정
         result_image = result_image.resize((1000, 1000), Image.Resampling.LANCZOS)
         status_messages.append("✅ 1000x1000 크기 조정 완료")