Update app.py

app.py

@@ -1,303 +1,175 @@
 import gradio as gr
 import os
-from PIL import Image, ImageDraw, ImageFont, ImageEnhance
+from PIL import Image, ImageDraw, ImageFont
 import requests
 import io
-import json
 
-# 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,
-        "negative": "basic, simple, childish",
-        "reference": "modern art gallery style"
-    },
-    "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"
-    },
-    "Design Minimaliste Plus": {
-        "prompt_prefix": "premium minimalist design, refined composition, luxury poster style",
-        "guidance": 11.5,
-        "steps": 65,
-        "negative": "cluttered, busy, complex",
-        "reference": "scandinavian design principles"
-    }
-}
+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; }
+"""
 
-# 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 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"
-        ]
-    },
-    "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"
-        ]
-    }
-}
-
-# 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):
+def create_interface():
+    def generate_image(format_size, orientation, prompt, style, quality, creativity, negative_prompt=""):
+        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:
-            # 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
+            # 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"
+            }
+            
+            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
+            
             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
+                    "negative_prompt": final_negative,
+                    "num_inference_steps": min(max(30, int(quality)), 50),
+                    "guidance_scale": min(max(5, creativity), 15),
+                    "width": base_width,
+                    "height": base_height
                 }
             }
-
-            print(f"Generating with enhanced prompt: {enhanced_prompt[:100]}...")
             
-            response = requests.post(
-                self.model_url,
-                headers=self.headers,
-                json=payload,
-                timeout=60
-            )
-
+            print(f"Envoi de la requête avec les paramètres: {payload}")
+            
+            response = requests.post(API_URL, headers=headers, json=payload, timeout=30)
+            
             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!"
+                return image, "✨ Image générée avec succès!"
             else:
-                return None, f"⚠️ Erreur {response.status_code}: {response.text}"
-
+                print(f"Erreur API: {response.text}")
+                return None, f"⚠️ Erreur {response.status_code}: Essayez de modifier vos paramètres"
+            
         except Exception as e:
-            print(f"Error: {str(e)}")
+            print(f"Exception: {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
-
-def create_interface():
-    artisan = EquityArtisanPro()
-
-    with gr.Blocks(theme=gr.themes.Soft(
-        primary_hue="indigo",
-        secondary_hue="purple",
-    )) as app:
+    with gr.Blocks(css=CUSTOM_CSS) as app:
         gr.HTML("""
-            <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 class="welcome">
+                <h1>🤖 Equity Artisan 3.0</h1>
+                <p>Créez des affiches professionnelles étape par étape avec notre assistant créatif augmenté.</p>
             </div>
         """)
-
-        with gr.Row():
-            # Panneau de contrôle
-            with gr.Column(scale=1):
-                with gr.Group():
-                    gr.Markdown("### 📐 Format & Style")
+        
+        with gr.Column(elem_classes="container"):
+            # Étape 1: Format
+            with gr.Group():
+                gr.Markdown("### 📏 Étape 1: Format et Orientation")
+                with gr.Row():
                     format_choice = gr.Dropdown(
-                        choices=list(PRO_DIMENSIONS.keys()),
-                        value="A4+",
-                        label="Format",
-                        info="Formats optimisés pour qualité professionnelle"
+                        choices=["A4", "A3", "A2", "A1", "A0"],
+                        value="A4",
+                        label="Format"
                     )
                     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",
-                        placeholder="Décrivez votre vision artistique...",
-                        info="Soyez précis et créatif"
+            # Étape 2: Création
+            with gr.Group():
+                gr.Markdown("### 🎨 Étape 2: Création Visuelle")
+                
+                with gr.Row():
+                    with gr.Column():
+                        style = gr.Dropdown(
+                            choices=["Moderne", "Artistique", "Corporate", "Impact"],
+                            value="Moderne",
+                            label="Style visuel"
+                        )
+                        
+                        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
+                        )
+                        
+                        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"
                     )
-
-                with gr.Group():
-                    gr.Markdown("### ✨ Paramètres Créatifs")
-                    quality = gr.Slider(
-                        minimum=70,
-                        maximum=100,
-                        value=85,
-                        label="Qualité Artistique",
-                        info="Influence la finesse des détails"
-                    )
-                    creativity = gr.Slider(
-                        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", 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="Statut",
-                    interactive=False
-                )
+                    
+            # Statut
+            status = gr.Textbox(
+                label="Status",
+                interactive=False,
+                value="Prêt à générer"
+            )
 
         # Événements
         generate_btn.click(
-            artisan.generate_artwork,
+            generate_image,
             inputs=[
                 format_choice,
                 orientation,
-                style,
-                collection,
                 prompt,
+                style,
                 quality,
-                creativity
+                creativity,
+                negative_prompt  # Ajout du negative_prompt aux inputs
             ],
             outputs=[image_output, status]
         )
+        
+        clear_btn.click(
+            lambda: (None, "Image effacée"),
+            outputs=[image_output, status]
+        )
 
     return app