Update app.py

app.py

@@ -10,10 +10,7 @@ import logging
 from dotenv import load_dotenv
 
 # Configuration du logging
-logging.basicConfig(
-    level=logging.DEBUG,
-    format='%(asctime)s - %(levelname)s - %(message)s'
-)
+logging.basicConfig(level=logging.DEBUG, format='%(asctime)s - %(levelname)s - %(message)s')
 logger = logging.getLogger(__name__)
 
 # Chargement des variables d'environnement
@@ -52,6 +49,10 @@ ART_STYLES = {
     "Japonais": {
         "prompt_prefix": "japanese art style poster, ukiyo-e inspired design",
         "negative_prompt": "western, modern, photographic"
+    },
+    "Ultra Réaliste": {
+        "prompt_prefix": "hyper-realistic poster, photographic quality, extremely detailed",
+        "negative_prompt": "cartoon, illustration, stylized, abstract"
     }
 }
 
@@ -90,28 +91,20 @@ class ImageGenerator:
         logger.info("ImageGenerator initialisé")
 
     def _build_prompt(self, params: Dict[str, Any]) -> str:
-        """Construction de prompt améliorée"""
         style_info = ART_STYLES.get(params["style"], ART_STYLES["Neo Vintage"])
         prompt = f"{style_info['prompt_prefix']}, {params['subject']}"
 
-        # Ajout des paramètres de composition
-        if params.get("layout"):
-            prompt += f", {COMPOSITION_PARAMS['Layouts'][params['layout']]}"
-        if params.get("ambiance"):
-            prompt += f", {COMPOSITION_PARAMS['Ambiances'][params['ambiance']]}"
-        if params.get("palette"):
-            prompt += f", {COMPOSITION_PARAMS['Palette'][params['palette']]}"
+        for param_type in ['layout', 'ambiance', 'palette']:
+            if params.get(param_type):
+                prompt += f", {COMPOSITION_PARAMS[param_type.capitalize()][params[param_type]]}"
 
-        # Ajout des ajustements fins
-        if params.get("detail_level"):
-            detail_strength = params["detail_level"]
-            prompt += f", {'highly detailed' if detail_strength > 7 else 'moderately detailed'}"
-        if params.get("contrast"):
-            contrast_strength = params["contrast"]
-            prompt += f", {'high contrast' if contrast_strength > 7 else 'balanced contrast'}"
-        if params.get("saturation"):
-            saturation_strength = params["saturation"]
-            prompt += f", {'vibrant colors' if saturation_strength > 7 else 'subtle colors'}"
+        for param, description in [
+            ("detail_level", "highly detailed" if params.get("detail_level", 0) > 7 else "moderately detailed"),
+            ("contrast", "high contrast" if params.get("contrast", 0) > 7 else "balanced contrast"),
+            ("saturation", "vibrant colors" if params.get("saturation", 0) > 7 else "subtle colors")
+        ]:
+            if params.get(param):
+                prompt += f", {description}"
 
         if params.get("title"):
             prompt += f", with text saying '{params['title']}'"
@@ -126,8 +119,6 @@ class ImageGenerator:
                 return None, "⚠️ Erreur: Token Hugging Face non configuré"
 
             prompt = self._build_prompt(params)
