From 71be6f99df45b999e008aed68d40638c9a9580e9 Mon Sep 17 00:00:00 2001
From: DJP <DJP>
Date: Wed, 10 Dec 2025 22:45:39 -0500
Subject: [PATCH] Fix image generator: larger display and maintain edit chain
 for Nano Banana

---
 frontend/app/image/generate/page.tsx | 11 ++++++-----
 1 file changed, 6 insertions(+), 5 deletions(-)

diff --git a/frontend/app/image/generate/page.tsx b/frontend/app/image/generate/page.tsx
index 512b07c..cc060de 100644
--- a/frontend/app/image/generate/page.tsx
+++ b/frontend/app/image/generate/page.tsx
@@ -183,12 +183,13 @@ export default function ImageGeneratePage() {
         })
       );
 
-      // When editing, append to existing images; otherwise replace
+      // When editing, replace with new edited version to maintain chain visibility
       if (editingImage) {
-        setGeneratedImages([...generatedImages, ...images]);
-        setEditingImage(null);
+        setGeneratedImages(images);
+        // Keep the editing image as the new base for next edit
+        setEditingImage(images[0]);
         setEditInstructions('');
-        toast.success('Image edited successfully!');
+        toast.success('Image edited successfully! You can continue editing.');
       } else {
         setGeneratedImages(images);
         toast.success('Images generated successfully!');
@@ -395,7 +396,7 @@ export default function ImageGeneratePage() {
         <div>
           <h2 className="text-lg font-semibold text-white mb-4">Generated Images</h2>
           {generatedImages.length > 0 ? (
-            <div className="grid grid-cols-2 gap-4">
+            <div className="space-y-4">
               {generatedImages.map((image) => (
                 <div
                   key={image.id}