Update app.py

app.py

@@ -44,35 +44,47 @@ def generate_image(prompt, style, format_size, material):
     """
     Génération d'image via l'API Hugging Face
     """
-    API_URL = "https://api-inference.huggingface.co/models/runwayml/stable-diffusion-v1-5"
+    API_URL = "https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-xl-base-1.0"
     headers = {"Authorization": f"Bearer {HF_TOKEN}"}
     
     # Style prompts pour améliorer la génération
     style_prompts = {
-        "signage": "professional signage design, minimalist, clean",
-        "poster": "artistic poster design, creative, expressive",
-        "silhouette": "decorative silhouette, elegant cutout design"
+        "signage": "professional signage design, minimalist, clean, photorealistic",
+        "poster": "artistic poster design, creative, expressive, highly detailed",
+        "silhouette": "decorative silhouette, elegant cutout design, clear shape"
     }
     
     # Enrichir le prompt
-    enhanced_prompt = f"{prompt}, {style_prompts[style]}, made of {material} cardboard, eco-friendly design, high quality, professional, detailed"
+    enhanced_prompt = f"a {material} cardboard {style} design showing {prompt}, {style_prompts[style]}, professional photography, 8k uhd, detailed"
+    
+    print(f"Envoi du prompt: {enhanced_prompt}")  # Pour debug
     
     payload = {
         "inputs": enhanced_prompt,
-        "negative_prompt": "low quality, blurry, bad anatomy, distorted, pixelated",
+        "negative_prompt": "low quality, blurry, bad anatomy, distorted, disfigured, pixelated",
+        "num_inference_steps": 40,
+        "guidance_scale": 7.5,
+        "width": 768,
+        "height": 768
     }
 
     try:
+        print("Envoi de la requête à l'API...")  # Pour debug
         response = requests.post(API_URL, headers=headers, json=payload)
+        print(f"Status code: {response.status_code}")  # Pour debug
+        
         if response.status_code == 200:
-            image_bytes = response.content
-            return Image.open(BytesIO(image_bytes))
+            return Image.open(BytesIO(response.content))
         else:
-            print(f"Erreur API : {response.status_code}")
-            return Image.new('RGB', (512, 512), color='lightgray')
+            print(f"Erreur API - Response: {response.text}")  # Pour debug
+            error_image = Image.new('RGB', (512, 512), color='lightgray')
+            # Ajouter un message d'erreur sur l'image
+            return error_image
+            
     except Exception as e:
-        print(f"Erreur de génération : {e}")
-        return Image.new('RGB', (512, 512), color='lightgray')
+        print(f"Exception lors de la génération: {str(e)}")  # Pour debug
+        error_image = Image.new('RGB', (512, 512), color='lightgray')
+        return error_image
 
 def update_suggestions(style):
     """