Update app.py

app.py

@@ -37,17 +37,15 @@ def apply_filter(image, filter_type, intensity):
         enhanced_image = cv2.cvtColor(np.array(image_pil), cv2.COLOR_RGB2BGR)
         return enhanced_image
     elif filter_type == "Warm Tone":
-        increase_red = np.array([[1.0 + 0.2 * normalized_intensity, 0.0, 0.0],
-                                 [0.0, 1.0, 0.0],
-                                 [0.0, 0.0, 1.0 - 0.2 * normalized_intensity]])
-        warm_image = cv2.transform(image, increase_red)
-        return warm_image
+        # 따뜻한 톤 적용
+        warm_image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
+        warm_image = cv2.applyColorMap(warm_image, cv2.COLORMAP_AUTUMN)
+        return cv2.cvtColor(warm_image, cv2.COLOR_RGB2BGR)
     elif filter_type == "Cold Tone":
-        increase_blue = np.array([[1.0 - 0.2 * normalized_intensity, 0.0, 0.0],
-                                  [0.0, 1.0, 0.0],
-                                  [0.0, 0.0, 1.0 + 0.2 * normalized_intensity]])
-        cold_image = cv2.transform(image, increase_blue)
-        return cold_image
+        # 차가운 톤 적용
+        cold_image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
+        cold_image = cv2.applyColorMap(cold_image, cv2.COLORMAP_WINTER)
+        return cv2.cvtColor(cold_image, cv2.COLOR_RGB2BGR)
     elif filter_type == "High-Key":
         high_key = cv2.convertScaleAbs(image, alpha=1.0 + 0.2 * normalized_intensity, beta=30)
         return high_key
@@ -62,14 +60,14 @@ def apply_filter(image, filter_type, intensity):
 
 def convert_to_grayscale(image):
     gray_image = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
-    return gray_image
+    return cv2.cvtColor(gray_image, cv2.COLOR_GRAY2BGR)  # 필터 적용 후 일관성을 위해 BGR로 변환
 
 def convert_and_save(image, filter_type, intensity):
     filtered_image = apply_filter(image, filter_type, intensity)
     
     # numpy array를 PIL 이미지로 변환
     original_image_pil = Image.fromarray(cv2.cvtColor(np.array(image), cv2.COLOR_RGB2BGR))
-    filtered_image_pil = Image.fromarray(cv2.cvtColor(filtered_image, cv2.COLOR_RGB2BGR))
+    filtered_image_pil = Image.fromarray(filtered_image)
 
     return original_image_pil, filtered_image_pil  # 원본과 필터 적용된 이미지를 반환