-
-            # Configuration de base
             payload = {
                 "inputs": prompt,
                 "parameters": {
@@ -140,12 +131,7 @@ class ImageGenerator:
             }
 
             logger.debug(f"Payload: {json.dumps(payload, indent=2)}")
-            response = requests.post(
-                self.API_URL,
-                headers=self.headers,
-                json=payload,
-                timeout=30
-            )
+            response = requests.post(self.API_URL, headers=self.headers, json=payload, timeout=30)
 
             if response.status_code == 200:
                 image = Image.open(io.BytesIO(response.content))
@@ -166,9 +152,12 @@ def create_interface():
     logger.info("Création de l'interface Gradio")
     css = """
     .container { max-width: 1200px; margin: auto; }
-    .welcome { text-align: center; margin: 20px 0; padding: 20px; background: #1e293b; border-radius: 10px; }
-    .controls-group { background: #2d3748; padding: 15px; border-radius: 5px; margin: 10px 0; }
-    .advanced-controls { background: #374151; padding: 12px; border-radius: 5px; margin: 8px 0; }
+    .welcome { text-align: center; margin: 20px 0; padding: 20px; background: #3498db; border-radius: 10px; color: white; }
+    .controls-group { background: #ecf0f1; padding: 15px; border-radius: 5px; margin: 10px 0; color: #2c3e50; }
+    .advanced-controls { background: #bdc3c7; padding: 12px; border-radius: 5px; margin: 8px 0; }
+    .gradio-slider input[type="range"] { accent-color: #3498db; }
+    .gradio-button { transition: all 0.3s ease; }
+    .gradio-button:hover { transform: translateY(-2px); box-shadow: 0 4px 6px rgba(52, 152, 219, 0.11), 0 1px 3px rgba(0, 0, 0, 0.08); }
     """
 
     generator = ImageGenerator()
@@ -176,118 +165,58 @@ def create_interface():
     with gr.Blocks(css=css) as app:
         gr.HTML("""
         <div class="welcome">
-            <h1>🎨 Equity Artisan 3.0</h1>
-            <p>Assistant de création d'affiches professionnelles</p>
+            <h1>🎨 Equity Artisan 4.0</h1>
+            <p>Assistant de création d'affiches professionnelles avancé</p>
         </div>
         """)
 
         with gr.Column(elem_classes="container"):
-            # Format et Orientation
             with gr.Group(elem_classes="controls-group"):
                 gr.Markdown("### 📐 Format et Orientation")
                 with gr.Row():
-                    format_size = gr.Dropdown(
-                        choices=["A4", "A3", "A2", "A1", "A0"],
-                        value="A4",
-                        label="Format"
-                    )
-                    orientation = gr.Radio(
-                        choices=["Portrait", "Paysage"],
-                        value="Portrait",
-                        label="Orientation"
-                    )
+                    format_size = gr.Dropdown(choices=["A4", "A3", "A2", "A1", "A0"], value="A4", label="Format")
+                    orientation = gr.Radio(choices=["Portrait", "Paysage"], value="Portrait", label="Orientation")
 
-            # Style et Composition
             with gr.Group(elem_classes="controls-group"):
                 gr.Markdown("### 🎨 Style et Composition")
                 with gr.Row():
-                    style = gr.Dropdown(
-                        choices=list(ART_STYLES.keys()),
-                        value="Neo Vintage",
-                        label="Style artistique"
-                    )
-                    layout = gr.Dropdown(
-                        choices=list(COMPOSITION_PARAMS["Layouts"].keys()),
-                        value="Centré",
-                        label="Composition"
-                    )
+                    style = gr.Dropdown(choices=list(ART_STYLES.keys()), value="Neo Vintage", label="Style artistique")
+                    layout = gr.Dropdown(choices=list(COMPOSITION_PARAMS["Layouts"].keys()), value="Centré", label="Composition")
                 with gr.Row():
-                    ambiance = gr.Dropdown(
-                        choices=list(COMPOSITION_PARAMS["Ambiances"].keys()),
-                        value="Dramatique",
-                        label="Ambiance"
-                    )
-                    palette = gr.Dropdown(
-                        choices=list(COMPOSITION_PARAMS["Palette"].keys()),
-                        value="Contrasté",
-                        label="Palette"
-                    )
+                    ambiance = gr.Dropdown(choices=list(COMPOSITION_PARAMS["Ambiances"].keys()), value="Dramatique", label="Ambiance")
+                    palette = gr.Dropdown(choices=list(COMPOSITION_PARAMS["Palette"].keys()), value="Contrasté", label="Palette")
 
-            # Contenu
             with gr.Group(elem_classes="controls-group"):
                 gr.Markdown("### 📝 Contenu")
-                subject = gr.Textbox(
-                    label="Description",
-                    placeholder="Décrivez votre vision..."
-                )
-                title = gr.Textbox(
-                    label="Titre",
-                    placeholder="Titre de l'affiche..."
-                )
+                subject = gr.Textbox(label="Description", placeholder="Décrivez votre vision...")
+                title = gr.Textbox(label="Titre", placeholder="Titre de l'affiche...")
 
-            # Ajustements fins
             with gr.Group(elem_classes="advanced-controls"):
                 gr.Markdown("### 🎯 Ajustements Fins")
                 with gr.Row():
-                    detail_level = gr.Slider(
-                        minimum=1, maximum=10, value=7, step=1,
-                        label="Niveau de Détail"
-                    )
-                    contrast = gr.Slider(
-                        minimum=1, maximum=10, value=5, step=1,
-                        label="Contraste"
-                    )
-                    saturation = gr.Slider(
-                        minimum=1, maximum=10, value=5, step=1,
-                        label="Saturation"
-                    )
+                    detail_level = gr.Slider(minimum=1, maximum=10, value=7, step=1, label="Niveau de Détail")
+                    contrast = gr.Slider(minimum=1, maximum=10, value=5, step=1, label="Contraste")
+                    saturation = gr.Slider(minimum=1, maximum=10, value=5, step=1, label="Saturation")
 
-            # Paramètres de génération
             with gr.Group(elem_classes="controls-group"):
                 with gr.Row():
-                    quality = gr.Slider(
-                        minimum=30, maximum=50, value=35,
-                        label="Qualité"
-                    )
-                    creativity = gr.Slider(
-                        minimum=5, maximum=15, value=7.5,
-                        label="Créativité"
-                    )
+                    quality = gr.Slider(minimum=30, maximum=50, value=35, label="Qualité")
+                    creativity = gr.Slider(minimum=5, maximum=15, value=7.5, label="Créativité")
 
-            # Boutons
             with gr.Row():
                 generate_btn = gr.Button("�� Générer", variant="primary")
                 clear_btn = gr.Button("🗑️ Effacer", variant="secondary")
 
-            # Sortie
             image_output = gr.Image(label="Aperçu")
             status = gr.Textbox(label="Statut", interactive=False)
 
         def generate(*args):
             logger.info("Démarrage d'une nouvelle génération")
             params = {
-                "format_size": args[0],
-                "orientation": args[1],
-                "style": args[2],
-                "layout": args[3],
-                "ambiance": args[4],
-                "palette": args[5],
-                "subject": args[6],
-                "title": args[7],
-                "detail_level": args[8],
-                "contrast": args[9],
-                "saturation": args[10],
-                "quality": args[11],
+                "format_size": args[0], "orientation": args[1], "style": args[2],
+                "layout": args[3], "ambiance": args[4], "palette": args[5],
+                "subject": args[6], "title": args[7], "detail_level": args[8],
+                "contrast": args[9], "saturation": args[10], "quality": args[11],
                 "creativity": args[12]
             }
             result = generator.generate(params)
@@ -296,17 +225,11 @@ def create_interface():
 
         generate_btn.click(
             generate,
-            inputs=[
-                format_size, orientation, style, layout, ambiance, palette,
-                subject, title, detail_level, contrast, saturation, quality, creativity
-            ],
+            inputs=[format_size, orientation, style, layout, ambiance, palette, subject, title, detail_level, contrast, saturation, quality, creativity],
             outputs=[image_output, status]
         )
 
-        clear_btn.click(
-            lambda: (None, "🗑️ Image effacée"),
-            outputs=[image_output, status]
-        )
+        clear_btn.click(lambda: (None, "🗑️ Image effacée"), outputs=[image_output, status])
 
     logger.info("Interface créée avec succès")
     return app