Fix mask issues, set default values for sliders.

app.py

@@ -34,6 +34,13 @@ def calculate_optimal_dimensions(image: Image.Image):
     return width, height
 
 
+def create_full_mask(image):
+    """Generate a fully white mask for the entire image."""
+    return Image.fromarray(
+        np.ones((image.height, image.width), dtype=np.uint8) * 255
+    ).convert("L")
+
+
 @spaces.GPU(duration=120)
 def infer(
     edit_images,
@@ -42,20 +49,18 @@ def infer(
     randomize_seed=False,
     width=1024,
     height=1024,
-    guidance_scale=3.5,
-    num_inference_steps=28,
+    guidance_scale=30,
+    num_inference_steps=50,
 ):
     if not edit_images["background"]:
         raise gr.Error("Please upload an image.")
-    if not edit_images["layers"]:
-        raise gr.Error("Please draw a mask.")
     image = edit_images["background"]
-    mask = edit_images["layers"][0]
+    mask = create_full_mask(image)  # Auto-generate full white mask
     width, height = calculate_optimal_dimensions(image)
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)
     pipe = FluxFillPipeline.from_pretrained(
-        "black-forest-labs/FLUX.1-Fill-dev", torch_dtype=torch.bfloat16
+        "black-forest-labs/FLUX.1-FILL-dev", torch_dtype=torch.bfloat16
     ).to("cuda")
     generator = torch.Generator("cuda").manual_seed(seed)
     try:
@@ -76,6 +81,7 @@ def infer(
 
 
 examples = [
+    "Add a glowing crescent moon on the forehead, vivid red eyes, and a shadowy, dark, misty background to the subject, while preserving the exact structure and details of the original human face, animal, or object",
     "a tiny astronaut hatching from an egg on the moon",
     "a cat holding a sign that says hello world",
     "an anime illustration of a wiener schnitzel",
@@ -99,11 +105,11 @@ with gr.Blocks(css=css) as demo:
         with gr.Row():
             with gr.Column():
                 edit_image = gr.ImageEditor(
-                    label="Upload and draw mask for inpainting",
+                    label="Upload image for editing (no mask required)",
                     type="pil",
                     sources=["upload", "webcam"],
                     image_mode="RGB",
-                    layers=False,
+                    layers=False,  # No layers needed as mask is auto-generated
                     brush=gr.Brush(colors=["#FFFFFF"], color_mode="fixed"),
                     height=600,
                 )
@@ -148,14 +154,14 @@ with gr.Blocks(css=css) as demo:
                     minimum=1,
                     maximum=30,
                     step=0.5,
-                    value=3.5,
+                    value=30,  # Updated default to 30
                 )
                 num_inference_steps = gr.Slider(
                     label="Number of inference steps",
                     minimum=1,
                     maximum=50,
                     step=1,
-                    value=28,
+                    value=50,  # Updated default to 50
                 )
         gr.on(
             triggers=[run_button.click, prompt.submit],