Update app.py

app.py

@@ -24,12 +24,16 @@ def _files_to_pil_list(files, n_use: int):
 
 
 def run_app(files, n_user_imgs, prompt, k_retrieve, n_i2i, n_t2i, strength_i2i, steps, gen_size, seed):
-    """Main app handler."""
     try:
-        user_imgs = _files_to_pil_list(files, n_user_imgs)
-        if len(user_imgs) == 0:
+        if not files:
             return [], [], [], "Error: Please upload at least 1 image."
 
+        max_allowed = int(n_user_imgs)  # the slider value (1–4)
+        if len(files) > max_allowed:
+            return [], [], [], f"Error: You uploaded {len(files)} images, but the limit is {max_allowed}. Please remove extra files."
+
+        user_imgs = _files_to_pil_list(files, max_allowed)
+
         retrieved, gen_i2i, gen_t2i, info = run_search_and_generate(
             user_imgs=user_imgs,
             user_prompt=prompt,
@@ -46,7 +50,6 @@ def run_app(files, n_user_imgs, prompt, k_retrieve, n_i2i, n_t2i, strength_i2i,
         i2i_gallery = [(im, f"img2img #{i+1}") for i, im in enumerate(gen_i2i)]
         t2i_gallery = [(im, f"txt2img #{i+1}") for i, im in enumerate(gen_t2i)]
         summary = "\n".join([f"{k}: {v}" for k, v in info.items()])
-
         return retr_gallery, i2i_gallery, t2i_gallery, summary
 
     except Exception as e: