Update app.py

app.py

@@ -113,52 +113,52 @@ class ImageGenerator:
             f"https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-xl-base-1.0"
             )
         
-        # Chargement du token Hugging Face depuis les variables d'environnement
-        token = os.getenv('HUGGINGFACE_TOKEN')
-        
-        if not token:
-            logger.error("HUGGINGFACE_TOKEN non trouvé!")
-        
-        self.headers = {"Authorization": f"Bearer {token}"}
-        
-        logger.info("ImageGenerator initialisé")
+       # Chargement du token Hugging Face depuis les variables d'environnement
+       token = os.getenv('HUGGINGFACE_TOKEN')
+       
+       if not token:
+           logger.error("HUGGINGFACE_TOKEN non trouvé!")
+       
+       self.headers = {"Authorization": f"Bearer {token}"}
+       
+       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']}"
+       """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']]}"
+       # 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' 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'}"
+       # 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'}"
 
-        if params.get("title"):
-            prompt += f", with text saying '{params['title']}'"
+       if params.get("title"):
+           prompt += f", with text saying '{params['title']}'"
 
-        logger.debug(f"Prompt final: {prompt}")
-        
-        return prompt
+       logger.debug(f"Prompt final: {prompt}")
+       
+       return prompt
 
     def generate(self, params: Dict[str, Any]) -> Tuple[Optional[Image.Image], str]:
         
@@ -172,13 +172,13 @@ class ImageGenerator:
              
              payload = {
                  # Construction de la requête pour l'API Hugging Face
-                 “inputs”: prompt,
-                 “parameters”: {
-                     “negative_prompt”: ART_STYLES[params["style"]]["negative_prompt"],
-                     “num_inference_steps”: min(int(35 * (params["quality"]/100)), 40),
-                     “guidance_scale”: min(7.5 * (params["creativity"]/10), 10.0),
-                     “width”: 768,
-                     “height”: 768 if params["orientation"] == “Portrait” else 512
+                 "inputs": prompt,
+                 "parameters": {
+                     "negative_prompt": ART_STYLES[params["style"]]["negative_prompt"],
+                     "num_inference_steps": min(int(35 * (params["quality"]/100)), 40),
+                     "guidance_scale": min(7.5 * (params["creativity"]/10), 10.0),
+                     "width": 768,
+                     "height": 768 if params["orientation"] == “Portrait” else 512
                  }
              }