Fix static output issue with LoRA loading and error handling

- Fix LoRA loading by properly unloading existing LoRAs first
- Add error handling for generation with fallback to basic pipe() call
- Add debug prints to track LoRA loading status
- Should resolve static output issues

🤖 Generated with [Claude Code](https://claude.ai/code)

Co-Authored-By: Claude <noreply@anthropic.com>

app.py

@@ -74,24 +74,40 @@ def infer(prompt, seed=42, randomize_seed=False, width=1024, height=1024, guidan
     generator = torch.Generator().manual_seed(seed)
     
     # Load selected LoRA
-    if lora_selection != "None" and lora_selection in loaded_loras:
-        try:
+    try:
+        # First unload any existing LoRAs
+        pipe.unload_lora_weights()
+        
+        if lora_selection != "None" and lora_selection in loaded_loras:
             pipe.load_lora_weights(loaded_loras[lora_selection])
-            pipe.fuse_lora(lora_scale=1.0)
-        except Exception as e:
-            print(f"Failed to load LoRA {lora_selection}: {e}")
+            print(f"Loaded LoRA: {lora_selection}")
+    except Exception as e:
+        print(f"Failed to load LoRA {lora_selection}: {e}")
     
-    for img in pipe.flux_pipe_call_that_returns_an_iterable_of_images(
+    try:
+        for img in pipe.flux_pipe_call_that_returns_an_iterable_of_images(
+                prompt=prompt,
+                guidance_scale=guidance_scale,
+                num_inference_steps=num_inference_steps,
+                width=width,
+                height=height,
+                generator=generator,
+                output_type="pil",
+                good_vae=good_vae,
+            ):
+                yield img, seed
+    except Exception as e:
+        print(f"Error during generation: {e}")
+        # Fallback to basic generation
+        img = pipe(
             prompt=prompt,
             guidance_scale=guidance_scale,
             num_inference_steps=num_inference_steps,
             width=width,
             height=height,
             generator=generator,
-            output_type="pil",
-            good_vae=good_vae,
-        ):
-            yield img, seed
+        ).images[0]
+        yield img, seed
     
 examples = [
     "a tiny astronaut hatching from an egg on the moon",