Update app.py

app.py

@@ -112,61 +112,6 @@ 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):
@@ -264,7 +209,6 @@ def create_interface():
     """
     
     generator = ImageGenerator()
-    generation_manager = GenerationManager()
     
     with gr.Blocks(css=css) as app:
         gr.HTML("""
@@ -425,268 +369,4 @@ def create_interface():
 if __name__ == "__main__":
     app = create_interface()
     logger.info("Démarrage de l'application")
-    app.launch()
-    # [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():
-   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()
-    
-    with gr.Blocks(css=css) as app:
-        gr.HTML("""
-            <div class="welcome">
-                <h1>🎨 Equity Artisan 3.0</h1>
-                <p>Assistant de création d'affiches professionnelles</p>
-            </div>
-        """)
-        
-        with gr.Column(elem_classes="container"):
-            # Format et Orientation
-            with gr.Group(elem_classes="controls-group"):
-                gr.Markdown("### 📐 Format et Orientation")
-                with gr.Row():
-                    format_size = gr.Dropdown(
-                        choices=["A4", "A3", "A2", "A1", "A0"],
-                        value="A4",
-                        label="Format"
-                    )
-                    orientation = gr.Radio(
-                        choices=["Portrait", "Paysage"],
-                        value="Portrait",
-                        label="Orientation"
-                    )
-            
-            # Style et Composition
-            with gr.Group(elem_classes="controls-group"):
-                gr.Markdown("### 🎨 Style et Composition")
-                with gr.Row():
-                    style = gr.Dropdown(
-                        choices=list(ART_STYLES.keys()),
-                        value="Art Moderne",
-                        label="Style artistique"
-                    )
-                    layout = gr.Dropdown(
-                        choices=list(COMPOSITION_PARAMS["Layouts"].keys()),
-                        value="Centré",
-                        label="Composition"
-                    )
-
-                with gr.Row():
-                    ambiance = gr.Dropdown(
-                        choices=list(COMPOSITION_PARAMS["Ambiances"].keys()),
-                        value="Dramatique",
-                        label="Ambiance"
-                    )
-                    palette = gr.Dropdown(
-                        choices=list(COMPOSITION_PARAMS["Palette"].keys()),
-                        value="Contrasté",
-                        label="Palette"
-                    )
-            
-            # Contenu
-            with gr.Group(elem_classes="controls-group"):
-                gr.Markdown("### 📝 Contenu")
-                subject = gr.Textbox(
-                    label="Description",
-                    placeholder="Décrivez votre vision..."
-                )
-                title = gr.Textbox(
-                    label="Titre",
-                    placeholder="Titre de l'affiche..."
-                )
-            
-            # Ajustements fins
-            with gr.Group(elem_classes="advanced-controls"):
-                gr.Markdown("### 🎯 Ajustements Fins")
-                with gr.Row():
-                    detail_level = gr.Slider(
-                        minimum=1,
-                        maximum=10,
-                        value=7,
-                        step=1,
-                        label="Niveau de Détail"
-                    )
-                    contrast = gr.Slider(
-                        minimum=1,
-                        maximum=10,
-                        value=5,
-                        step=1,
-                        label="Contraste"
-                    )
-                    saturation = gr.Slider(
-                        minimum=1,
-                        maximum=10,
-                        value=5,
-                        step=1,
-                        label="Saturation"
-                    )
-            
-            # Paramètres de génération
-            with gr.Group(elem_classes="controls-group"):
-                with gr.Row():
-                    quality = gr.Slider(
-                        minimum=30,
-                        maximum=50,
-                        value=35,
-                        label="Qualité"
-                    )
-                    creativity = gr.Slider(
-                        minimum=5,
-                        maximum=15,
-                        value=7.5,
-                        label="Créativité"
-                    )
-            
-            # Historique
-            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")
-            
-            # Boutons
-            with gr.Row():
-                generate_btn = gr.Button("✨ Générer", variant="primary")
-                clear_btn = gr.Button("🗑️ Effacer", variant="secondary")
-            
-            image_output = gr.Image(label="Aperçu")
-            status = gr.Textbox(label="Statut", interactive=False)
-
-            # Fonctions
-            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)
-                
-                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])
-
-            # Events
-            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
+    app.launch()