Update app.py

app.py

@@ -4,6 +4,83 @@ from PIL import Image, ImageDraw, ImageFont
 import requests
 import io
 
+# Configuration des styles artistiques
+ART_STYLES = {
+    "Renaissance": {
+        "prompt_prefix": "ultra-detailed renaissance style artwork, masterful anatomical details, dramatic chiaroscuro lighting",
+        "guidance": 15.0,
+        "steps": 80,
+        "negative_prompt": "blurry, low quality, flat lighting, cartoon, modern"
+    },
+    "Impressionnisme": {
+        "prompt_prefix": "impressionist art style, vibrant brushstrokes, atmospheric lighting, monet-like technique",
+        "guidance": 12.0,
+        "steps": 65,
+        "negative_prompt": "sharp details, photorealistic, digital art"
+    },
+    "Art Moderne": {
+        "prompt_prefix": "modern art composition, bold geometric shapes, abstract elements, strong visual impact",
+        "guidance": 11.0,
+        "steps": 60,
+        "negative_prompt": "realistic, traditional, detailed texture"
+    },
+    "Illustration Pro": {
+        "prompt_prefix": "professional editorial illustration, sophisticated art style, high-end magazine quality",
+        "guidance": 13.5,
+        "steps": 70,
+        "negative_prompt": "amateur, basic, photographic"
+    },
+    "Néo Vintage": {
+        "prompt_prefix": "vintage poster design, retro aesthetic, limited color palette, screen printing style",
+        "guidance": 12.5,
+        "steps": 65,
+        "negative_prompt": "modern, digital, photorealistic"
+    },
+    "Encre Artistique": {
+        "prompt_prefix": "artistic ink illustration, dynamic brushwork, high contrast black and white, expressive strokes",
+        "guidance": 14.0,
+        "steps": 75,
+        "negative_prompt": "color, photographic, digital"
+    },
+    "Pop Art": {
+        "prompt_prefix": "pop art style, bold colors, halftone dots, comic book aesthetic, Warhol-inspired",
+        "guidance": 11.5,
+        "steps": 60,
+        "negative_prompt": "realistic, painterly, subtle"
+    },
+    "Dark Fantasy": {
+        "prompt_prefix": "dark fantasy art, atmospheric mood, dramatic lighting, mythological elements",
+        "guidance": 13.0,
+        "steps": 70,
+        "negative_prompt": "bright, cheerful, realistic photo"
+    }
+}
+
+# Collections thématiques
+COLLECTIONS = {
+    "Nature Artistique": {
+        "prompts": ["artistic nature illustration", "stylized botanical elements"],
+        "styles": ["organic composition", "natural textures", "dynamic flow"],
+        "negative": "artificial, synthetic, urban"
+    },
+    "Portraits Expressifs": {
+        "prompts": ["expressive portrait art", "emotional character illustration"],
+        "styles": ["dramatic lighting", "powerful expression", "artistic interpretation"],
+        "negative": "flat lighting, emotionless, stiff pose"
+    },
+    "Urbain Stylisé": {
+        "prompts": ["stylized urban art", "city aesthetic interpretation"],
+        "styles": ["architectural elements", "urban textures", "street art influence"],
+        "negative": "rural, natural, realistic photo"
+    },
+    "Fantaisie Épique": {
+        "prompts": ["epic fantasy artwork", "mythological interpretation"],
+        "styles": ["magical elements", "dramatic atmosphere", "legendary creatures"],
+        "negative": "mundane, realistic, modern"
+    }
+}
+
+# CSS personnalisé
 CUSTOM_CSS = """
 .container { max-width: 1200px; margin: auto; }
 .welcome { text-align: center; margin: 20px 0; padding: 20px; background: #1e293b; border-radius: 10px; }
@@ -11,57 +88,58 @@ CUSTOM_CSS = """
 """
 
 def create_interface():
-    def generate_image(format_size, orientation, prompt, style, quality, creativity, negative_prompt=""):
+    def generate_image(format_size, orientation, prompt, style, quality, creativity, collection=None):
         API_URL = "https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-xl-base-1.0"
         headers = {"Authorization": f"Bearer {os.getenv('HUGGINGFACE_TOKEN')}"}
         
-        # Dimensions de base optimisées
-        base_width = 1024
-        base_height = 1024
-        
-        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
-        
         try:
-            # Styles prédéfinis
-            style_prompts = {
-                "Moderne": "modern clean poster design, professional layout",
-                "Artistique": "artistic poster composition, creative design",
-                "Corporate": "professional business poster, corporate style",
-                "Impact": "bold impactful poster design, attention-grabbing"
-            }
+            # Dimensions de base optimisées
+            base_width = 1024
+            base_height = 1024
             
-            enhanced_prompt = f"{style_prompts[style]}, {prompt}, high quality, detailed"
-            default_negative = "blurry, low quality, distorted, ugly, bad proportions, weird colors"
-            final_negative = negative_prompt if negative_prompt.strip() else default_negative
+            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
+
+            # Récupération des paramètres du style
+            style_config = ART_STYLES[style]
+            enhanced_prompt = f"{style_config['prompt_prefix']}, {prompt}"
             
+            if collection and collection in COLLECTIONS:
+                collection_data = COLLECTIONS[collection]
+                collection_elements = collection_data["prompts"] + collection_data["styles"]
+                enhanced_prompt += f", {', '.join(collection_elements)}"
+                negative_prompt = f"{style_config['negative_prompt']}, {collection_data['negative']}"
+            else:
+                negative_prompt = style_config['negative_prompt']
+
+            # Configuration de la génération
             payload = {
                 "inputs": enhanced_prompt,
                 "parameters": {
-                    "negative_prompt": final_negative,
-                    "num_inference_steps": min(max(30, int(quality)), 50),
-                    "guidance_scale": min(max(5, creativity), 15),
+                    "negative_prompt": negative_prompt,
+                    "num_inference_steps": int(style_config['steps'] * (quality/100)),
+                    "guidance_scale": style_config['guidance'] * (creativity/10),
                     "width": base_width,
                     "height": base_height
                 }
             }
             
-            print(f"Envoi de la requête avec les paramètres: {payload}")
+            print(f"Génération avec paramètres: {payload}")
             
-            response = requests.post(API_URL, headers=headers, json=payload, timeout=30)
+            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, "✨ Image générée avec succès!"
+                return image, f"✨ Création {style} réussie!"
             else:
                 print(f"Erreur API: {response.text}")
-                return None, f"⚠️ Erreur {response.status_code}: Essayez de modifier vos paramètres"
+                return None, f"⚠️ Erreur {response.status_code}: Ajustez les paramètres"
             
         except Exception as e:
             print(f"Exception: {str(e)}")
@@ -98,23 +176,21 @@ def create_interface():
                 with gr.Row():
                     with gr.Column():
                         style = gr.Dropdown(
-                            choices=["Moderne", "Artistique", "Corporate", "Impact"],
-                            value="Moderne",
-                            label="Style visuel"
+                            choices=list(ART_STYLES.keys()),
+                            value="Renaissance",
+                            label="Style artistique"
+                        )
+                        
+                        collection = gr.Dropdown(
+                            choices=list(COLLECTIONS.keys()),
+                            label="Collection (optionnel)",
+                            info="Thèmes artistiques spécialisés"
                         )
                         
                         prompt = gr.Textbox(
                             lines=3,
                             label="Description",
-                            placeholder="Décrivez l'affiche que vous souhaitez...",
-                            value="professional poster design with clean layout"
-                        )
-
-                        # Ajout du champ negative_prompt
-                        negative_prompt = gr.Textbox(
-                            label="Éléments à éviter (optionnel)",
-                            placeholder="Ex: flou, texte illisible, distorsions...",
-                            lines=2
+                            placeholder="Décrivez votre vision artistique..."
                         )
                         
                         with gr.Group(elem_classes="quality-controls"):
@@ -161,7 +237,7 @@ def create_interface():
                 style,
                 quality,
                 creativity,
-                negative_prompt  # Ajout du negative_prompt aux inputs
+                collection
             ],
             outputs=[image_output, status]
         )