Add an example

app.py

@@ -18,6 +18,48 @@ pipe = pipe.to(device)
 def noise_color(color, noise):
     return color + random.randint(- noise, noise)
 
+def check(
+    source_img,
+    enlarge_top,
+    enlarge_right,
+    enlarge_bottom,
+    enlarge_left,
+    prompt,
+    negative_prompt,
+    smooth_border,
+    denoising_steps,
+    num_inference_steps,
+    guidance_scale,
+    randomize_seed,
+    seed,
+    debug_mode,
+    progress = gr.Progress()):
+    if source_img is None:
+        raise gr.Error("Please provide an image.")
+
+    if prompt is None or prompt == "":
+        raise gr.Error("Please provide a prompt input.")
+
+    if (not (enlarge_top is None)) and enlarge_top < 0:
+        raise gr.Error("Please provide positive top margin.")
+
+    if (not (enlarge_right is None)) and enlarge_right < 0:
+        raise gr.Error("Please provide positive right margin.")
+
+    if (not (enlarge_bottom is None)) and enlarge_bottom < 0:
+        raise gr.Error("Please provide positive bottom margin.")
+
+    if (not (enlarge_left is None)) and enlarge_left < 0:
+        raise gr.Error("Please provide positive left margin.")
+
+    if (
+        (enlarge_top is None or enlarge_top == 0)
+        and (enlarge_right is None or enlarge_right == 0)
+        and (enlarge_bottom is None or enlarge_bottom == 0)
+        and (enlarge_left is None or enlarge_left == 0)
+    ):
+        raise gr.Error("At least one border must be enlarged.")
+
 def predict(
     source_img,
     enlarge_top,
@@ -37,9 +79,6 @@ def predict(
     start = time.time()
     progress(0, desc = "Preparing data...")
 
-    if source_img is None:
-        raise gr.Error("Please provide an image.")
-
     if enlarge_top is None or enlarge_top == "":
         enlarge_top = 0
 
@@ -52,15 +91,6 @@ def predict(
     if enlarge_left is None or enlarge_left == "":
         enlarge_left = 0
 
-    if enlarge_top < 0 or enlarge_right < 0 or enlarge_bottom < 0 or enlarge_left < 0:
-        raise gr.Error("Please only provide positive margins.")
-
-    if enlarge_top == 0 and enlarge_right == 0 and enlarge_bottom == 0 and enlarge_left == 0:
-        raise gr.Error("At least one border must be enlarged.")
-
-    if prompt is None or prompt == "":
-        raise gr.Error("Please provide a prompt input.")
-
     if negative_prompt is None:
         negative_prompt = ""
 
@@ -238,7 +268,23 @@ with gr.Blocks() as interface:
         mask_image = gr.Image(label = "Mask image", visible = False)
 
     submit.click(toggle_debug, debug_mode, [original_image, enlarged_image, mask_image], queue = False,
-                     show_progress = False).then(predict, inputs = [
+                     show_progress = False).then(check, inputs = [
+        source_img,
+        enlarge_top,
+        enlarge_right,
+        enlarge_bottom,
+        enlarge_left,
+        prompt,
+        negative_prompt,
+        smooth_border,
+        denoising_steps,
+        num_inference_steps,
+        guidance_scale,
+        randomize_seed,
+        seed,
+        debug_mode
+    ], outputs = [], queue = False,
+                     show_progress = False).success(predict, inputs = [
         source_img,
         enlarge_top,
         enlarge_right,
@@ -261,4 +307,49 @@ with gr.Blocks() as interface:
         mask_image
     ], scroll_to_output = True)
 
+    gr.Examples(
+	    inputs = [
+            source_img,
+            enlarge_top,
+            enlarge_right,
+            enlarge_bottom,
+            enlarge_left,
+            prompt,
+            negative_prompt,
+            smooth_border,
+            denoising_steps,
+            num_inference_steps,
+            guidance_scale,
+            randomize_seed,
+            seed,
+            debug_mode
+        ],
+	    outputs = [
+            uncropped_image,
+            information,
+            original_image,
+            enlarged_image,
+            mask_image
+        ],
+        examples = [
+                [
+                    "Example1.png",
+                    64,
+                    64,
+                    64,
+                    64,
+                    "Stone wall, front view, homogene light",
+                    "Border, frame, painting, scribbling, smear, noise, blur, watermark",
+                    20,
+                    1000,
+                    20,
+                    7,
+                    True,
+                    42,
+                    False
+                ],
+            ],
+        cache_examples = False,
+    )
+
     interface.queue().launch()