Update app.py

app.py

@@ -69,7 +69,11 @@ def convert_and_save(image, filter_type, intensity):
     original_image_pil = Image.fromarray(cv2.cvtColor(image_cv, cv2.COLOR_BGR2RGB))
     filtered_image_pil = Image.fromarray(cv2.cvtColor(filtered_image, cv2.COLOR_BGR2RGB))
 
-    return original_image_pil, filtered_image_pil  # 원본과 필터 적용된 이미지를 반환
+    # 필터 적용된 이미지를 JPG로 저장
+    output_path = "filtered_image.jpg"
+    filtered_image_pil.save(output_path)
+
+    return original_image_pil, filtered_image_pil, output_path  # 원본, 필터 적용된 이미지와 파일 경로 반환
 
 def get_filter_description(filter_type):
     descriptions = {
@@ -98,6 +102,7 @@ with gr.Blocks() as iface:
 
         with gr.Column():
             slider_output = ImageSlider(label="Before and After", type="pil")
+            download_link = gr.File(label="Download Filtered Image")
 
     filter_input.change(fn=get_filter_description, inputs=filter_input, outputs=description_output)
 
@@ -105,7 +110,7 @@ with gr.Blocks() as iface:
     process_button.click(
         fn=convert_and_save,
         inputs=[image_input, filter_input, intensity_slider],
-        outputs=slider_output
+        outputs=[slider_output, download_link]
     )
 
     iface.title = "인물 사진에 최적화된 필터"