Update app.py (#1)

- Update app.py (9742a7ce84cb8a4a5bd54fd0ea1cb2bda928441e)

app.py

@@ -273,15 +273,16 @@ Remove the background of image 2, and replace it with the background of image 1.
 """
 
 def use_output_as_input(output_images):
-    """Convert output images to input format for the gallery"""
+    """Convert output images to input format for the reference image"""
     if output_images is None or len(output_images) == 0:
-        return []
-    return output_images
+        return None
+    return output_images[0]
 
 # --- Main Inference Function (with hardcoded negative prompt) ---
 @spaces.GPU()
 def infer(
-    images,
+    reference_image,
+    pose_image,
     prompt,
     seed=42,
     randomize_seed=False,
@@ -297,7 +298,8 @@ def infer(
     Run image-editing inference using the Qwen-Image-Edit pipeline.
 
     Parameters:
-        images (list): Input images from the Gradio gallery (PIL or path-based).
+        reference_image: Reference image (PIL or path-based).
+        pose_image: Pose image (PIL or path-based).
         prompt (str): Editing instruction (may be rewritten by LLM if enabled).
         seed (int): Random seed for reproducibility.
         randomize_seed (bool): If True, overrides seed with a random value.
@@ -324,17 +326,30 @@ def infer(
     
     # Load input images into PIL Images
     pil_images = []
-    if images is not None:
-        for item in images:
-            try:
-                if isinstance(item[0], Image.Image):
-                    pil_images.append(item[0].convert("RGB"))
-                elif isinstance(item[0], str):
-                    pil_images.append(Image.open(item[0]).convert("RGB"))
-                elif hasattr(item, "name"):
-                    pil_images.append(Image.open(item.name).convert("RGB"))
-            except Exception:
-                continue
+    
+    # Process reference image (first)
+    if reference_image is not None:
+        try:
+            if isinstance(reference_image, Image.Image):
+                pil_images.append(reference_image.convert("RGB"))
+            elif isinstance(reference_image, str):
+                pil_images.append(Image.open(reference_image).convert("RGB"))
+            elif hasattr(reference_image, "name"):
+                pil_images.append(Image.open(reference_image.name).convert("RGB"))
+        except Exception:
+            pass
+    
+    # Process pose image (second)
+    if pose_image is not None:
+        try:
+            if isinstance(pose_image, Image.Image):
+                pil_images.append(pose_image.convert("RGB"))
+            elif isinstance(pose_image, str):
+                pil_images.append(Image.open(pose_image).convert("RGB"))
+            elif hasattr(pose_image, "name"):
+                pil_images.append(Image.open(pose_image.name).convert("RGB"))
+        except Exception:
+            pass
 
     if height==256 and width==256:
         height, width = None, None
@@ -379,40 +394,38 @@ css = """
 #edit_text{margin-top: -62px !important}
 """
 
-with gr.Blocks(css=css) as demo:
+with gr.Blocks(css=css, theme=gr.themes.Citrus()) as demo:
     with gr.Column(elem_id="col-container"):
         gr.HTML("""
         <div id="logo-title">
-            <img src="https://qianwen-res.oss-cn-beijing.aliyuncs.com/Qwen-Image/qwen_image_edit_logo.png" alt="Qwen-Image Edit Logo" width="400" style="display: block; margin: 0 auto;">
-            <h2 style="font-style: italic;color: #5b47d1;margin-top: -27px !important;margin-left: 96px">[Plus] Fast, 4-steps with LightX2V LoRA</h2>
+            <h1 style="color: #5b47d1;">🧘 Qwen Edit Any Pose</h1>
+            <h2 style="font-style: italic;color: #5b47d1;margin-top: -10px !important;">Fast 4-step pose transfer with AnyPose LoRA</h2>
         </div>
         """)
         gr.Markdown("""
-        [Learn more](https://github.com/QwenLM/Qwen-Image) about the Qwen-Image series. 
-        This demo uses the new [Qwen-Image-Edit-2511](https://huggingface.co/Qwen/Qwen-Image-Edit-2511) with the [Qwen-Image-Lightning-2511](https://huggingface.co/lightx2v/Qwen-Image-Edit-2511-Lightning) LoRA for accelerated inference.
-        Try on [Qwen Chat](https://chat.qwen.ai/), or [download model](https://huggingface.co/Qwen/Qwen-Image-Edit-2509) to run locally with ComfyUI or diffusers.
+        Transfer any pose from a reference image to your subject using [Qwen-Image-Edit-2511](https://huggingface.co/Qwen/Qwen-Image-Edit-2511) with [AnyPose LoRA](https://huggingface.co/lilylilith/AnyPose) and [Lightning LoRA](https://huggingface.co/lightx2v/Qwen-Image-Edit-2511-Lightning) for fast inference.
+        [Learn more](https://github.com/QwenLM/Qwen-Image) about the Qwen-Image series.
         """)
         with gr.Row():
-            with gr.Column():
-                input_images = gr.Gallery(label="Input Images", 
-                                          show_label=False, 
-                                          type="pil", 
-                                          interactive=True)
-
-            with gr.Column():
+            with gr.Column(scale=1):
+                with gr.Row():
+                    reference_image = gr.Image(label="Reference Image", type="pil", interactive=True)
+                    pose_image = gr.Image(label="Pose Image", type="pil", interactive=True)
+                
+                with gr.Row():
+                    prompt = gr.Text(
+                            label="Prompt",
+                            value=DEFAULT_LORA_PROMPT,
+                            show_label=False,
+                            visible=False
+                    )
+                    run_button = gr.Button("Edit Pose", variant="primary")
+
+            with gr.Column(scale=1):
                 result = gr.Gallery(label="Result", show_label=False, type="pil", interactive=False)
                 # Add this button right after the result gallery - initially hidden
                 use_output_btn = gr.Button("↗️ Use as input", variant="secondary", size="sm", visible=False)
 
-        with gr.Row():
-            prompt = gr.Text(
-                    label="Prompt",
-                    value=DEFAULT_LORA_PROMPT,
-                    show_label=False,
-                    visible=False
-            )
-            run_button = gr.Button("Edit Pose", variant="primary")
-
         with gr.Accordion("Advanced Settings", open=False):
             # Negative prompt UI element is removed here
 
@@ -469,7 +482,8 @@ with gr.Blocks(css=css) as demo:
         triggers=[run_button.click, prompt.submit],
         fn=infer,
         inputs=[
-            input_images,
+            reference_image,
+            pose_image,
             prompt,
             seed,
             randomize_seed,
@@ -486,7 +500,7 @@ with gr.Blocks(css=css) as demo:
     use_output_btn.click(
         fn=use_output_as_input,
         inputs=[result],
-        outputs=[input_images]
+        outputs=[reference_image]
     )
 
 if __name__ == "__main__":