Update app.py

app.py

@@ -1,4 +1,5 @@
 import gradio as gr
+from gradio_imageslider import ImageSlider
 import cv2
 import numpy as np
 
@@ -12,17 +13,24 @@ def convert_and_save(image):
     gray_image = convert_to_grayscale(image)
     output_path = "output.png"
     cv2.imwrite(output_path, gray_image)
-    return gray_image, output_path
+    
+    # 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=["image", "file"],
+    outputs=[slider, "file"],
     examples=["1.png", "2.jpg", "3.png"],  # 샘플 이미지 추가
     title="이미지 흑백 변환기",
     description="이미지를 업로드하면 흑백으로 변환하고, PNG 파일으로 다운로드할 수 있습니다."
 )
 
 if __name__ == "__main__":
-    iface.launch()
+    iface.launch()