Update app.py

app.py

@@ -4,147 +4,140 @@ 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"
+# 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
+ART_STYLES = {
     "Illustration Pro": {
-        "prompt_prefix": "professional editorial illustration, sophisticated art style, high-end magazine quality",
+        "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, basic, photographic"
+        "negative_prompt": "amateur, sketchy, rough, unfinished, low quality"
     },
-    "Néo Vintage": {
-        "prompt_prefix": "vintage poster design, retro aesthetic, limited color palette, screen printing style",
+    "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",
         "guidance": 12.5,
         "steps": 65,
-        "negative_prompt": "modern, digital, photorealistic"
+        "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"
     },
-    "Encre Artistique": {
-        "prompt_prefix": "artistic ink illustration, dynamic brushwork, high contrast black and white, expressive strokes",
+    "Dark Fantasy": {
+        "prompt_prefix": "dark fantasy art style, atmospheric mood, dramatic fantasy illustration",
+        "style_modifiers": "mystical atmosphere, dramatic lighting, ethereal quality",
         "guidance": 14.0,
         "steps": 75,
-        "negative_prompt": "color, photographic, digital"
+        "negative_prompt": "cheerful, bright, realistic photo, mundane"
     },
     "Pop Art": {
-        "prompt_prefix": "pop art style, bold colors, halftone dots, comic book aesthetic, Warhol-inspired",
+        "prompt_prefix": "pop art style illustration, bold graphic design, comic book aesthetic",
+        "style_modifiers": "vibrant colors, halftone dots, bold outlines, 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"
+        "negative_prompt": "realistic, subtle, painterly, traditional"
     }
 }
 
-# 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; }
 .quality-controls { margin: 10px 0; padding: 10px; background: #2d3748; border-radius: 5px; }
+.style-info { padding: 10px; background: #2d3748; border-radius: 5px; margin-top: 10px; }
 """
 
-def create_interface():
-    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')}"}
+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
+    
+    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']}"
+    
+    return final_prompt, style_config['negative_prompt']
+
+def generate_image(format_size, orientation, subject, style, 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
         
-        try:
-            # 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
+        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']
+        # Construction du prompt optimisé
+        enhanced_prompt, negative_prompt = enhance_prompt(subject, style, additional_details)
+        
+        print(f"Prompt final: {enhanced_prompt}")  # Debug
+        print(f"Prompt négatif: {negative_prompt}")  # Debug
 
-            # Configuration de la génération
-            payload = {
-                "inputs": enhanced_prompt,
-                "parameters": {
-                    "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
-                }
+        # Configuration de la requête
+        payload = {
+            "inputs": enhanced_prompt,
+            "parameters": {
+                "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
             }
+        }
+        
+        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!"
+        else:
+            print(f"Erreur API: {response.text}")  # Debug
+            return None, f"⚠️ Erreur {response.status_code}: Ajustez les paramètres"
             
-            print(f"Génération avec paramètres: {payload}")
-            
-            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!"
-            else:
-                print(f"Erreur API: {response.text}")
-                return None, f"⚠️ Erreur {response.status_code}: Ajustez les paramètres"
-            
-        except Exception as e:
-            print(f"Exception: {str(e)}")
-            return None, f"⚠️ Erreur: {str(e)}"
+    except Exception as e:
+        print(f"Exception: {str(e)}")  # Debug
+        return None, f"⚠️ Erreur: {str(e)}"
 
+def create_interface():
     with gr.Blocks(css=CUSTOM_CSS) as app:
         gr.HTML("""
             <div class="welcome">
@@ -177,20 +170,23 @@ def create_interface():
                     with gr.Column():
                         style = gr.Dropdown(
                             choices=list(ART_STYLES.keys()),
-                            value="Renaissance",
-                            label="Style artistique"
+                            value="Illustration Pro",
+                            label="Style artistique",
+                            info="Choisissez votre style d'illustration"
                         )
                         
-                        collection = gr.Dropdown(
-                            choices=list(COLLECTIONS.keys()),
-                            label="Collection (optionnel)",
-                            info="Thèmes artistiques spécialisés"
+                        subject = gr.Textbox(
+                            lines=1,
+                            label="Sujet principal",
+                            placeholder="Ex: libellule, loup, paysage...",
+                            info="Décrivez le sujet principal"
                         )
                         
-                        prompt = gr.Textbox(
-                            lines=3,
-                            label="Description",
-                            placeholder="Décrivez votre vision artistique..."
+                        additional_details = gr.Textbox(
+                            lines=2,
+                            label="Détails additionnels (optionnel)",
+                            placeholder="Ajoutez des détails spécifiques pour enrichir l'image...",
+                            info="Plus de détails = meilleur résultat"
                         )
                         
                         with gr.Group(elem_classes="quality-controls"):
@@ -233,11 +229,11 @@ def create_interface():
             inputs=[
                 format_choice,
                 orientation,
-                prompt,
+                subject,
                 style,
                 quality,
                 creativity,
-                collection
+                additional_details
             ],
             outputs=[image_output, status]
         )