Update app.py

app.py

@@ -3,9 +3,8 @@ import cv2
 import numpy as np
 from PIL import Image
 from io import BytesIO
-import os
 
-def process_image(file, 
+def process_image(file_path, 
                   convert_bw, 
                   denoise, 
                   sharpen, 
@@ -16,12 +15,11 @@ def process_image(file,
     """
     이미지 처리 함수
     """
-    if file is None:
+    if file_path is None:
         return None, None
 
-    # Read the binary data
-    image = Image.open(BytesIO(file)).convert("RGB")
-    original_image = image.copy()
+    # Open the image
+    image = Image.open(file_path).convert("RGB")
 
     # Convert PIL Image to OpenCV format
     img = np.array(image)
@@ -34,7 +32,7 @@ def process_image(file,
     # 샤프닝
     if sharpen:
         kernel = np.array([[0, -1, 0],
-                           [-1, 5,-1],
+                           [-1, 5, -1],
                            [0, -1, 0]])
         img = cv2.filter2D(img, -1, kernel)
 
@@ -69,36 +67,29 @@ def process_image(file,
     transformed_image = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
     transformed_image_pil = Image.fromarray(transformed_image)
 
-    # Prepare the download file
+    # Save the transformed image to BytesIO
     buf = BytesIO()
     transformed_image_pil.save(buf, format="JPEG")
-    buf.seek(0)
+    byte_data = buf.getvalue()
 
-    # Set the file name
-    # Extract original file name from the uploaded file
-    # Since gr.File with type='binary' provides bytes, the filename needs to be extracted separately
-    # For simplicity, set a default filename
-    original_filename = "transformed_image.jpg"
-    buf.name = original_filename
+    return transformed_image_pil, byte_data
 
-    return transformed_image_pil, buf
-
-def blend_images(original, transformed, alpha):
+def blend_images(original_image, transformed_image, alpha):
     """
     두 이미지를 알파 값에 따라 혼합하는 함수
     """
-    if original is None or transformed is None:
+    if original_image is None or transformed_image is None:
         return None
-    blended = Image.blend(original, transformed, alpha)
+    blended = Image.blend(original_image, transformed_image, alpha)
     return blended
 
-def get_original_image(file):
+def get_original_image(file_path):
     """
-    Get the original image as PIL.Image
+    원본 이미지를 PIL.Image 형식으로 반환하는 함수
     """
-    if file is None:
+    if file_path is None:
         return None
-    image = Image.open(BytesIO(file)).convert("RGB")
+    image = Image.open(file_path).convert("RGB")
     return image
 
 # Gradio 인터페이스 구성
@@ -108,7 +99,7 @@ def create_interface():
         
         with gr.Row():
             with gr.Column(scale=1):
-                input_image = gr.File(type="binary", label="원본 이미지 업로드")
+                input_image = gr.File(type="filepath", label="원본 이미지 업로드")
                 convert_bw = gr.Checkbox(label="흑백으로 변환", value=True)
                 denoise = gr.Checkbox(label="노이즈 제거", value=False)
                 sharpen = gr.Checkbox(label="샤프닝", value=False)
@@ -118,13 +109,13 @@ def create_interface():
                 saturation = gr.Slider(label="채도 조정", minimum=0.0, maximum=2.0, step=0.1, value=1.0)
                 submit = gr.Button("변환하기")
                 
-                # State를 사용하여 원본 및 변환된 이미지 저장
+                # State를 사용하여 원본 이미지 저장
                 original_state = gr.State()
                 transformed_state = gr.State()
             
             with gr.Column(scale=1):
                 transformed_image = gr.Image(type="pil", label="변환된 이미지")
-                download_btn = gr.File(label="JPG로 다운로드")
+                download_btn = gr.File(label="JPG로 다운로드", type="binary")
         
         # 이미지 처리 결과를 State에 저장하고 변환된 이미지 및 다운로드 파일 표시
         submit.click(
@@ -133,7 +124,7 @@ def create_interface():
             outputs=[transformed_image, download_btn]
         )
         
-        # 원본 이미지를 상태에 저장
+        # 원본 이미지를 State에 저장
         input_image.change(
             fn=get_original_image,
             inputs=input_image,