Update app.py

app.py

@@ -5,198 +5,296 @@ import requests
 import io
 import json
 
-# Styles artistiques professionnels
-ART_STYLES = {
-    "Illustration Vintage": {
-        "prompt_prefix": "professional vintage poster illustration, high detail screen print style, limited color palette",
+# Styles artistiques professionnels inspirés des meilleurs designers
+PRO_STYLES = {
+    "Neo Vintage": {
+        "prompt_prefix": "professional high-end vintage poster, screen printing style, halftone textures, limited color palette",
+        "guidance": 13.5,
+        "steps": 75,
+        "negative": "photorealistic, 3d, oversaturated, amateur",
+        "reference": "modern vintage illustration masters"
+    },
+    "Art Contemporain": {
+        "prompt_prefix": "contemporary art poster, bold artistic composition, professional design, gallery quality",
         "guidance": 12.5,
-        "steps": 70
+        "steps": 70,
+        "negative": "basic, simple, childish",
+        "reference": "modern art gallery style"
     },
-    "Art Déco": {
-        "prompt_prefix": "art deco style poster, geometric patterns, elegant composition, rich colors",
-        "guidance": 13.0,
-        "steps": 65
+    "Illustration Éditoriale": {
+        "prompt_prefix": "professional editorial illustration, sophisticated art style, magazine quality",
+        "guidance": 14.0,
+        "steps": 80,
+        "negative": "basic drawing, sketchy, unrefined",
+        "reference": "high-end magazine art"
     },
-    "Pop Art": {
-        "prompt_prefix": "pop art style, bold colors, halftone dots, comic book style, Andy Warhol inspired",
+    "Design Minimaliste Plus": {
+        "prompt_prefix": "premium minimalist design, refined composition, luxury poster style",
         "guidance": 11.5,
-        "steps": 60
-    },
-    "Minimaliste": {
-        "prompt_prefix": "minimalist design, clean lines, simple shapes, bold typography",
-        "guidance": 10.0,
-        "steps": 55
-    },
-    "Street Art": {
-        "prompt_prefix": "street art style, urban aesthetic, spray paint texture, bold contrast",
-        "guidance": 14.0,
-        "steps": 75
+        "steps": 65,
+        "negative": "cluttered, busy, complex",
+        "reference": "scandinavian design principles"
     }
 }
 
-# Collections thématiques
-COLLECTIONS = {
-    "Animaux Artistiques": {
-        "prompts": ["majestic animal portrait", "wildlife illustration"],
-        "negative": "realistic photo, blurry, photographic"
+# Collections artistiques spécialisées
+ARTISTIC_COLLECTIONS = {
+    "Portraits Animaliers Pro": {
+        "base_prompts": [
+            "majestic animal portrait",
+            "artistic wildlife illustration",
+            "powerful animal character"
+        ],
+        "style_elements": [
+            "dramatic lighting",
+            "expressive eyes",
+            "detailed textures",
+            "strong personality"
+        ],
+        "negative": "cute, cartoon, realistic photo",
+        "compositions": [
+            "centered portrait",
+            "dramatic angle",
+            "close-up detail"
+        ]
     },
-    "Nature Stylisée": {
-        "prompts": ["stylized botanical art", "decorative nature elements"],
-        "negative": "photorealistic, mundane, simple"
+    "Nature Graphique": {
+        "base_prompts": [
+            "graphic botanical art",
+            "stylized nature elements",
+            "organic patterns"
+        ],
+        "style_elements": [
+            "decorative details",
+            "flowing lines",
+            "dynamic composition"
+        ],
+        "negative": "photographic, plain, realistic",
+        "compositions": [
+            "geometric arrangement",
+            "pattern layout",
+            "abstract interpretation"
+        ]
     },
-    "Urbain Moderne": {
-        "prompts": ["modern urban design", "city life illustration"],
-        "negative": "traditional, rustic, old fashioned"
+    "Urban Art Plus": {
+        "base_prompts": [
+            "sophisticated urban art",
+            "street art fusion",
+            "modern city aesthetic"
+        ],
+        "style_elements": [
+            "street art textures",
+            "urban patterns",
+            "graffiti elements"
+        ],
+        "negative": "amateur graffiti, messy, unrefined",
+        "compositions": [
+            "dynamic urban composition",
+            "street art layout",
+            "modern city vibes"
+        ]
     }
 }
 
-def generate_pro_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')}"}
-    
-    # Paramètres optimisés pour version pro
-    dimensions = {
-        "A4": (1024, 1448),  # Augmenté pour qualité pro
-        "A3": (1448, 2048),
-        "A2": (2048, 2896),
-        "A1": (2896, 4096),
-        "A0": (4096, 5792)
-    }
-    
-    try:
-        width, height = dimensions[format_size]
-        if orientation == "Paysage":
-            width, height = height, width
-            
-        # Application du style artistique
-        style_config = ART_STYLES[style]
-        enhanced_prompt = f"{style_config['prompt_prefix']}, {prompt}"
-        
-        # Ajout des éléments de collection si spécifiés
-        if collection and collection in COLLECTIONS:
-            collection_prompts = COLLECTIONS[collection]["prompts"]
-            enhanced_prompt = f"{enhanced_prompt}, {', '.join(collection_prompts)}"
-            negative_prompt = COLLECTIONS[collection]["negative"]
-        else:
-            negative_prompt = "blurry, low quality, distorted, ugly, bad proportions"
-
-        payload = {
-            "inputs": enhanced_prompt,
-            "parameters": {
-                "negative_prompt": negative_prompt,
-                "num_inference_steps": style_config['steps'],
-                "guidance_scale": style_config['guidance'],
-                "width": width,
-                "height": height,
-                "quality": "production",  # Mode pro
-                "seed": int(creativity * 1000),  # Contrôle créatif
-                "num_images_per_prompt": 1,
-                "safety_checker": False,  # Plus de flexibilité artistique
-                "enhance_prompt": True,
-                "strength": quality / 100
+# Formats professionnels optimisés
+PRO_DIMENSIONS = {
+    "A4+": (1152, 1634),  # Optimisé pour qualité pro
+    "A3+": (1634, 2314),
+    "A2+": (2314, 3271),
+    "A1+": (3271, 4628),
+    "A0+": (4628, 6544)
+}
+
+class EquityArtisanPro:
+    def __init__(self):
+        self.model_url = "https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-xl-base-1.0"
+        self.headers = {"Authorization": f"Bearer {os.getenv('HUGGINGFACE_TOKEN')}"}
+
+    def generate_artwork(self, format_size, orientation, style, collection, prompt, quality, creativity):
+        try:
+            # Configuration des dimensions
+            width, height = PRO_DIMENSIONS[format_size]
+            if orientation == "Paysage":
+                width, height = height, width
+
+            # Construction du prompt artistique
+            style_config = PRO_STYLES[style]
+            collection_config = ARTISTIC_COLLECTIONS.get(collection)
+
+            # Prompt engineering avancé
+            enhanced_prompt = self._build_enhanced_prompt(
+                base_prompt=prompt,
+                style_config=style_config,
+                collection_config=collection_config
+            )
+
+            # Configuration avancée
+            payload = {
+                "inputs": enhanced_prompt,
+                "parameters": {
+                    "negative_prompt": self._build_negative_prompt(style_config, collection_config),
+                    "num_inference_steps": self._calculate_steps(style_config, quality),
+                    "guidance_scale": self._calculate_guidance(style_config, creativity),
+                    "width": width,
+                    "height": height,
+                    "seed": int(creativity * 1000)  # Contrôle créatif
+                }
             }
-        }
-        
-        response = requests.post(
-            API_URL,
-            headers=headers,
-            json=payload,
-            timeout=60  # Augmenté pour qualité pro
-        )
-        
-        if response.status_code == 200:
-            image = Image.open(io.BytesIO(response.content))
+
+            print(f"Generating with enhanced prompt: {enhanced_prompt[:100]}...")
             
-            # Post-traitement pro
+            response = requests.post(
+                self.model_url,
+                headers=self.headers,
+                json=payload,
+                timeout=60
+            )
+
+            if response.status_code == 200:
+                image = Image.open(io.BytesIO(response.content))
+                image = self._post_process_image(image, quality)
+                return image, "✨ Chef d'œuvre créé avec succès!"
+            else:
+                return None, f"⚠️ Erreur {response.status_code}: {response.text}"
+
+        except Exception as e:
+            print(f"Error: {str(e)}")
+            return None, f"⚠️ Erreur: {str(e)}"
+
+    def _build_enhanced_prompt(self, base_prompt, style_config, collection_config=None):
+        prompt_parts = [
+            style_config["prompt_prefix"],
+            base_prompt
+        ]
+
+        if collection_config:
+            prompt_parts.extend([
+                f"in style of {', '.join(collection_config['style_elements'])}",
+                f"with {', '.join(collection_config['base_prompts'])}",
+                f"featuring {', '.join(collection_config['compositions'])}"
+            ])
+
+        return ", ".join(prompt_parts) + ", professional poster design, masterpiece quality"
+
+    def _build_negative_prompt(self, style_config, collection_config=None):
+        negative_parts = [
+            style_config["negative"],
+            "poor quality, amateur design, basic composition"
+        ]
+        
+        if collection_config:
+            negative_parts.append(collection_config["negative"])
+
+        return ", ".join(negative_parts)
+
+    def _calculate_steps(self, style_config, quality):
+        base_steps = style_config["steps"]
+        return int(base_steps * (quality/85))  # Scaled by quality
+
+    def _calculate_guidance(self, style_config, creativity):
+        base_guidance = style_config["guidance"]
+        return base_guidance * (creativity/10)  # Scaled by creativity
+
+    def _post_process_image(self, image, quality):
+        if quality > 85:
             enhancer = ImageEnhance.Contrast(image)
             image = enhancer.enhance(1.2)
             enhancer = ImageEnhance.Sharpness(image)
             image = enhancer.enhance(1.3)
-            
-            return image, "✨ Création artistique réussie !"
-        return None, f"⚠️ Erreur {response.status_code}: Ajustez les paramètres"
-            
-    except Exception as e:
-        print(f"Erreur détaillée: {str(e)}")
-        return None, f"⚠️ Erreur: {str(e)}"
+        return image
 
 def create_interface():
-    with gr.Blocks(theme=gr.themes.Soft(primary_hue="purple")) as app:
+    artisan = EquityArtisanPro()
+
+    with gr.Blocks(theme=gr.themes.Soft(
+        primary_hue="indigo",
+        secondary_hue="purple",
+    )) as app:
         gr.HTML("""
-            <div style="text-align: center; padding: 20px;">
-                <h1 style="color: #7C3AED;">🎨 Equity Artisan 3.0 Pro</h1>
-                <p>Créez des affiches artistiques uniques avec notre IA spécialisée</p>
+            <div style="text-align: center; padding: 20px; background: linear-gradient(135deg, #4F46E5 0%, #7C3AED 100%); border-radius: 16px; margin-bottom: 24px; box-shadow: 0 4px 6px rgba(0, 0, 0, 0.1);">
+                <h1 style="color: white; font-size: 2.5em; margin-bottom: 10px;">🎨 Equity Artisan 3.0</h1>
+                <p style="color: #E5E7EB; font-size: 1.2em;">Créez des œuvres d'art uniques avec notre IA spécialisée</p>
             </div>
         """)
-        
+
         with gr.Row():
+            # Panneau de contrôle
             with gr.Column(scale=1):
-                format_choice = gr.Dropdown(
-                    choices=list(dimensions.keys()),
-                    value="A4",
-                    label="Format"
-                )
-                orientation = gr.Radio(
-                    choices=["Portrait", "Paysage"],
-                    value="Portrait",
-                    label="Orientation"
-                )
-                
-                style = gr.Dropdown(
-                    choices=list(ART_STYLES.keys()),
-                    value="Illustration Vintage",
-                    label="Style Artistique"
-                )
-                
-                collection = gr.Dropdown(
-                    choices=list(COLLECTIONS.keys()),
-                    label="Collection Thématique (optionnel)"
-                )
-                
                 with gr.Group():
+                    gr.Markdown("### 📐 Format & Style")
+                    format_choice = gr.Dropdown(
+                        choices=list(PRO_DIMENSIONS.keys()),
+                        value="A4+",
+                        label="Format",
+                        info="Formats optimisés pour qualité professionnelle"
+                    )
+                    orientation = gr.Radio(
+                        choices=["Portrait", "Paysage"],
+                        value="Portrait",
+                        label="Orientation"
+                    )
+                    style = gr.Dropdown(
+                        choices=list(PRO_STYLES.keys()),
+                        value="Neo Vintage",
+                        label="Style Artistique",
+                        info="Styles inspirés des grands maîtres"
+                    )
+
+                with gr.Group():
+                    gr.Markdown("### 🎨 Vision Artistique")
+                    collection = gr.Dropdown(
+                        choices=list(ARTISTIC_COLLECTIONS.keys()),
+                        label="Collection",
+                        info="Thèmes artistiques spécialisés"
+                    )
                     prompt = gr.Textbox(
                         lines=3,
-                        label="Description Artistique",
-                        placeholder="Décrivez votre vision créative..."
+                        label="Description",
+                        placeholder="Décrivez votre vision artistique...",
+                        info="Soyez précis et créatif"
                     )
-                    
+
+                with gr.Group():
+                    gr.Markdown("### ✨ Paramètres Créatifs")
                     quality = gr.Slider(
-                        minimum=50,
+                        minimum=70,
                         maximum=100,
                         value=85,
-                        label="Qualité Artistique"
+                        label="Qualité Artistique",
+                        info="Influence la finesse des détails"
                     )
-                    
                     creativity = gr.Slider(
-                        minimum=1,
-                        maximum=20,
-                        value=12,
-                        label="Niveau de Créativité"
+                        minimum=7,
+                        maximum=15,
+                        value=10,
+                        label="Expression Créative",
+                        info="Balance entre fidélité et originalité"
                     )
-                    
-                    generate_btn = gr.Button("✨ Créer", variant="primary")
 
+                generate_btn = gr.Button("✨ Créer", variant="primary", scale=1)
+
+            # Zone de prévisualisation
             with gr.Column(scale=2):
                 image_output = gr.Image(
                     label="Votre Création",
                     type="pil"
                 )
                 status = gr.Textbox(
-                    label="Status",
+                    label="Statut",
                     interactive=False
                 )
 
         # Événements
         generate_btn.click(
-            generate_pro_image,
+            artisan.generate_artwork,
             inputs=[
                 format_choice,
                 orientation,
-                prompt,
                 style,
+                collection,
+                prompt,
                 quality,
-                creativity,
-                collection
+                creativity
             ],
             outputs=[image_output, status]
         )