1

app.py

@@ -5,6 +5,7 @@ import os
 
 MODEL = "Qwen/Qwen-Image-Edit-2511"
 PROVIDER = "fal-ai"
+NUM_IMAGES = 5
 
 
 def image_to_data_uri(image_path):
@@ -15,13 +16,14 @@ def image_to_data_uri(image_path):
     return f"data:{mime};base64,{data}"
 
 
-def edit_images(img1, img2, img3, prompt, token: gr.OAuthToken | None = None):
+def edit_images(*args):
+    *imgs, prompt, token = args
     if not token:
         raise gr.Error("Please sign in with your Hugging Face account first.")
     if not prompt or prompt.strip() == "":
         raise gr.Error("Please enter a prompt.")
 
-    images = [img for img in [img1, img2, img3] if img is not None]
+    images = [img for img in imgs if img is not None]
     if len(images) == 0:
         raise gr.Error("Please upload at least one image.")
 
@@ -39,6 +41,14 @@ def edit_images(img1, img2, img3, prompt, token: gr.OAuthToken | None = None):
     return result
 
 
+def swap(a, b):
+    return b, a
+
+
+def send_to(output_img):
+    return output_img
+
+
 with gr.Blocks(fill_height=True) as demo:
     with gr.Sidebar():
         gr.Markdown("# Qwen Image Edit")
@@ -49,11 +59,19 @@ with gr.Blocks(fill_height=True) as demo:
         login_btn = gr.LoginButton("Sign in with Hugging Face")
 
     with gr.Row():
-        with gr.Column():
+        with gr.Column(scale=3):
             with gr.Row():
-                img1 = gr.Image(label="Image 1", type="filepath")
-                img2 = gr.Image(label="Image 2 (optional)", type="filepath")
-                img3 = gr.Image(label="Image 3 (optional)", type="filepath")
+                img = [
+                    gr.Image(
+                        label=f"Image {i+1}" if i == 0 else f"Image {i+1} (optional)",
+                        type="filepath",
+                    )
+                    for i in range(NUM_IMAGES)
+                ]
+            with gr.Row():
+                swap_btns = []
+                for i in range(NUM_IMAGES - 1):
+                    swap_btns.append(gr.Button(f"{i+1} ↔ {i+2}", size="sm"))
             prompt = gr.Textbox(
                 label="Edit Prompt",
                 placeholder="Describe how you want the image(s) edited...",
@@ -61,12 +79,26 @@ with gr.Blocks(fill_height=True) as demo:
             )
             submit_btn = gr.Button("Edit", variant="primary")
 
-        with gr.Column():
-            output_image = gr.Image(label="Result", type="pil")
+        with gr.Column(scale=2):
+            output_image = gr.Image(label="Result", type="filepath")
+            gr.Markdown("**Send result to input:**")
+            with gr.Row():
+                to_btns = [
+                    gr.Button(f"→ {i+1}", size="sm") for i in range(NUM_IMAGES)
+                ]
+
+    # Swap buttons
+    for idx, btn in enumerate(swap_btns):
+        btn.click(fn=swap, inputs=[img[idx], img[idx + 1]], outputs=[img[idx], img[idx + 1]])
+
+    # Send output to input
+    for idx, btn in enumerate(to_btns):
+        btn.click(fn=send_to, inputs=[output_image], outputs=[img[idx]])
 
+    # Edit
     submit_btn.click(
         fn=edit_images,
-        inputs=[img1, img2, img3, prompt],
+        inputs=[*img, prompt],
         outputs=[output_image],
     )