Update app.py

app.py

@@ -10,15 +10,13 @@ import tempfile
 # Initialize the SimpleLama model
 simple_lama = SimpleLama()
 def inpaint_image(input_data):
-    if input_data is None:
+    if input_data is None or not input_data['layers']:
         raise gr.Error("Please upload an image and draw a mask.")
    
-    # input_data is a dict with 'image' and 'mask' as PIL Images
-    image = input_data['image'].convert("RGB")
-    mask = input_data['mask'].convert("L")
-   
-    # In Gradio sketch, mask is white where drawn (to inpaint), black elsewhere
-    # But ensure it's binary if needed, though SimpleLama likely handles grayscale
+    # input_data is a dict with 'background', 'layers', 'composite'
+    image = input_data['background'].convert("RGB")
+    # Assume first layer is the mask drawn with white brush
+    mask = input_data['layers'][0].convert("L")
    
     # Perform inpainting
     result = simple_lama(image, mask)
@@ -29,9 +27,8 @@ def inpaint_image(input_data):
         tmp_path = tmp.name
    
     return result, tmp_path
-inputs = [gr.Image(
+inputs = [gr.ImageEditor(
             label="Upload and Mask Image",
-            tool="sketch",
             brush=gr.Brush(colors=["#FFFFFF"], default_color="#FFFFFF"),
             interactive=True
         )]
@@ -41,7 +38,7 @@ examples = [
 ]
 # For HF Spaces, requirements.txt would include:
 # simple-lama-inpainting
-# gradio
+# gradio>=4.39.0
 # pillow
 # torch (if not included in simple-lama)
 title = "LaMa Image Inpainting"