From 43509333ec2a09cfbab3ccf3a9a9339fb809912d Mon Sep 17 00:00:00 2001
From: Pazz <pascalk89@gmail.com>
Date: Tue, 1 Aug 2023 18:06:25 +0200
Subject: [PATCH] fix: save images from ui when a batch was run

When doing batches, it was not possible to save images from the ui
because the prompts and images sent to the ui didn't align.
---
 scripts/onebuttonprompt.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/scripts/onebuttonprompt.py b/scripts/onebuttonprompt.py
index 2ddec0d..5f09eba 100644
--- a/scripts/onebuttonprompt.py
+++ b/scripts/onebuttonprompt.py
@@ -1062,9 +1062,14 @@ class Script(scripts.Script):
             processed = process_images(p)
             images += processed.images
             infotexts += processed.infotexts
+            
+            # prompt and seed info for batch grid
             all_seeds.append(processed.seed)
             all_prompts.append(processed.prompt)
             
+            # prompt and seed info for individual images
+            all_seeds += processed.all_seeds
+            all_prompts += processed.all_prompts
             
             state.job = f"{state.job_no} out of {state.job_count}"