Update app.py

app.py

@@ -1,18 +1,37 @@
 import gradio as gr
 from rembg import remove
+from PIL import Image
 
-def process_image(input_img):
-    """Processes an image by removing its background and resizing it."""
-    img = remove(input_img)
-    return img
+def replace_background(foreground_img, background_img):
+    # 去背處理
+    fg = remove(foreground_img).convert("RGBA")
+    
+    # 調整背景尺寸符合前景圖
+    bg = background_img.convert("RGBA").resize(fg.size)
+
+    # 合成圖片
+    combined = Image.alpha_composite(bg, fg)
+    return combined.convert("RGB")
 
 with gr.Blocks() as demo:
-    gr.Interface(
-        fn=process_image, 
-        inputs = [gr.Image(label="Input Image", interactive=True)], 
-        outputs = gr.Image(label="Processed Image"),
-        examples=[["images/sample1.png"]],
-        title="📸 Image Background Remover")
+    gr.Markdown("# 🖼️ 去背 + 替換風景背景")
+
+    with gr.Row():
+        input_img = gr.Image(label="上傳人物圖片", type="pil")
+        bg_img = gr.Image(label="上傳或選擇背景圖片", type="pil")
+
+    output_img = gr.Image(label="輸出圖片")
+
+    examples = [
+        ["images/person1.png", "images/beach.jpg"],
+        ["images/person2.png", "images/mountain.jpg"]
+    ]
+
+    btn = gr.Button("去背並替換背景")
+    btn.click(fn=replace_background, inputs=[input_img, bg_img], outputs=output_img)
+
+    gr.Examples(examples=examples, inputs=[input_img, bg_img])
 
 if __name__ == "__main__":
     demo.launch()
+