Update app.py

app.py

@@ -1,6 +1,5 @@
 import gradio as gr
-from PIL import Image, ImageEnhance, ImageOps
-import numpy as np
+from PIL import Image, ImageEnhance, ImageOps, ImageFilter
 
 def convert_to_moody_bw(image):
     # 이미지를 흑백으로 변환
@@ -15,26 +14,31 @@ def convert_to_moody_bw(image):
     bw_image = enhancer.enhance(0.9)
     
     # 약간의 흐림 효과 추가
-    bw_image = bw_image.filter(Image.Filter.GaussianBlur(radius=1))
+    bw_image = bw_image.filter(ImageFilter.GaussianBlur(radius=1))
     
     return bw_image
 
-def download_jpg(image):
-    # 이미지를 JPG 형식으로 저장
-    return image.convert("RGB")
+def convert_and_prepare_download(image):
+    bw_image = convert_to_moody_bw(image)
+    # JPG 형식으로 변환
+    jpg_image = bw_image.convert("RGB")
+    return bw_image, jpg_image
 
-# Gradio 인터페이스 설정
 with gr.Blocks() as demo:
     gr.Markdown("## 분위기 있는 흑백 사진 변환기")
+    
     with gr.Row():
         with gr.Column():
             input_image = gr.Image(type="pil", label="원본 이미지 업로드")
             convert_button = gr.Button("변환하기")
         with gr.Column():
             output_image = gr.Image(type="pil", label="변환된 이미지")
-            download_button = gr.Button("JPG로 다운로드")
+            download_link = gr.File(label="JPG로 다운로드")
     
-    convert_button.click(fn=convert_to_moody_bw, inputs=input_image, outputs=output_image)
-    download_button.click(fn=download_jpg, inputs=output_image, outputs=gr.File())
-
-demo.launch()
+    convert_button.click(
+        fn=convert_and_prepare_download, 
+        inputs=input_image, 
+        outputs=[output_image, download_link]
+    )
+    
+demo.launch()