Update app.py

app.py

@@ -62,6 +62,31 @@ def upscale_with_clarity(image_path):
     except Exception as e:
         return None, f"화질 개선 오류: {str(e)}"
 
+def force_white_background(image, threshold=240):
+    """생성된 이미지의 배경을 강제로 완전한 흰색으로 변경"""
+    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)
+        
+        return result_image, "완전한 흰색 배경 강제 적용 완료"
+        
+    except Exception as e:
+        return image, f"배경 흰색 변경 실패: {str(e)}"
+
 def create_coupang_thumbnail_fixed(image_path):
     """고정된 설정으로 쿠팡 썸네일 생성"""
     if not REPLICATE_API_TOKEN:
@@ -82,7 +107,12 @@ def create_coupang_thumbnail_fixed(image_path):
         
         response = requests.get(output)
         if response.status_code == 200:
-            return Image.open(io.BytesIO(response.content)), "썸네일 생성 완료"
+            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}"
         else:
             return None, "썸네일 생성 실패"