Update app.py

app.py

@@ -17,13 +17,10 @@ def process_image(file,
     이미지 처리 함수
     """
     if file is None:
-        return None, None, None, None
+        return None, None
 
-    # Extract original filename
-    original_filename = os.path.basename(file.name)
-
-    # Open the image
-    image = Image.open(file).convert("RGB")
+    # Read the binary data
+    image = Image.open(BytesIO(file)).convert("RGB")
     original_image = image.copy()
 
     # Convert PIL Image to OpenCV format
@@ -70,24 +67,21 @@ def process_image(file,
 
     # 변환된 이미지를 PIL 형식으로 변환
     transformed_image = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
-    transformed_image = Image.fromarray(transformed_image)
+    transformed_image_pil = Image.fromarray(transformed_image)
 
     # Prepare the download file
     buf = BytesIO()
-    transformed_image.save(buf, format="JPEG")
+    transformed_image_pil.save(buf, format="JPEG")
     buf.seek(0)
 
-    # 새로운 파일명 설정 (예: original_transformed.jpg)
-    if '.' in original_filename:
-        base, ext = original_filename.rsplit('.', 1)
-    else:
-        base, ext = original_filename, 'jpg'
-    new_filename = f"{base}_transformed.{ext}"
-
-    # gr.File을 위한 파일 이름 설정
-    buf.name = new_filename
+    # 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, original_image, transformed_image, buf
+    return transformed_image_pil, buf
 
 def blend_images(original, transformed, alpha):
     """
@@ -98,6 +92,15 @@ def blend_images(original, transformed, alpha):
     blended = Image.blend(original, transformed, alpha)
     return blended
 
+def get_original_image(file):
+    """
+    Get the original image as PIL.Image
+    """
+    if file is None:
+        return None
+    image = Image.open(BytesIO(file)).convert("RGB")
+    return image
+
 # Gradio 인터페이스 구성
 def create_interface():
     with gr.Blocks() as demo:
@@ -105,7 +108,7 @@ def create_interface():
         
         with gr.Row():
             with gr.Column(scale=1):
-                input_image = gr.File(type="file", label="원본 이미지 업로드")
+                input_image = gr.File(type="binary", label="원본 이미지 업로드")
                 convert_bw = gr.Checkbox(label="흑백으로 변환", value=True)
                 denoise = gr.Checkbox(label="노이즈 제거", value=False)
                 sharpen = gr.Checkbox(label="샤프닝", value=False)
@@ -127,7 +130,14 @@ def create_interface():
         submit.click(
             fn=process_image, 
             inputs=[input_image, convert_bw, denoise, sharpen, gamma, brightness, contrast, saturation],
-            outputs=[transformed_image, original_state, transformed_state, download_btn]
+            outputs=[transformed_image, download_btn]
+        )
+        
+        # 원본 이미지를 상태에 저장
+        input_image.change(
+            fn=get_original_image,
+            inputs=input_image,
+            outputs=original_state
         )
         
         gr.Markdown("### 이미지 비교")
@@ -139,7 +149,7 @@ def create_interface():
         # 슬라이더 조작 시 이미지 혼합
         alpha_slider.change(
             fn=blend_images,
-            inputs=[original_state, transformed_state, alpha_slider],
+            inputs=[original_state, transformed_image, alpha_slider],
             outputs=blended_image
         )