Update app.py

app.py

@@ -1,49 +1,32 @@
 import gradio as gr
-from PIL import Image
+import cv2
+import numpy as np
 from gradio_imageslider import ImageSlider
 
 def convert_to_grayscale(image):
-    """
-    이미지를 흑백으로 변환하고 다시 RGB 형식으로 변환하여 3채널 이미지를 반환합니다.
-    """
-    gray_image = image.convert("L")  # 흑백으로 변환
-    gray_image = gray_image.convert("RGB")  # 다시 RGB로 변환
+    # 이미지를 흑백으로 변환
+    gray_image = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
     return gray_image
 
 def convert_and_save(image):
-    """
-    업로드된 이미지를 흑백으로 변환하고, 원본과 변환된 이미지를 슬라이드 형태로 반환하며,
-    변환된 이미지를 JPG 파일로 저장합니다.
-    """
-    original_image = image  # 원본 이미지 (PIL 형식)
-    gray_image = convert_to_grayscale(image)  # 흑백 이미지 (PIL 형식)
+    # 이미지를 흑백으로 변환
+    gray_image = convert_to_grayscale(image)
     
-    # 흑백 이미지 저장
+    # 변환된 이미지를 JPG로 저장
     output_path = "output.jpg"
-    gray_image.save(output_path, "JPEG")
+    cv2.imwrite(output_path, gray_image)
     
-    # 슬라이드에 사용할 두 이미지 리스트로 반환
-    return [original_image, gray_image], output_path
+    # 슬라이드 출력용 리스트 [원본이미지, 흑백이미지] 와 저장된 파일 경로 반환
+    return [image, gray_image], output_path
 
 # Gradio 인터페이스 정의
-with gr.Blocks() as demo:
-    gr.Markdown("# 이미지 흑백 변환기")
-    gr.Markdown("이미지를 업로드하면 흑백으로 변환하고, 변경 전후를 슬라이드 형태로 비교할 수 있습니다.")
-    
-    with gr.Row():
-        image_input = gr.Image(label="이미지 업로드", type="pil")
-        file_output = gr.File(label="다운로드 가능한 JPG 파일")
-    
-    with gr.Row():
-        slider_output = ImageSlider(label="변경 전후 비교")
-    
-    # 버튼을 눌렀을 때 함수 실행
-    upload_btn = gr.Button("변환하기")
-    upload_btn.click(
-        fn=convert_and_save,
-        inputs=image_input,
-        outputs=[slider_output, file_output]
-    )
+iface = gr.Interface(
+    fn=convert_and_save,
+    inputs="image",
+    outputs=[ImageSlider(label="변경 전후 보기"), "file"],
+    title="이미지 흑백 변환기",
+    description="이미지를 업로드하면 흑백으로 변환하고, 결과 이미지를 슬라이드로 비교해볼 수 있으며, JPG 파일도 다운로드할 수 있습니다."
+)
 
 if __name__ == "__main__":
-    demo.launch()
+    iface.launch()