Update app.py

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
 import cv2
 import numpy as np
+from PIL import Image
 from gradio_imageslider import ImageSlider
 
 def convert_to_grayscale(image):
@@ -11,16 +12,21 @@ def convert_to_grayscale(image):
     return gray_image_bgr
 
 def convert_and_save(image):
-    # 원본 이미지를 복사
-    original_image = image.copy()
+    # 원본 이미지를 복사하여 PIL 형식으로 변환
+    original_image = Image.fromarray(cv2.cvtColor(image, cv2.COLOR_BGR2RGB))
+    
     # 이미지를 흑백으로 변환
     gray_image = convert_to_grayscale(image)
+    gray_image_pil = Image.fromarray(cv2.cvtColor(gray_image, cv2.COLOR_BGR2RGB))
+    
     # 출력 파일 경로 설정
     output_path = "output.jpg"
     # 흑백 이미지 저장
-    cv2.imwrite(output_path, gray_image)
-    # 슬라이드에 사용할 두 이미지 반환
-    return (original_image, gray_image), output_path
+    gray_image_bgr = cv2.cvtColor(gray_image_pil, cv2.COLOR_RGB2BGR)
+    cv2.imwrite(output_path, gray_image_bgr)
+    
+    # 슬라이드에 사용할 두 이미지 반환 (PIL 형식)
+    return (original_image, gray_image_pil), output_path
 
 # Gradio 인터페이스 정의
 with gr.Blocks() as demo:
@@ -32,7 +38,7 @@ with gr.Blocks() as demo:
         file_output = gr.File(label="다운로드 가능한 JPG 파일")
     
     with gr.Row():
-        slider_output = ImageSlider(label="변경 전후 비교", type="numpy")
+        slider_output = ImageSlider(label="변경 전후 비교")
     
     # 버튼을 눌렀을 때 함수 실행
     upload_btn = gr.Button("변환하기")