22

app.py

@@ -93,9 +93,10 @@ pipe = ZImageControlPipeline(
 pipe.to("cuda", torch.bfloat16)
 print("pipe ready.")
 
-def prepare(edit_dict):
+def prepare(edit_dict, prompt):
     # return edit_dict['background']
-    return image_utils.replace_transparent(edit_dict['layers'][0], (0, 0, 0))
+    output_image = image_utils.replace_transparent(edit_dict['layers'][0], (0, 0, 0))
+    return output_image, prompt
 
 @spaces.GPU
 def inference(
@@ -271,8 +272,8 @@ with gr.Blocks() as demo:
                 output_image = gr.Image(label="Generated image", show_label=False)
                 # polished_prompt = gr.Textbox(label="Polished prompt", interactive=False)
 
-                with gr.Accordion("Preprocessor data", open=True):
-                    mask_image = gr.Image(label="Mask image")
+                with gr.Accordion("Preprocessor data", open=False):
+                    mask_image = gr.Image(label="Generated Mask", interactive=False)
                     
         gr.Examples(examples=examples, inputs=[edit_dict, prompt])
         gr.Markdown(read_file("static/footer.md"))
@@ -280,8 +281,8 @@ with gr.Blocks() as demo:
     # edit_dict.upload(fn=lambda x: x, inputs=[mask_image], outputs=[input_image])
     run_button.click(
         fn=prepare,
-        inputs=[edit_dict],
-        outputs=[mask_image]
+        inputs=[edit_dict, prompt],
+        outputs=[mask_image, prompt]
     ).then(
         fn=inference,
         inputs=[
@@ -289,8 +290,6 @@ with gr.Blocks() as demo:
             negative_prompt,
             edit_dict,
             mask_image,
-            # image_scale,
-            # control_mode,
             control_context_scale,
             seed,
             randomize_seed,

examples/0data.json

@@ -1,27 +1,22 @@
 [
 	[
 		"examples/pose2.jpg",
-		"Woman wearing jeans and tanktop, from dubai",
-		"HED"
+		"Red bikini"
 	],
 	[
 		"examples/bottle.jpg",
-		"A man holding a bottle",
-		"HED"
+		"a cup of coffee"
 	],
 	[
 		"examples/room.jpg",
-		"modern architecture, living room",
-		"Depth"
+		"Starry night painting by Van Gogh"
 	],
 	[
 		"examples/pose1.jpg",
-		"A female paladin. Mountain background.",
-		"Pose"
+		"Silver armor"
 	],
 	[
 		"examples/bird.jpg",
-		"A bird sitting on a branch, cartoon.",
-		"Canny"
+		"Green feather."
 	]
 ]