app.py

import gradio as gr
from PIL import Image, ImageEnhance, ImageFilter

# 필터 적용 함수
def apply_filter(image, filter_type, intensity):
    image = image.convert("RGB")
    
    if filter_type == "Soft Glow":
        return image.filter(ImageFilter.GaussianBlur(radius=intensity))
    elif filter_type == "Portrait Enhancer":
        enhancer = ImageEnhance.Color(image)
        return enhancer.enhance(1 + intensity / 10)
    elif filter_type == "Warm Tone":
        r, g, b = image.split()
        r = r.point(lambda i: min(255, i + intensity * 10))
        return Image.merge("RGB", (r, g, b))
    elif filter_type == "Cold Tone":
        r, g, b = image.split()
        b = b.point(lambda i: min(255, i + intensity * 10))
        return Image.merge("RGB", (r, g, b))
    elif filter_type == "High-Key":
        enhancer = ImageEnhance.Brightness(image)
        return enhancer.enhance(1 + intensity / 10)
    elif filter_type == "Low-Key":
        enhancer = ImageEnhance.Brightness(image)
        return enhancer.enhance(1 - intensity / 10)
    elif filter_type == "Haze":
        return image.filter(ImageFilter.BLUR)
    elif filter_type == "Monochrome":
        return image.convert("L").convert("RGB")
    else:
        return image

# 초기값 설정 함수
def set_initial_intensity(filter_type):
    # 필터별 초기값 지정
    initial_values = {
        "Soft Glow": 3,
        "Portrait Enhancer": 2,
        "Warm Tone": 5,
        "Cold Tone": 5,
        "High-Key": 3,
        "Low-Key": 3,
        "Haze": 2,
        "Monochrome": 0
    }
    return initial_values.get(filter_type, 0)

# Gradio UI 구성
with gr.Blocks() as demo:
    gr.Markdown("# 인물 사진 필터 적용기")
    
    with gr.Row():
        with gr.Column():
            image_input = gr.Image(type="pil", label="이미지 업로드")
            filter_type = gr.Dropdown(
                choices=[
                    "Soft Glow", "Portrait Enhancer", "Warm Tone", 
                    "Cold Tone", "High-Key", "Low-Key", "Haze", "Monochrome"
                ], 
                value="Soft Glow",  # 기본값 설정
                label="필터 선택"
            )
            intensity = gr.Slider(
                0, 10, value=3, step=1, label="필터 강도"
            )
        with gr.Column():
            filtered_image = gr.Image(type="pil", label="필터 적용 이미지")
    
    # 필터 적용 및 결과 연결
    def process_image(image, filter_type, intensity):
        if image is None:
            return None
        filtered_image = apply_filter(image, filter_type, intensity)
        return filtered_image
    
    def update_intensity_on_filter_change(filter_type):
        return set_initial_intensity(filter_type)
    
    # UI 이벤트 연결
    filter_type.change(
        update_intensity_on_filter_change,
        inputs=filter_type,
        outputs=intensity
    )
    
    image_input.change(
        process_image,
        inputs=[image_input, filter_type, intensity],
        outputs=[filtered_image]
    )
    
    intensity.change(
        process_image,
        inputs=[image_input, filter_type, intensity],
        outputs=[filtered_image]
    )

# 앱 실행
if __name__ == "__main__":
    demo.launch()