Update app.py

app.py

@@ -10,6 +10,8 @@ from realesrgan.utils import RealESRGANer
 import torch
 import cv2
 import gradio as gr
+from gradio_imageslider import ImageSlider  # 슬라이더 컴포넌트 추가
+from PIL import Image  # PIL을 사용하여 numpy 이미지를 PIL 이미지로 변환
 
 # 필수 모델 다운로드
 if not os.path.exists('realesr-general-x4v3.pth'):
@@ -33,18 +35,29 @@ upsampler = RealESRGANer(scale=4, model_path=model_path, model=model, tile=0, ti
 
 def upscaler(img, version, scale):
     try:
-        img = cv2.imread(img, cv2.IMREAD_UNCHANGED)
-        if len(img.shape) == 3 and img.shape[2] == 4:
+        # 입력된 img는 파일 경로임
+        image_array = cv2.imread(img, cv2.IMREAD_UNCHANGED)
+        if image_array is None:
+            print("이미지 로드 실패")
+            return None, None
+
+        if len(image_array.shape) == 3 and image_array.shape[2] == 4:
             img_mode = 'RGBA'
-        elif len(img.shape) == 2:
+        elif len(image_array.shape) == 2:
             img_mode = None
-            img = cv2.cvtColor(img, cv2.COLOR_GRAY2BGR)
+            image_array = cv2.cvtColor(image_array, cv2.COLOR_GRAY2BGR)
         else:
             img_mode = None
 
-        h, w = img.shape[0:2]
+        h, w = image_array.shape[0:2]
         if h < 300:
-            img = cv2.resize(img, (w * 2, h * 2), interpolation=cv2.INTER_LANCZOS4)
+            image_array = cv2.resize(image_array, (w * 2, h * 2), interpolation=cv2.INTER_LANCZOS4)
+
+        # 변경 전 원본 이미지를 RGB (또는 RGBA)로 변환
+        if img_mode == 'RGBA':
+            original_rgb = cv2.cvtColor(image_array, cv2.COLOR_BGRA2RGBA)
+        else:
+            original_rgb = cv2.cvtColor(image_array, cv2.COLOR_BGR2RGB)
 
         face_enhancer = GFPGANer(
             model_path=f'{version}.pth', 
@@ -55,20 +68,26 @@ def upscaler(img, version, scale):
         )
 
         try:
-            _, _, output = face_enhancer.enhance(img, has_aligned=False, only_center_face=False, paste_back=True)
+            _, _, output = face_enhancer.enhance(image_array, has_aligned=False, only_center_face=False, paste_back=True)
         except RuntimeError as error:
             print('오류', error)
+            return None, None
 
         try:
             if scale != 2:
                 interpolation = cv2.INTER_AREA if scale < 2 else cv2.INTER_LANCZOS4
-                h, w = img.shape[0:2]
+                h, w = image_array.shape[0:2]
                 output = cv2.resize(output, (int(w * scale / 2), int(h * scale / 2)), interpolation=interpolation)
         except Exception as error:
             print('잘못된 재스케일링 입력.', error)
 
-        output = cv2.cvtColor(output, cv2.COLOR_BGR2RGB)
-        return output
+        output_rgb = cv2.cvtColor(output, cv2.COLOR_BGR2RGB)
+
+        # numpy 이미지를 PIL 이미지로 변환 (슬라이더에서 type="pil" 사용)
+        original_pil = Image.fromarray(original_rgb)
+        output_pil = Image.fromarray(output_rgb)
+        # 변경 전/후 이미지를 튜플로 반환하여 슬라이더에 표시
+        return original_pil, output_pil
     except Exception as error:
         print('전역 예외', error)
         return None, None
@@ -82,7 +101,7 @@ if __name__ == "__main__":
                 gr.Radio(['GFPGANv1.2', 'GFPGANv1.3', 'GFPGANv1.4', 'RestoreFormer'], type="value", label="버전", value="GFPGANv1.4", visible=False),
                 gr.Number(label="재스케일링 계수", value=0, visible=False),
             ], [
-                gr.Image(type="numpy", label="출력"),
+                ImageSlider(label="출력", type="pil")
             ],
             title=title,
             examples=[["예제.png", "GFPGANv1.4", 0]],