app.py

import gradio as gr
from gradio_imageslider import ImageSlider
import cv2
import numpy as np

def convert_to_grayscale(image):
    # 이미지를 흑백으로 변환
    gray_image = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
    return gray_image

def convert_and_save(image):
    # 이미지를 흑백으로 변환하고, 다운로드 가능한 PNG 파일로 저장
    gray_image = convert_to_grayscale(image)
    output_path = "output.png"
    cv2.imwrite(output_path, gray_image)
    
    # Gradio는 RGB 이미지를 기대하므로 흑백 이미지를 RGB로 변환
    gray_rgb = cv2.cvtColor(gray_image, cv2.COLOR_GRAY2RGB)
    original_rgb = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
    
    # 변경 전후 이미지를 슬라이드로 반환
    return [gray_rgb, original_rgb], output_path

# Gradio 인터페이스 정의
slider = ImageSlider(label="변환 전후 이미지")
iface = gr.Interface(
    fn=convert_and_save,
    inputs="image",
    outputs=[slider, "file"],
    examples=["1.png", "2.jpg", "3.png"],  # 샘플 이미지 추가
    title="이미지 흑백 변환기",
    description="이미지를 업로드하면 흑백으로 변환하고, PNG 파일으로 다운로드할 수 있습니다."
)

if __name__ == "__main__":
    iface.launch()