Update app.py

app.py

@@ -433,7 +433,7 @@ class GenerationManager:
 
 # Modifiez votre fonction create_interface comme ceci
 def create_interface():
-    def create_interface():
+   def create_interface():
     logger.info("Création de l'interface Gradio")
     
     css = """
@@ -444,85 +444,194 @@ def create_interface():
     """
     
     generator = ImageGenerator()
-    generation_manager = GenerationManager()  # Ajout du gestionnaire
+    generation_manager = GenerationManager()
     
     with gr.Blocks(css=css) as app:
-        # [Gardez toute votre interface existante jusqu'à la fin]
+        gr.HTML("""
+            <div class="welcome">
+                <h1>🎨 Equity Artisan 3.0</h1>
+                <p>Assistant de création d'affiches professionnelles</p>
+            </div>
+        """)
         
-        # 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.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")
-        
-        # 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)
+                with gr.Row():
+                    refresh_btn = gr.Button("🔄 Actualiser")
             
-            # Ajout de la sauvegarde automatique
-            if result[0] is not None:
-                generation_manager.save_generation(result[0], params)
+            # Boutons
+            with gr.Row():
+                generate_btn = gr.Button("✨ Générer", variant="primary")
+                clear_btn = gr.Button("🗑️ Effacer", variant="secondary")
             
-            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])
+            image_output = gr.Image(label="Aperçu")
+            status = gr.Textbox(label="Statut", interactive=False)
 
-        # 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]
-        )
+            # 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