Update app.py

app.py

@@ -224,25 +224,44 @@ def blend_images(bg_img, obj_img, mask_img, method):
     
     return (blend_img * 255).astype(np.uint8)
 
-examples = [
-    ["img1.jpg", "img2.jpg", "mask1.jpg", "Poisson"],
-    ["img3.jpg", "img4.jpg", "mask2.jpg", "Mixed Gradient"],
-    ["img6.jpg", "img9.jpg", "mask3.jpg", "Laplacian"]
-]
-
-iface = gr.Interface(
-    fn=blend_images,
-    inputs=[
-        gr.Image(label="Background Image", type="numpy"),
-        gr.Image(label="Object Image", type="numpy"),
-        gr.Image(label="Mask Image", type="numpy"),
-        gr.Radio(["Poisson", "Mixed Gradient", "Laplacian"], label="Blending Method")
-    ],
-    outputs=gr.Image(label="Blended Image"),
-    title="Image Blending with Examples",
-    description="Choose from example images or upload your own to blend using different methods.",
-    examples=examples,
-    cache_examples=True
-)
+with gr.Blocks(theme='bethecloud/storj_theme') as iface:
+    gr.HTML("<h1>Image Blending with Multiple Methods</h1>")
+    
+    with gr.Row():
+        with gr.Column():
+            bg_img = gr.Image(label="Background Image", type="numpy")
+            obj_img = gr.Image(label="Object Image", type="numpy")
+        with gr.Column():
+            mask_img = gr.ImageEditor(
+                label="Mask Image",
+                type="numpy",
+                brush=gr.Brush(colors=["#ffffff"], color_mode="fixed"),
+                eraser=gr.Eraser(),
+                height=300,
+                width=300,
+            )
+            method = gr.Radio(["Poisson", "Mixed Gradient", "Laplacian"], label="Blending Method", value="Poisson")
+    
+    blend_button = gr.Button("Blend Images")
+    
+    output_image = gr.Image(label="Blended Image")
+    
+    blend_button.click(
+        blend_images,
+        inputs=[bg_img, obj_img, mask_img, method],
+        outputs=output_image
+    )
+    
+    gr.Examples(
+        examples=[
+            ["img1.jpg", "img2.jpg", "mask1.jpg", "Poisson"],
+            ["img3.jpg", "img4.jpg", "mask2.jpg", "Mixed Gradient"],
+            ["img6.jpg", "img9.jpg", "mask3.jpg", "Laplacian"]
+        ],
+        inputs=[bg_img, obj_img, mask_img, method],
+        outputs=output_image,
+        fn=blend_images,
+        cache_examples=True,
+    )
 
 iface.launch()