History v3

app.py

@@ -154,32 +154,31 @@ def generate(
             ).images
 
         if images:
-            for image in images:
-                # Create thumbnail
-                thumbnail = image.copy()
-                thumbnail.thumbnail((256, 256))
-                
-                # Convert thumbnail to base64
-                buffered = BytesIO()
-                thumbnail.save(buffered, format="PNG")
-                img_str = base64.b64encode(buffered.getvalue()).decode()
-                
-                # Add to history
-                image_history.appendleft({
-                    "thumbnail": f"data:image/png;base64,{img_str}",
-                    "prompt": prompt,
-                    "negative_prompt": negative_prompt,
-                    "seed": seed,
-                    "width": width,
-                    "height": height,
-                })
-
-        if images and IS_COLAB:
-            for image in images:
+            image = images[0]
+            # Create thumbnail
+            thumbnail = image.copy()
+            thumbnail.thumbnail((256, 256))
+            
+            # Convert thumbnail to base64
+            buffered = BytesIO()
+            thumbnail.save(buffered, format="PNG")
+            img_str = base64.b64encode(buffered.getvalue()).decode()
+            
+            # Add to history
+            image_history.appendleft({
+                "thumbnail": f"data:image/png;base64,{img_str}",
+                "prompt": prompt,
+                "negative_prompt": negative_prompt,
+                "seed": seed,
+                "width": width,
+                "height": height,
+            })
+
+            if IS_COLAB:
                 filepath = utils.save_image(image, metadata, OUTPUT_DIR)
                 logger.info(f"Image saved as {filepath} with metadata")
 
-        return images, metadata, list(image_history)
+        return image, metadata, list(image_history)
 
     except Exception as e:
         logger.exception(f"An error occurred: {e}")