1

app.py

@@ -15,14 +15,16 @@ def image_to_data_uri(image_path):
     return f"data:{mime};base64,{data}"
 
 
-def edit_images(images, prompt, token: gr.OAuthToken | None = None):
+def edit_images(img1, img2, img3, prompt, token: gr.OAuthToken | None = None):
     if not token:
         raise gr.Error("Please sign in with your Hugging Face account first.")
-    if not images or len(images) == 0:
-        raise gr.Error("Please upload at least one image.")
     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]
+    if len(images) == 0:
+        raise gr.Error("Please upload at least one image.")
+
     client = InferenceClient(provider=PROVIDER, token=token.token)
 
     image_urls = [image_to_data_uri(img) for img in images]
@@ -48,12 +50,10 @@ with gr.Blocks(fill_height=True) as demo:
 
     with gr.Row():
         with gr.Column():
-            input_images = gr.File(
-                label="Upload Images",
-                file_count="multiple",
-                file_types=["image"],
-                type="filepath",
-            )
+            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")
             prompt = gr.Textbox(
                 label="Edit Prompt",
                 placeholder="Describe how you want the image(s) edited...",
@@ -66,7 +66,7 @@ with gr.Blocks(fill_height=True) as demo:
 
     submit_btn.click(
         fn=edit_images,
-        inputs=[input_images, prompt],
+        inputs=[img1, img2, img3, prompt],
         outputs=[output_image],
     )