import gradio as gr
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):
    # 이미지를 흑백으로 변환하고, 다운로드 가능한 JPG 파일로 저장
    gray_image = convert_to_grayscale(image)
    output_path = "output.jpg"
    cv2.imwrite(output_path, gray_image)
    return gray_image, output_path

# Gradio 인터페이스 정의
iface = gr.Interface(
    fn=convert_and_save,
    inputs="image",
    outputs=["image", "file"],
    title="이미지 흑백 변환기",
    description="이미지를 업로드하면 흑백으로 변환하고, JPG 파일로 다운로드할 수 있습니다."
)

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