Update app.py

app.py

@@ -1,89 +1,85 @@
 import gradio as gr
 import os
-from PIL import Image, ImageDraw, ImageFont
+from PIL import Image, ImageDraw, ImageFont, ImageEnhance
 import requests
 import io
+import json
 
-# Styles artistiques optimisés pour le texte
-ART_STYLES = {
-    "Neo Vintage": {
-        "prompt_prefix": "vintage propaganda poster style, retro advertisement design",
-        "text_style": "clear vintage typography, screen printing style lettering, clean text layout",
-        "guidance": 12.5,
-        "steps": 70,
-        "negative_prompt": "blurry text, illegible writing, damaged letters, distorted fonts, gibberish"
-    },
-    "Art Déco": {
-        "prompt_prefix": "art deco poster design, 1920s luxury advertisement style",
-        "text_style": "elegant art deco typography, decorative lettering, geometric text design",
-        "guidance": 13.0,
-        "steps": 75,
-        "negative_prompt": "modern fonts, messy text, unclear letters, random characters"
-    },
-    "Moderne": {
-        "prompt_prefix": "modern minimalist poster design, contemporary professional layout",
-        "text_style": "clean sans-serif typography, precise text placement, minimal design",
-        "guidance": 11.5,
-        "steps": 65,
-        "negative_prompt": "cluttered text, unreadable fonts, messy layout, poor typography"
-    },
-    "Pop Art": {
-        "prompt_prefix": "pop art style poster, comic book inspired design",
-        "text_style": "bold comic book typography, dynamic text placement, retro lettering",
-        "guidance": 11.0,
-        "steps": 60,
-        "negative_prompt": "subtle text, unclear typography, messy letters"
-    }
-}
+# [Insérer ici le code des ART_STYLES, TEXT_EFFECTS, et THEME_COLLECTIONS précédents]
 
-# CSS personnalisé
+# Interface utilisateur améliorée avec CSS personnalisé
 CUSTOM_CSS = """
 .container { max-width: 1200px; margin: auto; }
 .welcome { text-align: center; margin: 20px 0; padding: 20px; background: #1e293b; border-radius: 10px; }
 .quality-controls { margin: 10px 0; padding: 10px; background: #2d3748; border-radius: 5px; }
-.text-input { background: #2d3748; padding: 15px; border-radius: 5px; margin: 10px 0; }
+.style-group { background: #2d3748; padding: 15px; border-radius: 5px; margin: 10px 0; }
+.text-effects { background: #374151; padding: 12px; border-radius: 5px; margin: 8px 0; }
+.preview-panel { position: relative; }
+.parameters { display: flex; gap: 10px; }
 """
 
-def generate_enhanced_prompt(subject, style, title="", subtitle="", additional_details=""):
-    """Génère un prompt optimisé pour le texte"""
+def enhance_prompt(subject, style, text_effect, collection=None, additional_details=""):
+    """Génération de prompt optimisée"""
     style_config = ART_STYLES[style]
+    text_config = TEXT_EFFECTS[text_effect]
     
     # Construction du prompt de base
     base_prompt = f"{style_config['prompt_prefix']}, {subject}"
     
-    # Ajout des éléments de texte si présents
-    text_elements = []
-    if title:
-        text_elements.append(f"with main title text '{title}'")
-    if subtitle:
-        text_elements.append(f"subtitle text '{subtitle}'")
-    
-    text_prompt = ", ".join(text_elements)
-    if text_prompt:
-        base_prompt += f", {text_prompt}, {style_config['text_style']}"
+    # Ajout des effets de texte
+    if text_effect != "Standard":
+        base_prompt += f", {text_config['prompt_suffix']}"
     
+    # Ajout des éléments de collection
+    if collection and collection in THEME_COLLECTIONS:
+        collection_data = THEME_COLLECTIONS[collection]
+        collection_elements = (
+            collection_data["prompts"] + 
+            collection_data["styles"]
+        )
+        base_prompt += f", {', '.join(collection_elements)}"
+        
+    # Ajout des détails supplémentaires
     if additional_details:
         base_prompt += f", {additional_details}"
     
-    return base_prompt, style_config['negative_prompt']
+    # Construction du prompt négatif
+    negative_prompt = style_config['negative_prompt']
+    if collection:
+        negative_prompt += f", {THEME_COLLECTIONS[collection]['negative']}"
+    
+    return base_prompt, negative_prompt
 
