Update app.py

app.py

@@ -31,6 +31,51 @@ BACKGROUND_REMOVER_MODEL = os.getenv("BACKGROUND_REMOVER_MODEL", "")
 # 세션별 이미지 저장용 딕셔너리
 session_images = {}
 
+def adjust_brightness(image, value):
+    """이미지 밝기 조절"""
+    value = float(value - 1) * 100  # 0-2 범위를 -100에서 +100으로 변환
+    hsv = cv2.cvtColor(image, cv2.COLOR_BGR2HSV)
+    h, s, v = cv2.split(hsv)
+    v = cv2.add(v, value)
+    v = np.clip(v, 0, 255)
+    final_hsv = cv2.merge((h, s, v))
+    return cv2.cvtColor(final_hsv, cv2.COLOR_HSV2BGR)
+
+def adjust_contrast(image, value):
+    """이미지 대비 조절"""
+    value = float(value)
+    return np.clip(image * value, 0, 255).astype(np.uint8)
+
+def adjust_saturation(image, value):
+    """이미지 채도 조절"""
+    value = float(value - 1) * 100  # 0-2 범위를 -100에서 +100으로 변환
+    hsv = cv2.cvtColor(image, cv2.COLOR_BGR2HSV)
+    h, s, v = cv2.split(hsv)
+    s = cv2.add(s, value)
+    s = np.clip(s, 0, 255)
+    final_hsv = cv2.merge((h, s, v))
+    return cv2.cvtColor(final_hsv, cv2.COLOR_HSV2BGR)
+
+def adjust_temperature(image, value):
+    """이미지 색온도 조절 (색상 밸런스)"""
+    value = float(value) * 30  # 효과 스케일 조절
+    b, g, r = cv2.split(image)
+    if value > 0:  # 따뜻하게
+        r = cv2.add(r, value)
+        b = cv2.subtract(b, value)
+    else:  # 차갑게
+        r = cv2.add(r, value)
+        b = cv2.subtract(b, value)
+    
+    r = np.clip(r, 0, 255)
+    b = np.clip(b, 0, 255)
+    return cv2.merge([b, g, r])
+
+def get_korean_timestamp():
+    """한국 시간 타임스탬프 생성"""
+    korea_time = datetime.utcnow() + timedelta(hours=9)
+    return korea_time.strftime('%Y%m%d_%H%M%S')
+
 def remove_background(image, session_id):
     """배경 제거 기능 - Replicate API 사용"""
     logger.info(f"세션 {session_id}: 배경 제거 시작")
@@ -91,11 +136,6 @@ def remove_background(image, session_id):
         if os.path.exists(temp_file_path):
             os.unlink(temp_file_path)
 
-def get_korean_timestamp():
-    """한국 시간 타임스탬프 생성"""
-    korea_time = datetime.utcnow() + timedelta(hours=9)
-    return korea_time.strftime('%Y%m%d_%H%M%S')
-
 def handle_upload(image, session_id=None):
     """이미지 업로드 처리 및 배경 제거"""
     if image is None: