Update app.py

app.py

@@ -4,112 +4,87 @@ from PIL import Image, ImageDraw, ImageFont
 import requests
 import io
 
-# Dictionnaire de sujets spécifiques
-SUBJECT_PROMPTS = {
-    "insectes": {
-        "libellule": "hyper realistic dragonfly in flight with translucent wings, iridescent body, macro photography, detailed insect illustration, natural lighting, entomological precision",
-        "papillon": "detailed butterfly with intricate wing patterns, highly detailed macro photography, precise lepidoptera illustration",
-        "scarabée": "detailed beetle with shiny carapace, precise entomological macro photography"
-    },
-    "animaux": {
-        "loup": "majestic wolf portrait, detailed fur texture, highly detailed wild canine features, expressive eyes, professional wildlife photography",
-        "lion": "majestic lion portrait, detailed mane, powerful feline features, dramatic lighting, professional wildlife photography",
-        "renard": "elegant fox portrait, detailed fur texture, cunning vulpine features, professional wildlife photography"
-    }
-}
-
-# Styles artistiques améliorés
+# Styles artistiques optimisés pour le texte
 ART_STYLES = {
-    "Illustration Pro": {
-        "prompt_prefix": "professional high-end illustration, perfect composition, editorial quality artwork",
-        "style_modifiers": "sharp clean lines, elegant design, magazine quality finish",
-        "guidance": 13.5,
-        "steps": 70,
-        "negative_prompt": "amateur, sketchy, rough, unfinished, low quality"
-    },
     "Neo Vintage": {
-        "prompt_prefix": "vintage style poster art, retro illustration, classic design aesthetic",
-        "style_modifiers": "halftone texture, limited color palette, aged paper effect, vintage typography",
+        "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": 65,
-        "negative_prompt": "modern, digital, glossy, contemporary"
-    },
-    "Art Moderne": {
-        "prompt_prefix": "modern art style, contemporary design, abstract artistic interpretation",
-        "style_modifiers": "geometric shapes, bold colors, minimalist composition",
-        "guidance": 11.0,
-        "steps": 60,
-        "negative_prompt": "classical, photorealistic, traditional"
+        "steps": 70,
+        "negative_prompt": "blurry text, illegible writing, damaged letters, distorted fonts, gibberish"
     },
-    "Dark Fantasy": {
-        "prompt_prefix": "dark fantasy art style, atmospheric mood, dramatic fantasy illustration",
-        "style_modifiers": "mystical atmosphere, dramatic lighting, ethereal quality",
-        "guidance": 14.0,
+    "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": "cheerful, bright, realistic photo, mundane"
+        "negative_prompt": "modern fonts, messy text, unclear letters, random characters"
     },
-    "Pop Art": {
-        "prompt_prefix": "pop art style illustration, bold graphic design, comic book aesthetic",
-        "style_modifiers": "vibrant colors, halftone dots, bold outlines, Warhol-inspired",
+    "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": "realistic, subtle, painterly, traditional"
+        "negative_prompt": "subtle text, unclear typography, messy letters"
     }
 }
 
