Update app.py

app.py

@@ -1,33 +1,58 @@
 import gradio as gr
-from PIL import Image
+from PIL import Image, ImageEnhance, ImageOps, ImageFilter
 
-def convert_to_grayscale(image):
-    # 이미지를 흑백으로 변환
-    gray_image = image.convert("L")
-    return gray_image
+# 필터 효과 정의
+def apply_filter(image, filter_name, intensity):
+    if filter_name == "Soft Glow":
+        return image.filter(ImageFilter.GaussianBlur(intensity))
+    elif filter_name == "Portrait Enhancer":
+        enhancer = ImageEnhance.Brightness(image)
+        return enhancer.enhance(1 + (intensity / 100))
+    elif filter_name == "Warm Tone":
+        red, green, blue = image.split()
+        red = red.point(lambda p: min(255, p + intensity))
+        return Image.merge("RGB", (red, green, blue))
+    elif filter_name == "Cold Tone":
+        red, green, blue = image.split()
+        blue = blue.point(lambda p: min(255, p + intensity))
+        return Image.merge("RGB", (red, green, blue))
+    elif filter_name == "High-Key":
+        enhancer = ImageEnhance.Brightness(image)
+        return enhancer.enhance(1.5 + (intensity / 200))
+    elif filter_name == "Low-Key":
+        enhancer = ImageEnhance.Contrast(image)
+        return enhancer.enhance(0.8 - (intensity / 200))
+    elif filter_name == "Haze":
+        return image.filter(ImageFilter.BLUR)
+    elif filter_name == "Monochrome":
+        return ImageOps.grayscale(image)
+    else:
+        return image
 
 # Gradio UI 구성
 with gr.Blocks() as app:
-    gr.Markdown("# 이미지를 흑백으로 변환")
+    gr.Markdown("# 인물 사진 필터 적용 앱")
     
     with gr.Row():
-        input_image = gr.Image(label="원본 이미지", type="pil")
-        output_image = gr.Image(label="흑백 이미지", type="pil")
+        with gr.Column():
+            input_image = gr.Image(label="원본 이미지", type="pil")
+            filter_name = gr.Dropdown(
+                ["Soft Glow", "Portrait Enhancer", "Warm Tone", "Cold Tone", 
+                 "High-Key", "Low-Key", "Haze", "Monochrome"],
+                label="필터 선택"
+            )
+            intensity = gr.Slider(0, 100, step=1, label="필터 강도")
+            apply_button = gr.Button("필터 적용")
         
-    convert_button = gr.Button("변환하기")
-    download_button = gr.Button("JPG로 다운로드")
+        with gr.Column():
+            output_image = gr.Image(label="필터 적용 이미지", type="pil")
+            comparison = gr.Slider([("Before", input_image), ("After", output_image)], label="변경 전후 비교")
     
-    convert_button.click(
-        fn=convert_to_grayscale,
-        inputs=input_image,
+    apply_button.click(
+        fn=apply_filter,
+        inputs=[input_image, filter_name, intensity],
         outputs=output_image
     )
-    
-    download_button.click(
-        fn=lambda img: img.save("converted_image.jpg", "JPEG") or "converted_image.jpg",
-        inputs=output_image,
-        outputs=gr.File(label="JPG 파일 다운로드")
-    )
 
 # 실행
 if __name__ == "__main__":