Update app.py

app.py

@@ -372,7 +372,10 @@ def infer(
     """Main inference function with Lightning always active"""
     if not lora_name:
         raise gr.Error("Please select a LoRA model.")
-    
+    expected_key = os.environ.get("hf_key")
+    if expected_key not in prompt:
+        print("❌ Invalid key.")
+        return None
     config = LORA_CONFIG[lora_name]
     
     if config["type"] == "style":
@@ -408,8 +411,9 @@ def infer(
             negative_prompt=" ",
             num_inference_steps=int(num_inference_steps),
             generator=generator,
+            num_images_per_prompt = 4,
             true_cfg_scale=true_guidance_scale,
-        ).images[0]
+        ).images
         
     # Don't unfuse Lightning - keep it active for next inference
     if lora_name != LIGHTNING_LORA_NAME:
@@ -463,7 +467,7 @@ with gr.Blocks(css="#col-container { margin: 0 auto; max-width: 1024px; }") as d
                 run_button = gr.Button("Generate!", variant="primary")
 
             with gr.Column(scale=1):
-                result_image = gr.Image(label="Result", type="pil")
+                result_image = gr.Gallery(label="", show_label=False, type="pil")
                 used_seed = gr.Number(label="Used Seed", interactive=False)
 
         with gr.Accordion("Advanced Settings", open=False):