Update app.py

app.py

@@ -16,12 +16,18 @@ def apply_filter(image, filter_type):
         enhanced = cv2.detailEnhance(image, sigma_s=10, sigma_r=0.15)
         return enhanced
     elif filter_type == "Warm Tone":
-        # 따뜻한 톤 효과
-        warm_image = cv2.applyColorMap(image, cv2.COLORMAP_AUTUMN)
+        # 따뜻한 톤 효과 (빨강 및 노랑 톤을 강조)
+        increase_red = np.array([[1.2, 0.0, 0.0],
+                                 [0.0, 1.0, 0.0],
+                                 [0.0, 0.0, 0.8]])
+        warm_image = cv2.transform(image, increase_red)
         return warm_image
     elif filter_type == "Cold Tone":
-        # 차가운 톤 효과
-        cold_image = cv2.applyColorMap(image, cv2.COLORMAP_WINTER)
+        # 차가운 톤 효과 (파랑 톤을 강조)
+        increase_blue = np.array([[0.8, 0.0, 0.0],
+                                  [0.0, 1.0, 0.0],
+                                  [0.0, 0.0, 1.2]])
+        cold_image = cv2.transform(image, increase_blue)
         return cold_image
     elif filter_type == "High-Key":
         # High-Key 효과 (밝기 증가)
@@ -36,7 +42,6 @@ def apply_filter(image, filter_type):
         haze = cv2.addWeighted(image, 0.7, np.full(image.shape, 255, dtype=np.uint8), 0.3, 0)
         return haze
     else:
-        # 필터를 적용하지 않음
         return image
 
 def convert_to_grayscale(image):
@@ -58,7 +63,7 @@ iface = gr.Interface(
     inputs=[
         "image", 
         gr.Radio(
-            ["None", "Grayscale", "Soft Glow", "Portrait Enhancer", "Warm Tone", "Cold Tone", "High-Key", "Low-Key", "Haze"], 
+            ["Grayscale", "Soft Glow", "Portrait Enhancer", "Warm Tone", "Cold Tone", "High-Key", "Low-Key", "Haze"], 
             label="필터 선택"
         )
     ],