Update app.py

app.py

@@ -1,34 +1,71 @@
 import gradio as gr
 import cv2
 import numpy as np
+import os
+from PIL import Image
 
-# 이미지 업스케일 함수
 def upscale_image(input_image, upscale_factor):
-    upscale_factor = int(upscale_factor)  # 업스케일 비율을 정수형으로 변환
-    # 입력 이미지를 주어진 업스케일 비율로 크기 변경 (보간법: INTER_CUBIC)
-    output_image = cv2.resize(input_image, None, fx=upscale_factor, fy=upscale_factor, interpolation=cv2.INTER_CUBIC)
-    # 이미지를 PNG 형식으로 저장
-    output_image_bgr = cv2.cvtColor(output_image, cv2.COLOR_RGB2BGR)  # RGB에서 BGR로 변환
-    output_filename = "upscaled_image.png"
-    cv2.imwrite(output_filename, output_image_bgr)
-    return output_filename
+    try:
+        # PIL Image로 변환
+        if isinstance(input_image, np.ndarray):
+            input_image = Image.fromarray(input_image)
+        
+        # 현재 이미지 크기
+        current_width, current_height = input_image.size
+        
+        # 새로운 크기 계산
+        new_width = int(current_width * upscale_factor)
+        new_height = int(current_height * upscale_factor)
+        
+        # 이미지 리사이즈
+        output_image = input_image.resize((new_width, new_height), Image.Resampling.LANCZOS)
+        
+        # 임시 파일 저장을 위한 디렉토리 확인 및 생성
+        if not os.path.exists("temp"):
+            os.makedirs("temp")
+            
+        # 결과 이미지 저장
+        output_path = os.path.join("temp", "upscaled_image.png")
+        output_image.save(output_path, "PNG", quality=100)
+        
+        return output_path
+        
+    except Exception as e:
+        print(f"Error in upscale_image: {str(e)}")
+        raise gr.Error("이미지 처리 중 오류가 발생했습니다. 다시 시도해주세요.")
 
 # 인터페이스 설명
 DESCRIPTION = """
-이미지의 크기와 품질을 향상시켜 보세요! (you can increase the size and quality of your images)
-"""
+# 이미지 업스케일러 (Image Upscaler)
+이미지의 크기와 품질을 향상시켜 보세요!
+You can increase the size and quality of your images!
 
-# 업스케일 레벨 선택 라디오 버튼
-radio_input = gr.Radio(label="업스케일 레벨을 선택해 주세요 (Select Upscaling level)", choices=[2, 4, 6, 8, 10], value=2)
+## 사용 방법 (How to use):
+1. 이미지를 업로드하세요 (Upload an image)
+2. 업스케일 레벨을 선택하세요 (Select upscale level)
+3. Submit 버튼을 클릭하세요 (Click submit)
+"""
 
 # Gradio 인터페이스 정의
 demo = gr.Interface(
     fn=upscale_image,
-    inputs=[gr.Image(label="입력 이미지 (Input Image)", type="numpy"), radio_input],
+    inputs=[
+        gr.Image(label="입력 이미지 (Input Image)", type="pil"),
+        gr.Radio(
+            label="업스케일 레벨 (Upscale Level)", 
+            choices=[2, 4, 6, 8],
+            value=2,
+            type="number"
+        )
+    ],
     outputs=gr.File(label="업스케일된 이미지 다운로드 (Download Upscaled Image)"),
     title="Image Upscaler",
-    description=DESCRIPTION
+    description=DESCRIPTION,
+    allow_flagging="never",
+    examples=[
+        ["example.jpg", 2]  # examples 폴더에 example.jpg 파일을 추가해야 합니다
+    ]
 )
 
 # 인터페이스 실행
-demo.launch(show_api=False)
+demo.launch()