+# 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; }
-.style-info { padding: 10px; background: #2d3748; border-radius: 5px; margin-top: 10px; }
+.text-input { background: #2d3748; padding: 15px; border-radius: 5px; margin: 10px 0; }
 """
 
-def enhance_prompt(subject, style, additional_details=""):
-    """Améliore le prompt avec des détails spécifiques"""
-    # Recherche dans le dictionnaire de sujets
-    base_prompt = None
-    for category, subjects in SUBJECT_PROMPTS.items():
-        if subject.lower() in subjects:
-            base_prompt = subjects[subject.lower()]
-            break
+def generate_enhanced_prompt(subject, style, title="", subtitle="", additional_details=""):
+    """Génère un prompt optimisé pour le texte"""
+    style_config = ART_STYLES[style]
+    
+    # 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']}"
     
-    if not base_prompt:
-        base_prompt = f"highly detailed {subject}"
-
-    # Ajout des détails spécifiques
     if additional_details:
-        base_prompt = f"{base_prompt}, {additional_details}"
-
-    # Application du style
-    style_config = ART_STYLES[style]
-    final_prompt = f"{style_config['prompt_prefix']}, {base_prompt}, {style_config['style_modifiers']}"
+        base_prompt += f", {additional_details}"
     
-    return final_prompt, style_config['negative_prompt']
+    return base_prompt, style_config['negative_prompt']
 
-def generate_image(format_size, orientation, subject, style, quality, creativity, additional_details=""):
+def generate_image(format_size, orientation, subject, style, title, subtitle, quality, creativity, additional_details=""):
     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 de base optimisées
-        base_width = 1024
-        base_height = 1024
+        # Dimensions optimisées
+        width, height = (768, 1024) if format_size == "A4" else (1024, 1024)
+        if orientation == "Paysage":
+            width, height = height, width
+            
+        # Génération du prompt optimisé
+        enhanced_prompt, negative_prompt = generate_enhanced_prompt(
+            subject, style, title, subtitle, additional_details
+        )
         
-        if orientation == "Portrait":
-            if format_size in ["A4", "A3"]:
-                base_width = 768
-                base_height = 1024
-        else:
-            if format_size in ["A4", "A3"]:
-                base_width = 1024
-                base_height = 768
-
-        # Construction du prompt optimisé
-        enhanced_prompt, negative_prompt = enhance_prompt(subject, style, additional_details)
+        print(f"Prompt: {enhanced_prompt}")  # Debug
+        print(f"Negative prompt: {negative_prompt}")  # Debug
         
-        print(f"Prompt final: {enhanced_prompt}")  # Debug
-        print(f"Prompt négatif: {negative_prompt}")  # Debug
-
         # Configuration de la requête
         payload = {
             "inputs": enhanced_prompt,
@@ -117,18 +92,16 @@ def generate_image(format_size, orientation, subject, style, quality, creativity
                 "negative_prompt": negative_prompt,
                 "num_inference_steps": int(ART_STYLES[style]['steps'] * (quality/100)),
                 "guidance_scale": ART_STYLES[style]['guidance'] * (creativity/10),
-                "width": base_width,
-                "height": base_height
+                "width": width,
+                "height": height
             }
         }
         
-        print(f"Paramètres de génération: {payload}")  # Debug
-        
         response = requests.post(API_URL, headers=headers, json=payload, timeout=60)
         
         if response.status_code == 200:
             image = Image.open(io.BytesIO(response.content))
-            return image, f"✨ Création {style} réussie!"
+            return image, f"✨ Création {style} réussie avec texte intégré!"
         else:
             print(f"Erreur API: {response.text}")  # Debug
             return None, f"⚠️ Erreur {response.status_code}: Ajustez les paramètres"
@@ -162,30 +135,42 @@ def create_interface():
                         label="Orientation"
                     )
 
-            # Étape 2: Création
+            # Étape 2: Création Visuelle
             with gr.Group():
                 gr.Markdown("### 🎨 Étape 2: Création Visuelle")
-                
                 with gr.Row():
                     with gr.Column():
                         style = gr.Dropdown(
                             choices=list(ART_STYLES.keys()),
-                            value="Illustration Pro",
+                            value="Neo Vintage",
                             label="Style artistique",
-                            info="Choisissez votre style d'illustration"
+                            info="Choisissez le style de votre affiche"
                         )
                         
                         subject = gr.Textbox(
-                            lines=1,
                             label="Sujet principal",
-                            placeholder="Ex: libellule, loup, paysage...",
-                            info="Décrivez le sujet principal"
+                            placeholder="Ex: loup, paysage urbain...",
+                            info="Décrivez l'élément principal"
                         )
                         
+                        # 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 spécifiques pour enrichir l'image...",
+                            placeholder="Ajoutez des détails pour enrichir l'image...",
                             info="Plus de détails = meilleur résultat"
                         )
                         
@@ -231,6 +216,8 @@ def create_interface():
                 orientation,
                 subject,
                 style,
+                title,
+                subtitle,
                 quality,
                 creativity,
                 additional_details