Update app.py

app.py

@@ -96,16 +96,29 @@ def merge_images(background_image, foreground_image):
 title = "Background Removal"
 description = "This is a demo for BRIA RMBG 1.4 using the BRIA RMBG-1.4 image matting model as backbone."
 
-demo = gr.Interface(
-    fn=process,
-    inputs=[
-        gr.Image(type="numpy", label="Image to remove background"),
-        gr.Image(type="pil", label="Optional: Background Image")  # 'optional=True'는 Gradio 버전에 따라 필요 없을 수 있음
-    ],
-    outputs="image",
-    title=title,
-    description=description
-)
 
+def add_blue_background(image):
+    # 배경 제거된 이미지에 푸른색 배경을 추가하는 함수
+    blue_background = Image.new("RGBA", image.size, "blue")
+    final_image = Image.alpha_composite(blue_background, image.convert("RGBA"))
+    return final_image
+
+# Gradio 인터페이스에서 이미지 입력과 두 개의 출력을 설정합니다.
+inputs = gr.inputs.Image(type="pil")
+outputs = [gr.outputs.Image(type="pil", label="Background Removed"),
+           gr.outputs.Image(type="pil", label="Blue Background Added")]
+
+def update_interface(image):
+    transparent_image = process(image)
+    blue_background_image = add_blue_background(transparent_image)
+    return transparent_image, blue_background_image
+
+demo = gr.Interface(fn=update_interface,
+                    inputs=inputs,
+                    outputs=outputs,
+                    title="Background Removal and Add Blue Background",
+                    description="This adds a blue background to the transparent image.")
+
+# 'add' 버튼을 제거했으나, 백그라운드 추가는 자동으로 진행됩니다.
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()