Fix image generator: larger display and maintain edit chain for Nano Banana

This commit is contained in:
DJP 2025-12-10 22:45:39 -05:00
parent c7ef5ee2ef
commit 71be6f99df

View file

@ -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}