Update app.py

app.py

@@ -430,6 +430,160 @@ class GenerationManager:
 
 # [Gardez votre classe ImageGenerator existante]
 
+# Modifiez votre fonction create_interface comme ceci
+def create_interface():
+    logger.info("Création de l'interface Gradio")
+    
+    css = """
+    .container { max-width: 1200px; margin: auto; }
+    .welcome { text-align: center; margin: 20px 0; padding: 20px; background: #1e293b; border-radius: 10px; }
+    .controls-group { background: #2d3748; padding: 15px; border-radius: 5px; margin: 10px 0; }
+    .advanced-controls { background: #374151; padding: 12px; border-radius: 5px; margin: 8px 0; }
+    """
+    
+    generator = ImageGenerator()
+    generation_manager = GenerationManager()  # Ajout du gestionnaire
+    
+    with gr.Blocks(css=css) as app:
+        # [Gardez toute votre interface existante jusqu'à la fin]
+        
+        # Ajoutez cet onglet à la fin, avant les boutons de génération
+        with gr.Tab("📜 Historique"):
+            with gr.Row():
+                history_gallery = gr.Gallery(
+                    label="Générations Récentes",
+                    columns=3,
+                    height=400,
+                    show_label=True
+                )
+                
+            with gr.Row():
+                refresh_btn = gr.Button("🔄 Actualiser")
+        
+        # Modification de la fonction generate existante
+        def generate(*args):
+            logger.info("Démarrage d'une nouvelle génération")
+            params = {
+                "format_size": args[0],
+                "orientation": args[1],
+                "style": args[2],
+                "layout": args[3],
+                "ambiance": args[4],
+                "palette": args[5],
+                "subject": args[6],
+                "title": args[7],
+                "detail_level": args[8],
+                "contrast": args[9],
+                "saturation": args[10],
+                "quality": args[11],
+                "creativity": args[12]
+            }
+            result = generator.generate(params)
+            
+            # Ajout de la sauvegarde automatique
+            if result[0] is not None:
+                generation_manager.save_generation(result[0], params)
+            
+            logger.info(f"Génération terminée avec statut: {result[1]}")
+            return result
+
+        def refresh_history():
+            recent = generation_manager.get_recent_generations()
+            return gr.Gallery.update(value=[entry["image_path"] for entry in recent])
+
+        # Connexion des boutons
+        generate_btn.click(
+            generate,
+            inputs=[
+                format_size,
+                orientation,
+                style,
+                layout,
+                ambiance,
+                palette,
+                subject,
+                title,
+                detail_level,
+                contrast,
+                saturation,
+                quality,
+                creativity
+            ],
+            outputs=[image_output, status]
+        )
+        
+        refresh_btn.click(
+            refresh_history,
+            outputs=[history_gallery]
+        )
+        
+        clear_btn.click(
+            lambda: (None, "🗑️ Image effacée"),
+            outputs=[image_output, status]
+        )
+    
+    logger.info("Interface créée avec succès")
+    return app
+    # [Gardez tous vos imports existants]
+# Ajoutez ces imports en haut du fichier
+from pathlib import Path
+import time
+import json
+from datetime import datetime
+
+# [Gardez toutes vos configurations existantes (ART_STYLES, etc.)]
+
+# Ajoutez cette classe juste avant la classe ImageGenerator
+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()
+
+    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
+
+        # 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
+
+    def _save_history(self):
+        with open(self.history_file, "w", encoding="utf-8") as f:
+            json.dump(self.history, f, indent=2, ensure_ascii=False)
+
+    def get_recent_generations(self, limit=10):
+        return sorted(
+            self.history,
+            key=lambda x: x["created_at"],
+            reverse=True
+        )[:limit]
+
+# [Gardez votre classe ImageGenerator existante]
+
 # Modifiez votre fonction create_interface comme ceci
 def create_interface():
     logger.info("Création de l'interface Gradio")