Update interface to match desired layout and tools

app.py

@@ -44,13 +44,15 @@ def calculate_optimal_dimensions(image: Image.Image):
 
 # Inpainting function
 def infer(edit_images, prompt):
-    image = edit_images["background"]
+    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]
     width, height = calculate_optimal_dimensions(image)
     seed = random.randint(0, np.iinfo(np.int32).max)
-    generator = torch.Generator(device=device).manual_seed(seed)
+    generator = torch.Generator(device).manual_seed(seed)
     try:
         output = pipe(
             prompt=prompt,
@@ -70,18 +72,25 @@ def infer(edit_images, prompt):
 # Gradio interface setup
 with gr.Blocks() as demo:
     gr.Markdown("# FLUX.1 Fill [dev]")
-    edit_image = gr.ImageEditor(
-        label="Upload and draw mask for inpainting",
-        type="pil",
-        sources=["upload", "webcam"],
-        image_mode="RGB",
-        layers=False,
-        brush=gr.Brush(colors=["#FFFFFF"], color_mode="fixed"),
-        height=600,
-    )
-    prompt = gr.Textbox(label="Prompt", placeholder="Enter your prompt")
-    run_button = gr.Button("Run")
-    result = gr.Image(label="Result")
+    with gr.Row():
+        with gr.Column(scale=1):
+            edit_image = gr.ImageEditor(
+                label="Upload and draw mask for inpainting",
+                type="pil",
+                sources=["upload"],
+                image_mode="RGB",
+                layers=False,
+                brush=gr.Brush(colors=["#FFFFFF"], color_mode="fixed"),
+                height=400,
+            )
+        with gr.Column(scale=2):
+            prompt = gr.Textbox(
+                label="Prompt",
+                value="add a golden crescent moon on the forehead, glowing red cat eyes",
+                placeholder="Enter your prompt",
+            )
+            run_button = gr.Button("Run")
+            result = gr.Image(label="Result")
     run_button.click(infer, inputs=[edit_image, prompt], outputs=result)
 
 # Launch the demo