-def generate_image(format_size, orientation, subject, style, title, subtitle, quality, creativity, additional_details=""):
+def generate_image(format_size, orientation, subject, style, text_effect, collection, 
+                  title, subtitle, quality, creativity, additional_details=""):
+    """Fonction de génération d'image améliorée"""
     API_URL = "https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-xl-base-1.0"
     headers = {"Authorization": f"Bearer {os.getenv('HUGGINGFACE_TOKEN')}"}
     
     try:
-        # Dimensions optimisées
-        width, height = (768, 1024) if format_size == "A4" else (1024, 1024)
-        if orientation == "Paysage":
-            width, height = height, width
-            
+        # Optimisation des dimensions
+        base_width = 1024
+        base_height = 1024
+        if format_size in ["A4", "A3"]:
+            if orientation == "Portrait":
+                base_width = 768
+                base_height = 1024
+            else:
+                base_width = 1024
+                base_height = 768
+                
         # Génération du prompt optimisé
-        enhanced_prompt, negative_prompt = generate_enhanced_prompt(
-            subject, style, title, subtitle, additional_details
+        enhanced_prompt, negative_prompt = enhance_prompt(
+            subject, style, text_effect, collection, additional_details
         )
         
-        print(f"Prompt: {enhanced_prompt}")  # Debug
-        print(f"Negative prompt: {negative_prompt}")  # Debug
+        # Ajout des textes si présents
+        if title or subtitle:
+            enhanced_prompt += f", with text elements: '{title}'"
+            if subtitle:
+                enhanced_prompt += f" and subtitle: '{subtitle}'"
+                
+        print(f"Prompt final: {enhanced_prompt}")  # Debug
         
         # Configuration de la requête
         payload = {
@@ -92,8 +88,9 @@ def generate_image(format_size, orientation, subject, style, title, subtitle, qu
                 "negative_prompt": negative_prompt,
                 "num_inference_steps": int(ART_STYLES[style]['steps'] * (quality/100)),
                 "guidance_scale": ART_STYLES[style]['guidance'] * (creativity/10),
-                "width": width,
-                "height": height
+                "width": base_width,
+                "height": base_height,
+                "seed": int(creativity * 1000)  # Pour la reproductibilité
             }
         }
         
@@ -101,7 +98,7 @@ def generate_image(format_size, orientation, subject, style, title, subtitle, qu
         
         if response.status_code == 200:
             image = Image.open(io.BytesIO(response.content))
-            return image, f"✨ Création {style} réussie avec texte intégré!"
+            return image, f"✨ Création {style} avec effet {text_effect} réussie!"
         else:
             print(f"Erreur API: {response.text}")  # Debug
             return None, f"⚠️ Erreur {response.status_code}: Ajustez les paramètres"
@@ -112,17 +109,18 @@ def generate_image(format_size, orientation, subject, style, title, subtitle, qu
 
 def create_interface():
     with gr.Blocks(css=CUSTOM_CSS) as app:
+        # En-tête
         gr.HTML("""
             <div class="welcome">
                 <h1>🤖 Equity Artisan 3.0</h1>
-                <p>Créez des affiches professionnelles étape par étape avec notre assistant créatif augmenté.</p>
+                <p>Créez des affiches artistiques professionnelles avec notre assistant créatif augmenté.</p>
             </div>
         """)
         
         with gr.Column(elem_classes="container"):
             # Étape 1: Format
-            with gr.Group():
-                gr.Markdown("### 📏 Étape 1: Format et Orientation")
+            with gr.Group(elem_classes="style-group"):
+                gr.Markdown("### 📏 Format et Orientation")
                 with gr.Row():
                     format_choice = gr.Dropdown(
                         choices=["A4", "A3", "A2", "A1", "A0"],
@@ -135,78 +133,89 @@ def create_interface():
                         label="Orientation"
                     )
 
-            # Étape 2: Création Visuelle
-            with gr.Group():
-                gr.Markdown("### 🎨 Étape 2: Création Visuelle")
+            # Étape 2: Style et Contenu
+            with gr.Group(elem_classes="style-group"):
+                gr.Markdown("### 🎨 Style et Contenu")
                 with gr.Row():
-                    with gr.Column():
+                    with gr.Column(scale=1):
                         style = gr.Dropdown(
                             choices=list(ART_STYLES.keys()),
                             value="Neo Vintage",
                             label="Style artistique",
-                            info="Choisissez le style de votre affiche"
+                            info="Choisissez le style visuel"
                         )
-                        
+                        text_effect = gr.Dropdown(
+                            choices=list(TEXT_EFFECTS.keys()),
+                            value="Standard",
+                            label="Effet de texte",
+                            info="Style du texte"
+                        )
+                        collection = gr.Dropdown(
+                            choices=list(THEME_COLLECTIONS.keys()),
+                            label="Collection thématique",
+                            info="Thème global"
+                        )
+                    
+                    with gr.Column(scale=2):
                         subject = gr.Textbox(
                             label="Sujet principal",
-                            placeholder="Ex: loup, paysage urbain...",
-                            info="Décrivez l'élément principal"
+                            placeholder="Ex: loup, paysage urbain..."
                         )
-                        
-                        # Nouvelle section texte
-                        with gr.Group(elem_classes="text-input"):
-                            gr.Markdown("#### 📝 Texte de l'affiche")
-                            title = gr.Textbox(
-                                label="Titre principal",
-                                placeholder="Texte principal qui apparaîtra sur l'affiche",
-                                info="Ce texte sera intégré de manière lisible"
-                            )
-                            subtitle = gr.Textbox(
-                                label="Sous-titre (optionnel)",
-                                placeholder="Texte secondaire plus petit",
-                                info="Texte complémentaire"
-                            )
-                        
                         additional_details = gr.Textbox(
                             lines=2,
                             label="Détails additionnels (optionnel)",
-                            placeholder="Ajoutez des détails pour enrichir l'image...",
-                            info="Plus de détails = meilleur résultat"
+                            placeholder="Ajoutez des détails spécifiques..."
                         )
-                        
-                        with gr.Group(elem_classes="quality-controls"):
-                            quality = gr.Slider(
-                                minimum=30,
-                                maximum=50,
-                                value=35,
-                                step=5,
-                                label="Qualité",
-                                info="Influence le niveau de détail"
-                            )
-                            
-                            creativity = gr.Slider(
-                                minimum=5,
-                                maximum=15,
-                                value=7.5,
-                                step=0.5,
-                                label="Créativité",
-                                info="Balance entre fidélité et créativité"
-                            )
-                        
-                        generate_btn = gr.Button("✨ Générer", variant="primary")
-                        clear_btn = gr.Button("🗑️ Effacer", variant="secondary")
-                        
-                    image_output = gr.Image(
-                        label="Aperçu",
-                        type="pil"
+
+            # Étape 3: Texte
+            with gr.Group(elem_classes="text-effects"):
+                gr.Markdown("### ✍️ Texte de l'affiche")
+                with gr.Row():
+                    title = gr.Textbox(
+                        label="Titre principal",
+                        placeholder="Texte principal de l'affiche"
                     )
-                    
-            # Statut
-            status = gr.Textbox(
-                label="Status",
-                interactive=False,
-                value="Prêt à générer"
-            )
+                    subtitle = gr.Textbox(
+                        label="Sous-titre (optionnel)",
+                        placeholder="Texte secondaire"
+                    )
+
+            # Étape 4: Paramètres de génération
+            with gr.Group(elem_classes="style-group"):
+                gr.Markdown("### ⚙️ Paramètres")
+                with gr.Row():
+                    quality = gr.Slider(
+                        minimum=30,
+                        maximum=50,
+                        value=35,
+                        step=5,
+                        label="Qualité",
+                        info="Niveau de détail"
+                    )
+                    creativity = gr.Slider(
+                        minimum=5,
+                        maximum=15,
+                        value=7.5,
+                        step=0.5,
+                        label="Créativité",
+                        info="Balance entre fidélité et originalité"
+                    )
+
+            # Boutons de génération
+            with gr.Row():
+                generate_btn = gr.Button("✨ Générer", variant="primary")
+                clear_btn = gr.Button("🗑️ Effacer", variant="secondary")
+
+            # Zone de prévisualisation
+            with gr.Group(elem_classes="preview-panel"):
+                image_output = gr.Image(
+                    label="Aperçu",
+                    type="pil"
+                )
+                status = gr.Textbox(
+                    label="Statut",
+                    interactive=False
+                )
 
         # Événements
         generate_btn.click(
@@ -216,6 +225,8 @@ def create_interface():
                 orientation,
                 subject,
                 style,
+                text_effect,
+                collection,
                 title,
                 subtitle,
                 quality,