Update app.py

app.py

@@ -4,8 +4,11 @@ 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
 
@@ -31,12 +34,11 @@ 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, "num_images": 1},
+            # CORRECTED: The 'edit' endpoint always expects 'image_urls' as a list.
+            arguments={"prompt": prompt, "image_urls": [image_url]},
         )
     else:
         result = fal_client.run(
@@ -46,16 +48,13 @@ def run_single_image_logic(prompt: str, image: Optional[str] = None) -> str:
 
 def run_multi_image_logic(prompt: str, images: List[str]) -> str:
     """
-    Uploads multiple images
+    Handles multi-image editing by sending a list of URLs in a single API call.
     """
     get_fal_key()
     if not images:
         raise gr.Error("Please upload at least one image in the 'Multiple Images' tab.")
 
-    # 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={
@@ -64,8 +63,6 @@ def run_multi_image_logic(prompt: str, images: List[str]) -> str:
             "num_images": 1
         },
     )
-
-    # 3. Return the single resulting image URL
     return result["images"][0]["url"]
 
 # --- Gradio App UI ---