Update app.py

app.py

@@ -19,64 +19,97 @@ logger = logging.getLogger(__name__)
 # Chargement des variables d'environnement
 load_dotenv()
 
-# Styles artistiques étendus
+# Styles artistiques complets
 ART_STYLES = {
-    "Art Moderne": {
-        "prompt_prefix": "modern art style poster, professional design",
-        "negative_prompt": "traditional, photorealistic, cluttered, busy design"
+    # Styles Classiques
+    "Photoréaliste": {
+        "prompt_prefix": "hyperrealistic photograph, extremely detailed, studio quality, professional photography, 8k uhd",
+        "negative_prompt": "artistic, painterly, abstract, cartoon, illustration, low quality"
     },
-    "Neo Vintage": {
-        "prompt_prefix": "vintage style advertising poster, retro design",
-        "negative_prompt": "modern, digital, contemporary style"
+    "Expressionniste": {
+        "prompt_prefix": "expressive painting style, intense emotional art, bold brushstrokes, vibrant colors, van gogh inspired",
+        "negative_prompt": "realistic, subtle, photographic, clean lines, digital art"
+    },
+    "Impressionniste": {
+        "prompt_prefix": "impressionist painting style, soft light, visible brushstrokes, outdoor scene, monet inspired",
+        "negative_prompt": "sharp details, high contrast, digital, modern"
+    },
+    "Art Abstrait": {
+        "prompt_prefix": "abstract art, geometric shapes, non-representational, kandinsky style, pure artistic expression",
+        "negative_prompt": "realistic, figurative, photographic, literal"
+    },
+    "Surréaliste": {
+        "prompt_prefix": "surrealist art, dreamlike imagery, symbolic elements, dali inspired, metaphysical art",
+        "negative_prompt": "realistic, conventional, ordinary, literal"
+    },
+
+    # Styles Modernes
+    "Art Moderne": {
+        "prompt_prefix": "modern art style poster, professional design, contemporary aesthetic",
+        "negative_prompt": "traditional, cluttered, busy design, vintage"
     },
     "Pop Art": {
-        "prompt_prefix": "pop art style poster, bold design",
-        "negative_prompt": "subtle, realistic, traditional art"
+        "prompt_prefix": "pop art style poster, bold colors, repeated patterns, screen print effect, warhol inspired",
+        "negative_prompt": "subtle, realistic, traditional, painterly"
     },
     "Minimaliste": {
-        "prompt_prefix": "minimalist design poster, clean composition",
-        "negative_prompt": "complex, detailed, ornate, busy"
+        "prompt_prefix": "minimalist design poster, clean composition, elegant simplicity",
+        "negative_prompt": "complex, detailed, ornate, busy, cluttered"
+    },
+    "Cubiste": {
+        "prompt_prefix": "cubist art style, geometric fragmentation, multiple perspectives, picasso inspired",
+        "negative_prompt": "realistic, single perspective, traditional, photographic"
+    },
+    "Futuriste": {
+        "prompt_prefix": "futuristic art style, dynamic movement, technological elements, speed and motion",
+        "negative_prompt": "static, traditional, classical, historical"
+    },
+
+    # Styles Spéciaux
+    "Neo Vintage": {
+        "prompt_prefix": "vintage style advertising poster, retro design, classic aesthetic",
+        "negative_prompt": "modern, digital, contemporary style"
     },
     "Cyberpunk": {
-        "prompt_prefix": "cyberpunk style poster, neon lights, futuristic design",
+        "prompt_prefix": "cyberpunk style poster, neon lights, futuristic design, high-tech aesthetic",
         "negative_prompt": "vintage, natural, rustic, traditional"
     },
-    "Aquarelle": {
-        "prompt_prefix": "watercolor art style poster, fluid artistic design",
-        "negative_prompt": "digital, sharp, photorealistic"
+    "Japonais": {
+        "prompt_prefix": "japanese art style poster, ukiyo-e inspired design, traditional japanese aesthetic",
+        "negative_prompt": "western, modern, photographic"
     },
     "Art Déco": {
-        "prompt_prefix": "art deco style poster, geometric patterns, luxury design",
-        "negative_prompt": "modern, minimalist, casual"
+        "prompt_prefix": "art deco style poster, geometric patterns, luxury design, 1920s aesthetic",
+        "negative_prompt": "modern, minimalist, casual, contemporary"
     },
-    "Japonais": {
-        "prompt_prefix": "japanese art style poster, ukiyo-e inspired design",
-        "negative_prompt": "western, modern, photographic"
+    "Symboliste": {
+        "prompt_prefix": "symbolic art, decorative patterns, gold elements, mystical atmosphere, klimt inspired",
+        "negative_prompt": "realistic, simple, plain, literal"
     }
 }
 
-# Paramètres de composition
+# Paramètres de composition enrichis
 COMPOSITION_PARAMS = {
     "Layouts": {
-        "Centré": "centered composition, balanced layout",
-        "Asymétrique": "dynamic asymmetrical composition",
-        "Grille": "grid-based layout, structured composition",
-        "Diagonal": "diagonal dynamic composition",
-        "Minimaliste": "minimal composition, lots of whitespace"
+        "Centré": "centered composition, balanced layout, harmonious arrangement",
+        "Asymétrique": "dynamic asymmetrical composition, creative balance",
+        "Grille": "grid-based layout, structured composition, organized design",
+        "Diagonal": "diagonal dynamic composition, energetic flow",
+        "Minimaliste": "minimal composition, lots of whitespace, elegant spacing"
     },
     "Ambiances": {
-        "Dramatique": "dramatic lighting, high contrast",
-        "Doux": "soft lighting, gentle atmosphere",
-        "Vibrant": "vibrant colors, energetic mood",
-        "Mystérieux": "mysterious atmosphere, moody lighting",
-        "Serein": "peaceful atmosphere, calm mood"
+        "Dramatique": "dramatic lighting, high contrast, intense mood",
+        "Doux": "soft lighting, gentle atmosphere, subtle mood",
+        "Vibrant": "vibrant colors, energetic mood, dynamic atmosphere",
+        "Mystérieux": "mysterious atmosphere, moody lighting, enigmatic feel",
+        "Serein": "peaceful atmosphere, calm mood, tranquil setting"
     },
     "Palette": {
-        "Monochrome": "monochromatic color scheme",
-        "Contrasté": "high contrast color palette",
-        "Pastel": "soft pastel color palette",
-        "Terre": "earthy color palette",
-        "Néon": "neon color palette"
+        "Monochrome": "monochromatic color scheme, sophisticated tones",
+        "Contrasté": "high contrast color palette, bold color combinations",
+        "Pastel": "soft pastel color palette, gentle colors",
+        "Terre": "earthy color palette, natural tones",
+        "Néon": "neon color palette, vibrant glowing colors"
     }
 }
 
@@ -90,8 +123,8 @@ class ImageGenerator:
         logger.info("ImageGenerator initialisé")
 
     def _build_prompt(self, params: Dict[str, Any]) -> str:
-        """Construction de prompt améliorée"""
-        style_info = ART_STYLES.get(params["style"], ART_STYLES["Neo Vintage"])
+        """Construction de prompt améliorée avec plus de détails"""
+        style_info = ART_STYLES.get(params["style"], ART_STYLES["Art Moderne"])
         prompt = f"{style_info['prompt_prefix']}, {params['subject']}"
 
         # Ajout des paramètres de composition
@@ -105,15 +138,15 @@ class ImageGenerator:
         # Ajout des ajustements fins
         if params.get("detail_level"):
             detail_strength = params["detail_level"]
-            prompt += f", {'highly detailed' if detail_strength > 7 else 'moderately detailed'}"
+            prompt += f", {'highly detailed, intricate' if detail_strength > 7 else 'moderately detailed'}"
 
         if params.get("contrast"):
             contrast_strength = params["contrast"]
-            prompt += f", {'high contrast' if contrast_strength > 7 else 'balanced contrast'}"
+            prompt += f", {'high contrast, dramatic lighting' if contrast_strength > 7 else 'balanced contrast'}"
 
         if params.get("saturation"):
             saturation_strength = params["saturation"]
-            prompt += f", {'vibrant colors' if saturation_strength > 7 else 'subtle colors'}"
+            prompt += f", {'vibrant saturated colors' if saturation_strength > 7 else 'subtle muted colors'}"
 
         if params.get("title"):
             prompt += f", with text saying '{params['title']}'"
@@ -130,7 +163,6 @@ class ImageGenerator:
 
             prompt = self._build_prompt(params)
             
-            # Configuration de base
             payload = {
                 "inputs": prompt,
                 "parameters": {
@@ -187,7 +219,6 @@ def create_interface():
         """)
         
         with gr.Column(elem_classes="container"):
-            # Format et Orientation
             with gr.Group(elem_classes="controls-group"):
                 gr.Markdown("### 📐 Format et Orientation")
                 with gr.Row():
@@ -202,13 +233,12 @@ def create_interface():
                         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="Neo Vintage",
+                        value="Art Moderne",
                         label="Style artistique"
                     )
                     layout = gr.Dropdown(
@@ -229,7 +259,6 @@ def create_interface():
                         label="Palette"
                     )
             
-            # Contenu
             with gr.Group(elem_classes="controls-group"):
                 gr.Markdown("### 📝 Contenu")
                 subject = gr.Textbox(
@@ -241,7 +270,6 @@ def create_interface():
                     placeholder="Titre de l'affiche..."
                 )
             
-            # Ajustements fins
             with gr.Group(elem_classes="advanced-controls"):
                 gr.Markdown("### 🎯 Ajustements Fins")
                 with gr.Row():
@@ -267,7 +295,6 @@ def create_interface():
                         label="Saturation"
                     )
             
-            # Paramètres de génération
             with gr.Group(elem_classes="controls-group"):
                 with gr.Row():
                     quality = gr.Slider(
@@ -283,12 +310,10 @@ def create_interface():
                         label="Créativité"
                     )
             
-            # Boutons
             with gr.Row():
                 generate_btn = gr.Button("✨ Générer", variant="primary")
                 clear_btn = gr.Button("🗑️ Effacer", variant="secondary")
             
-            # Sortie
             image_output = gr.Image(label="Aperçu")
             status = gr.Textbox(label="Statut", interactive=False)