Update app.py

app.py

@@ -112,6 +112,61 @@ COMPOSITION_PARAMS = {
         "Néon": "neon color palette, vibrant glowing colors"
     }
 }
+class GenerationManager:
+    def __init__(self):
+        self.save_dir = Path("generations")
+        self.save_dir.mkdir(exist_ok=True)
+        self.history_file = self.save_dir / "history.json"
+        self.history = self.load_history()
+        logger.info("Système de génération initialisé")
+
+    def load_history(self):
+        if self.history_file.exists():
+            try:
+                with open(self.history_file, "r", encoding="utf-8") as f:
+                    return json.load(f)
+            except Exception as e:
+                logger.error(f"Erreur chargement historique: {e}")
+                return []
+        return []
+
+    def save_generation(self, image, params):
+        timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
+        image_name = f"generation_{timestamp}.png"
+        image_path = self.save_dir / image_name
+
+        try:
+            # Sauvegarde de l'image
+            image.save(image_path)
+
+            # Création de l'entrée
+            entry = {
+                "id": timestamp,
+                "image_path": str(image_path),
+                "params": params,
+                "created_at": timestamp
+            }
+
+            self.history.append(entry)
+            self._save_history()
+            return entry
+        except Exception as e:
+            logger.error(f"Erreur sauvegarde génération: {e}")
+            return None
+
+    def _save_history(self):
+        try:
+            with open(self.history_file, "w", encoding="utf-8") as f:
+                json.dump(self.history, f, indent=2, ensure_ascii=False)
+        except Exception as e:
+            logger.error(f"Erreur sauvegarde historique: {e}")
+
+    def get_recent_generations(self, limit=10):
+        return sorted(
+            self.history,
+            key=lambda x: x["created_at"],
+            reverse=True
+        )[:limit]
 
 class ImageGenerator:
     def __init__(self):