Update app.py

app.py

@@ -1,32 +1,29 @@
 import gradio as gr
-import cv2
-import numpy as np
 from PIL import Image
 from gradio_imageslider import ImageSlider
 
 def convert_to_grayscale(image):
-    # 이미지를 흑백으로 변환
-    gray_image = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
-    # 흑백 이미지를 BGR 형식으로 다시 변환하여 3채널 이미지로 만듦
-    gray_image_bgr = cv2.cvtColor(gray_image, cv2.COLOR_GRAY2BGR)
-    return gray_image_bgr
+    """
+    이미지를 흑백으로 변환하고 다시 RGB 형식으로 변환하여 3채널 이미지를 반환합니다.
+    """
+    gray_image = image.convert("L")  # 흑백으로 변환
+    gray_image = gray_image.convert("RGB")  # 다시 RGB로 변환
+    return gray_image
 
 def convert_and_save(image):
-    # 원본 이미지를 복사하여 PIL 형식으로 변환
-    original_image = Image.fromarray(cv2.cvtColor(image, cv2.COLOR_BGR2RGB))
+    """
+    업로드된 이미지를 흑백으로 변환하고, 원본과 변환된 이미지를 슬라이드 형태로 반환하며,
+    변환된 이미지를 JPG 파일로 저장합니다.
+    """
+    original_image = image  # 원본 이미지 (PIL 형식)
+    gray_image = convert_to_grayscale(image)  # 흑백 이미지 (PIL 형식)
     
-    # 이미지를 흑백으로 변환
-    gray_image = convert_to_grayscale(image)
-    gray_image_pil = Image.fromarray(cv2.cvtColor(gray_image, cv2.COLOR_BGR2RGB))
-    
-    # 출력 파일 경로 설정
-    output_path = "output.jpg"
     # 흑백 이미지 저장
-    gray_image_bgr = cv2.cvtColor(gray_image_pil, cv2.COLOR_RGB2BGR)
-    cv2.imwrite(output_path, gray_image_bgr)
+    output_path = "output.jpg"
+    gray_image.save(output_path, "JPEG")
     
-    # 슬라이드에 사용할 두 이미지 반환 (PIL 형식)
-    return (original_image, gray_image_pil), output_path
+    # 슬라이드에 사용할 두 이미지 리스트로 반환
+    return [original_image, gray_image], output_path
 
 # Gradio 인터페이스 정의
 with gr.Blocks() as demo:
@@ -34,7 +31,7 @@ with gr.Blocks() as demo:
     gr.Markdown("이미지를 업로드하면 흑백으로 변환하고, 변경 전후를 슬라이드 형태로 비교할 수 있습니다.")
     
     with gr.Row():
-        image_input = gr.Image(label="이미지 업로드", type="numpy")
+        image_input = gr.Image(label="이미지 업로드", type="pil")
         file_output = gr.File(label="다운로드 가능한 JPG 파일")
     
     with gr.Row():