Update app.py

app.py

@@ -112,9 +112,7 @@ COMPOSITION_PARAMS = {
         "Terre": "earthy color palette, natural tones",
         "Néon": "neon color palette, vibrant glowing colors"
     }
-}
-
-class ImageGenerator:
+}class ImageGenerator:
     def __init__(self):
         self.API_URL = "https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-xl-base-1.0"
         token = os.getenv('HUGGINGFACE_TOKEN')
@@ -124,65 +122,40 @@ class ImageGenerator:
         logger.info("ImageGenerator initialisé")
 
     def _build_prompt(self, params: Dict[str, Any]) -> str:
-    """Construction de prompt améliorée avec le PromptEnhancer"""
-    try:
-        # Initialisation du PromptEnhancer
-        enhancer = PromptEnhancer()
-        # Préparation du prompt initial
-        base_prompt = f"{params['subject']}"
-        if params.get('title'):
-            base_prompt += f", with text '{params['title']}'"
-        
-        # Amélioration du prompt
-        enhanced_prompt = enhancer.enhance_prompt(base_prompt, style_context)
-        
-        # Analyse de l'efficacité du prompt
-        prompt_analysis = enhancer.analyze_prompt_effectiveness(enhanced_prompt)
-        logger.debug(f"Analyse du prompt: {json.dumps(prompt_analysis, indent=2)}")
-        
-        return enhanced_prompt
-        except Exception as e:
-        logger.error(f"Erreur dans la construction du prompt: {str(e)}")
-        # Fallback sur le prompt basique en cas d'erreur
-        return f"{style_info['prompt_prefix']}, {params['subject']}"
-        
-        # Récupération du style
-        style_info = ART_STYLES.get(params["style"], ART_STYLES["Art Moderne"])
-        
-        # Construction du contexte de style
-        style_context = {
-            "prompt_prefix": style_info['prompt_prefix'],
-            "layout": COMPOSITION_PARAMS['Layouts'][params['layout']],
-            "ambiance": COMPOSITION_PARAMS['Ambiances'][params['ambiance']],
-            "palette": COMPOSITION_PARAMS['Palette'][params['palette']]
-        }
-
-        # 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']]}"
-
-        # Ajout des ajustements fins
-        if params.get("detail_level"):
-            detail_strength = params["detail_level"]
-            prompt += f", {'highly detailed, intricate' if detail_strength > 7 else 'moderately detailed'}"
-
-        if params.get("contrast"):
-            contrast_strength = params["contrast"]
-            prompt += f", {'high contrast, dramatic lighting' if contrast_strength > 7 else 'balanced contrast'}"
-
-        if params.get("saturation"):
-            saturation_strength = params["saturation"]
-            prompt += f", {'vibrant saturated colors' if saturation_strength > 7 else 'subtle muted colors'}"
-
-        if params.get("title"):
-            prompt += f", with text saying '{params['title']}'"
+        """Construction de prompt améliorée avec le PromptEnhancer"""
+        try:
+            # Initialisation du PromptEnhancer
+            enhancer = PromptEnhancer()
+            
+            # Récupération du style
+            style_info = ART_STYLES.get(params["style"], ART_STYLES["Art Moderne"])
+            
+            # Construction du contexte de style
+            style_context = {
+                "prompt_prefix": style_info['prompt_prefix'],
+                "layout": COMPOSITION_PARAMS['Layouts'][params['layout']],
+                "ambiance": COMPOSITION_PARAMS['Ambiances'][params['ambiance']],
+                "palette": COMPOSITION_PARAMS['Palette'][params['palette']]
+            }
+            
+            # Préparation du prompt initial
+            base_prompt = f"{params['subject']}"
+            if params.get('title'):
+                base_prompt += f", with text '{params['title']}'"
+            
+            # Amélioration du prompt
+            enhanced_prompt = enhancer.enhance_prompt(base_prompt, style_context)
+            
+            # Analyse de l'efficacité du prompt
+            prompt_analysis = enhancer.analyze_prompt_effectiveness(enhanced_prompt)
+            logger.debug(f"Analyse du prompt: {json.dumps(prompt_analysis, indent=2)}")
+            
+            return enhanced_prompt
 
-        logger.debug(f"Prompt final: {prompt}")
-        return prompt
+        except Exception as e:
+            logger.error(f"Erreur dans la construction du prompt: {str(e)}")
+            # Fallback sur le prompt basique en cas d'erreur
+            return f"{style_info['prompt_prefix']}, {params['subject']}"
 
     def generate(self, params: Dict[str, Any]) -> Tuple[Optional[Image.Image], str]:
         try: