Update app.py

app.py

@@ -4,11 +4,8 @@ import os
 from typing import Optional, List
 from huggingface_hub import whoami
 
-# It is recommended to create this as a Secret on your Hugging Face Space
-# For example: FAL_KEY = "fal_key_..."
 FAL_KEY = os.getenv("FAL_KEY", "")
 
-# Set the key for the fal_client
 if FAL_KEY:
     fal_client.api_key = FAL_KEY
 
@@ -34,7 +31,9 @@ def run_single_image_logic(prompt: str, image: Optional[str] = None) -> str:
     """Handles text-to-image or single image-to-image and returns a single URL string."""
     get_fal_key()
     if image:
+        print(image)
         image_url = fal_client.upload_file(image)
+        print(image_url)
         result = fal_client.run(
             "fal-ai/nano-banana/edit",
             arguments={"prompt": prompt, "image_url": image_url},
@@ -47,21 +46,27 @@ def run_single_image_logic(prompt: str, image: Optional[str] = None) -> str:
 
 def run_multi_image_logic(prompt: str, images: List[str]) -> str:
     """
-    Handles multi-image editing and returns the URL of the LAST generated image.
+    Uploads multiple images
     """
     get_fal_key()
     if not images:
         raise gr.Error("Please upload at least one image in the 'Multiple Images' tab.")
 
-    last_image_url = ""
-    for image_path in images:
-        image_url = fal_client.upload_file(image_path)
-        result = fal_client.run(
-            "fal-ai/nano-banana/edit",
-            arguments={"prompt": prompt, "image_url": image_url},
-        )
-        last_image_url = result["images"][0]["url"]
-    return last_image_url
+    # 1. Upload all images and collect their URLs
+    image_urls = [fal_client.upload_file(image_path) for image_path in images]
+
+    # 2. Make a single API call with the list of URLs
+    result = fal_client.run(
+        "fal-ai/nano-banana/edit",
+        arguments={
+            "prompt": prompt,
+            "image_urls": image_urls,
+            "num_images": 1
+        },
+    )
+
+    # 3. Return the single resulting image URL
+    return result["images"][0]["url"]
 
 # --- Gradio App UI ---
 with gr.Blocks(theme=gr.themes.Citrus()) as demo: