Add hidden mask image component for API access

- Add mask_in component with visible=False to keep UI clean
- Update generate function parameter order to match Gradio inputs
- Mask component is hidden from web UI but available for API users
- Enables ComfyUI and other API integrations to pass optional mask images

app.py

@@ -281,8 +281,8 @@ def generate(
     base_img: Image.Image, 
     garment_img: Image.Image, 
     workflow_choice: str, 
-    request: gr.Request,
-    mask_img: Optional[Image.Image] = None  # NEW: Optional mask parameter
+    mask_img: Optional[Image.Image],  # NEW: Optional mask parameter  
+    request: gr.Request
 ) -> Image.Image:
     if base_img is None or garment_img is None:
         raise gr.Error("Please provide both images.")
@@ -462,6 +462,13 @@ with gr.Blocks(title="YOURMIRROR.IO - SM4LL-VTON Demo") as demo:
                 height=IMG_SIZE,
                 width=IMG_SIZE,
             )
+            mask_in = gr.Image(
+                label="Mask Image (Optional)",
+                type="pil",
+                height=IMG_SIZE,
+                width=IMG_SIZE,
+                visible=False,  # Hidden from UI but available for API
+            )
 
         # Third column: Result
         with gr.Column(scale=2):
@@ -528,7 +535,7 @@ with gr.Blocks(title="YOURMIRROR.IO - SM4LL-VTON Demo") as demo:
     # Wire up interaction
     generate_btn.click(
         generate,
-        inputs=[base_in, garment_in, workflow_selector],
+        inputs=[base_in, garment_in, workflow_selector, mask_in],
         outputs=result_out,
     )