22

README.md

@@ -1,5 +1,5 @@
 ---
-title: ZIT Inpaint
+title: Z-Image Turbo (ZIT) - Inpaint
 emoji: 💃
 colorFrom: blue
 colorTo: yellow
@@ -8,7 +8,11 @@ sdk_version: 6.1.0
 app_file: app.py
 pinned: false
 license: apache-2.0
-short_description: Edit image using a mask (selective changes to areas)
+short_description: Edit image by changing specific areas (mask)
+models:
+    - alibaba-pai/Z-Image-Turbo-Fun-Controlnet-Union-2.0
+    - alibaba-pai/Z-Image-Turbo-Fun-Controlnet-Union
+    - Tongyi-MAI/Z-Image-Turbo
 ---
 
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -95,6 +95,7 @@ print("pipe ready.")
 
 def prepare(edit_dict, prompt):
     # return edit_dict['background']
+    if not prompt: prompt = "Ultra HD, 4K"
     output_image = image_utils.replace_transparent(edit_dict['layers'][0], (0, 0, 0))
     return output_image, prompt
 
@@ -104,8 +105,6 @@ def inference(
     negative_prompt,
     edit_dict,
     mask_image,
-    # image_scale=1.0,
-    # control_mode='Canny',
     control_context_scale = 0.75,
     seed=42,
     randomize_seed=True,
@@ -122,36 +121,11 @@ def inference(
         print("Error: edit_dict is empty.")
         return None
     
-    # print(edit_dict)
-    # input_image, width, height = scale_image(input_image, image_scale)
-    # control_mode='HED'
-    # processor_id = 'canny'
-    # if control_mode == 'HED':
-    #     processor_id = 'softedge_hed'
-    # if control_mode =='Depth':
-    #     processor_id = 'depth_midas'
-    # if control_mode =='MLSD':
-    #     processor_id = 'mlsd'
-    # if control_mode =='Pose':
-    #     processor_id = 'openpose_full'
-
-    # print(f"DEBUG: processor_id={processor_id}")
-    # processor = Processor(processor_id)
-
-    # Width must be divisible by 16
-    
-    # control_image, width, height = image_utils.rescale_image(input_image, image_scale, 16)
-    # control_image = control_image.resize((1024, 1024))
     width, height = edit_dict['background'].size
 
     print("DEBUG: control_image_torch")
     sample_size = [height, width]
-    # control_image = processor(control_image, to_pil=True)
-    # control_image = control_image.resize((width, height))
-    # control_image_torch = get_image_latent(control_image, sample_size=sample_size)[:, :, 0]
 
-    # mask_image = edit_dict['layers'][0]
-    # mask_image = edit_dict['composite']
     if mask_image is not None:
         mask_image = get_image_latent(mask_image, sample_size=sample_size)[:, :1, 0]
     else:
@@ -221,12 +195,7 @@ with gr.Blocks() as demo:
                     placeholder="Enter your prompt",
                     # container=False,
                 )
-                # is_polish_prompt = gr.Checkbox(label="Polish prompt", value=True)
-                # control_mode = gr.Radio(
-                #     choices=["Canny", "Depth", "HED", "MLSD", "Pose"],
-                #     value="Canny",
-                #     label="Control Mode"
-                # )
+
                 run_button = gr.Button("Generate", variant="primary")
                 with gr.Accordion("Advanced Settings", open=False):
                     negative_prompt = gr.Textbox(

examples/0data.json

@@ -7,16 +7,8 @@
 		"examples/bottle.jpg",
 		"a cup of coffee"
 	],
-	[
-		"examples/room.jpg",
-		"Starry night painting by Van Gogh"
-	],
 	[
 		"examples/pose1.jpg",
 		"Silver armor"
-	],
-	[
-		"examples/bird.jpg",
-		"Green feather."
 	]
 ]

static/footer.md

@@ -1,16 +1,24 @@
 
+## What does this app do?
+This AI image editor allows you to transform your pictures by simply applying a mask and providing a text prompt. You can select specific areas of your image with the mask and then describe the desired changes using natural language. The app will then intelligently edit the selected area of your image based on your instructions.
+
 ## Usage
-- **Polish Prompt**: ZIT needs a detailed prompt, which you can get by enabling polish prompt.
-- **Context Scale**: Similar to strength, the higher the value, the more detail is preserved. The recommended control_context_scale range is 0.65 to 0.80.
-- **Image Scale**: Upscale/downscale image resolution.
+- **Edit Image (required)**: Upload the image you want to edit, then mark the area you want to change.
+- **Prompt (required)**: Type a description of the changes you want to make to the masked area.
+- **Context Scale**: Higher values preserve more detail from the original image.
+
+### Tips
+If the result isn't great, try marking a bigger area or lowering the *context scale*.
+
 
-# Version History
-2025-12-13: Updated to Controlnet Union v2.0
+## ComfyUI
+As of this writing, ComfyUI integration isn't supported yet.  
+You can follow updates here: <https://github.com/comfyanonymous/ComfyUI/pull/11304>
+
+## Version History
+2025-12-14: Published to huggingface space.
 
 ## References
 - **alibaba-pai - Controlnet Union v2.0**: <https://huggingface.co/alibaba-pai/Z-Image-Turbo-Fun-Controlnet-Union-2.0>
 - **VideoX-Fun**: <https://github.com/aigc-apps/VideoX-Fun>
 - **Tongyi-MAI**: <https://huggingface.co/Tongyi-MAI/Z-Image-Turbo>
-
-
-<!-- https://github.com/comfyanonymous/ComfyUI/pull/11062 -->

utils/repo_utils.py

@@ -46,6 +46,7 @@ def move_folder(source_path, destination_path):
         if shutil.os.path.exists(destination_path):
             shutil.rmtree(destination_path)  # Remove existing destination folder
         shutil.copytree(source_path, destination_path)
+        print(f"Folder '{source_path}' has copied to '{destination_path}")
     except Exception as e:
         print(f"Error move_folder: {e}")