Update app.py

app.py

@@ -5,9 +5,89 @@ import requests
 import io
 import json
 
-# [Insérer ici le code des ART_STYLES, TEXT_EFFECTS, et THEME_COLLECTIONS précédents]
+# Définition des styles
+ART_STYLES = {
+    "Renaissance": {
+        "prompt_prefix": "renaissance style masterpiece, classical composition, chiaroscuro lighting",
+        "text_style": "elegant classical typography, golden ratio composition",
+        "guidance": 15.0,
+        "steps": 80,
+        "negative_prompt": "modern, digital, low quality, blurry"
+    },
+    "Impressionnisme": {
+        "prompt_prefix": "impressionist painting style, vibrant brushstrokes, atmospheric lighting",
+        "text_style": "artistic text integration, painterly typography",
+        "guidance": 12.0,
+        "steps": 65,
+        "negative_prompt": "sharp details, digital art, photorealistic"
+    },
+    "Art Moderne": {
+        "prompt_prefix": "modern art style, geometric shapes, bold composition",
+        "text_style": "modern typography, geometric letter design",
+        "guidance": 11.0,
+        "steps": 60,
+        "negative_prompt": "traditional, realistic, detailed texture"
+    },
+    "Neo Vintage": {
+        "prompt_prefix": "vintage style poster, retro advertisement design, classic illustration",
+        "text_style": "vintage typography, retro lettering style",
+        "guidance": 12.5,
+        "steps": 65,
+        "negative_prompt": "modern, digital, contemporary"
+    },
+    "Pop Art": {
+        "prompt_prefix": "pop art style, comic book aesthetic, bold colors, halftone dots",
+        "text_style": "bold comic book typography, retro lettering",
+        "guidance": 11.5,
+        "steps": 60,
+        "negative_prompt": "subtle, realistic, painterly"
+    }
+}
 
-# Interface utilisateur améliorée avec CSS personnalisé
+# Effets de texte
+TEXT_EFFECTS = {
+    "Standard": {
+        "prompt_suffix": "with clear readable text",
+        "text_params": {"weight": 1.0}
+    },
+    "Néon": {
+        "prompt_suffix": "with glowing neon text effect",
+        "text_params": {"weight": 1.2}
+    },
+    "3D": {
+        "prompt_suffix": "with 3D text effect, depth and shadows",
+        "text_params": {"weight": 1.3}
+    },
+    "Métallique": {
+        "prompt_suffix": "with metallic text effect, reflective surface",
+        "text_params": {"weight": 1.1}
+    },
+    "Graffiti": {
+        "prompt_suffix": "with graffiti style text, urban art typography",
+        "text_params": {"weight": 1.2}
+    }
+}
+
+# Collections thématiques
+THEME_COLLECTIONS = {
+    "Nature": {
+        "prompts": ["natural elements", "organic composition"],
+        "styles": ["flowing lines", "natural textures"],
+        "negative": "artificial, synthetic"
+    },
+    "Urbain": {
+        "prompts": ["urban landscape", "city elements"],
+        "styles": ["street art", "architectural elements"],
+        "negative": "rural, natural"
+    },
+    "Tech": {
+        "prompts": ["technological elements", "digital aesthetic"],
+        "styles": ["circuit patterns", "tech elements"],
+        "negative": "organic, traditional"
+    }
+}
+
+# CSS personnalisé
 CUSTOM_CSS = """
 .container { max-width: 1200px; margin: auto; }
 .welcome { text-align: center; margin: 20px 0; padding: 20px; background: #1e293b; border-radius: 10px; }
@@ -23,27 +103,19 @@ def enhance_prompt(subject, style, text_effect, collection=None, additional_deta
     style_config = ART_STYLES[style]
     text_config = TEXT_EFFECTS[text_effect]
     
-    # Construction du prompt de base
     base_prompt = f"{style_config['prompt_prefix']}, {subject}"
     
-    # Ajout des effets de texte
     if text_effect != "Standard":
         base_prompt += f", {text_config['prompt_suffix']}"
     
-    # Ajout des éléments de collection
     if collection and collection in THEME_COLLECTIONS:
         collection_data = THEME_COLLECTIONS[collection]
-        collection_elements = (
-            collection_data["prompts"] + 
-            collection_data["styles"]
-        )
+        collection_elements = collection_data["prompts"] + collection_data["styles"]
         base_prompt += f", {', '.join(collection_elements)}"
         
-    # Ajout des détails supplémentaires
     if additional_details:
         base_prompt += f", {additional_details}"
     
-    # Construction du prompt négatif
     negative_prompt = style_config['negative_prompt']
     if collection:
         negative_prompt += f", {THEME_COLLECTIONS[collection]['negative']}"
@@ -52,36 +124,33 @@ def enhance_prompt(subject, style, text_effect, collection=None, additional_deta
 
 def generate_image(format_size, orientation, subject, style, text_effect, collection, 
                   title, subtitle, quality, creativity, additional_details=""):
-    """Fonction de génération d'image améliorée"""
+    """Fonction de génération d'image"""
     API_URL = "https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-xl-base-1.0"
     headers = {"Authorization": f"Bearer {os.getenv('HUGGINGFACE_TOKEN')}"}
     
     try:
-        # Optimisation des dimensions
         base_width = 1024
         base_height = 1024
-        if format_size in ["A4", "A3"]:
-            if orientation == "Portrait":
+        if orientation == "Portrait":
+            if format_size in ["A4", "A3"]:
                 base_width = 768
                 base_height = 1024
-            else:
+        else:
+            if format_size in ["A4", "A3"]:
                 base_width = 1024
                 base_height = 768
-                
-        # Génération du prompt optimisé
+
         enhanced_prompt, negative_prompt = enhance_prompt(
             subject, style, text_effect, collection, additional_details
         )
         
-        # Ajout des textes si présents
-        if title or subtitle:
-            enhanced_prompt += f", with text elements: '{title}'"
-            if subtitle:
-                enhanced_prompt += f" and subtitle: '{subtitle}'"
-                
-        print(f"Prompt final: {enhanced_prompt}")  # Debug
+        if title:
+            enhanced_prompt += f", with text: '{title}'"
+        if subtitle:
+            enhanced_prompt += f", subtitle: '{subtitle}'"
+            
+        print(f"Prompt: {enhanced_prompt}")  # Debug
         
-        # Configuration de la requête
         payload = {
             "inputs": enhanced_prompt,
             "parameters": {
@@ -89,8 +158,7 @@ def generate_image(format_size, orientation, subject, style, text_effect, collec
                 "num_inference_steps": int(ART_STYLES[style]['steps'] * (quality/100)),
                 "guidance_scale": ART_STYLES[style]['guidance'] * (creativity/10),
                 "width": base_width,
-                "height": base_height,
-                "seed": int(creativity * 1000)  # Pour la reproductibilité
+                "height": base_height
             }
         }
         
@@ -100,16 +168,15 @@ def generate_image(format_size, orientation, subject, style, text_effect, collec
             image = Image.open(io.BytesIO(response.content))
             return image, f"✨ Création {style} avec effet {text_effect} réussie!"
         else:
-            print(f"Erreur API: {response.text}")  # Debug
+            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)}")  # Debug
+        print(f"Exception: {str(e)}")
         return None, f"⚠️ Erreur: {str(e)}"
 
 def create_interface():
     with gr.Blocks(css=CUSTOM_CSS) as app:
-        # En-tête
         gr.HTML("""
             <div class="welcome">
                 <h1>🤖 Equity Artisan 3.0</h1>
@@ -118,7 +185,7 @@ def create_interface():
         """)
         
         with gr.Column(elem_classes="container"):
-            # Étape 1: Format
+            # Format
             with gr.Group(elem_classes="style-group"):
                 gr.Markdown("### 📏 Format et Orientation")
                 with gr.Row():
@@ -133,29 +200,26 @@ def create_interface():
                         label="Orientation"
                     )
 
-            # Étape 2: Style et Contenu
+            # Style et Contenu
             with gr.Group(elem_classes="style-group"):
-                gr.Markdown("### 🎨 Style et Contenu")
+                gr.Markdown("### 🎨 Style et Création")
                 with gr.Row():
                     with gr.Column(scale=1):
                         style = gr.Dropdown(
                             choices=list(ART_STYLES.keys()),
                             value="Neo Vintage",
-                            label="Style artistique",
-                            info="Choisissez le style visuel"
+                            label="Style artistique"
                         )
                         text_effect = gr.Dropdown(
                             choices=list(TEXT_EFFECTS.keys()),
                             value="Standard",
-                            label="Effet de texte",
-                            info="Style du texte"
+                            label="Effet de texte"
                         )
                         collection = gr.Dropdown(
                             choices=list(THEME_COLLECTIONS.keys()),
-                            label="Collection thématique",
-                            info="Thème global"
+                            label="Collection (optionnel)"
                         )
-                    
+                
                     with gr.Column(scale=2):
                         subject = gr.Textbox(
                             label="Sujet principal",
@@ -163,61 +227,49 @@ def create_interface():
                         )
                         additional_details = gr.Textbox(
                             lines=2,
-                            label="Détails additionnels (optionnel)",
-                            placeholder="Ajoutez des détails spécifiques..."
+                            label="Détails additionnels",
+                            placeholder="Ajoutez des détails..."
                         )
 
-            # Étape 3: Texte
+            # Texte
             with gr.Group(elem_classes="text-effects"):
-                gr.Markdown("### ✍️ Texte de l'affiche")
+                gr.Markdown("### ✍️ Texte")
                 with gr.Row():
                     title = gr.Textbox(
                         label="Titre principal",
-                        placeholder="Texte principal de l'affiche"
+                        placeholder="Texte principal..."
                     )
                     subtitle = gr.Textbox(
-                        label="Sous-titre (optionnel)",
-                        placeholder="Texte secondaire"
+                        label="Sous-titre",
+                        placeholder="Texte secondaire..."
                     )
 
-            # Étape 4: Paramètres de génération
-            with gr.Group(elem_classes="style-group"):
-                gr.Markdown("### ⚙️ Paramètres")
+            # Paramètres
+            with gr.Group(elem_classes="quality-controls"):
                 with gr.Row():
                     quality = gr.Slider(
                         minimum=30,
                         maximum=50,
                         value=35,
                         step=5,
-                        label="Qualité",
-                        info="Niveau de détail"
+                        label="Qualité"
                     )
                     creativity = gr.Slider(
                         minimum=5,
                         maximum=15,
                         value=7.5,
                         step=0.5,
-                        label="Créativité",
-                        info="Balance entre fidélité et originalité"
+                        label="Créativité"
                     )
 
-            # Boutons de génération
             with gr.Row():
                 generate_btn = gr.Button("✨ Générer", variant="primary")
                 clear_btn = gr.Button("🗑️ Effacer", variant="secondary")
 
-            # Zone de prévisualisation
-            with gr.Group(elem_classes="preview-panel"):
-                image_output = gr.Image(
-                    label="Aperçu",
-                    type="pil"
-                )
-                status = gr.Textbox(
-                    label="Statut",
-                    interactive=False
-                )
-
-        # Événements
+            image_output = gr.Image(label="Aperçu", type="pil")
+            status = gr.Textbox(label="Statut", interactive=False)
+
+        # Events
         generate_btn.click(
             generate_image,
             inputs=[