Update app.py

app.py

@@ -5,8 +5,8 @@ import requests
 from io import BytesIO
 import os
 
-# Token Hugging Face - À remplacer par votre token
-HF_TOKEN = "hf_..." # Mettez votre token ici
+# Token Hugging Face
+HF_TOKEN = "hf_..." # Assurez-vous que ce token est correct
 
 # Styles disponibles
 STYLES = {
@@ -47,17 +47,15 @@ def generate_image(prompt, style, format_size, material):
     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, photorealistic",
         "poster": "artistic poster design, creative, expressive, highly detailed",
         "silhouette": "decorative silhouette, elegant cutout design, clear shape"
     }
     
-    # Enrichir le prompt
     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
+    print(f"Envoi du prompt: {enhanced_prompt}")
     
     payload = {
         "inputs": enhanced_prompt,
@@ -69,21 +67,25 @@ def generate_image(prompt, style, format_size, material):
     }
 
     try:
-        print("Envoi de la requête à l'API...")  # Pour debug
+        print("Envoi de la requête à l'API...")
         response = requests.post(API_URL, headers=headers, json=payload)
-        print(f"Status code: {response.status_code}")  # Pour debug
+        print(f"Status code: {response.status_code}")
         
         if response.status_code == 200:
+            print("Image générée avec succès")
             return Image.open(BytesIO(response.content))
         else:
-            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
+            print(f"Erreur API - Response: {response.text}")
+            error_image = Image.new('RGB', (512, 512), color='red')
             return error_image
             
+    except requests.exceptions.RequestException as e:
+        print(f"Erreur de requête : {str(e)}")
+        error_image = Image.new('RGB', (512, 512), color='orange')
+        return error_image
     except Exception as e:
-        print(f"Exception lors de la génération: {str(e)}")  # Pour debug
-        error_image = Image.new('RGB', (512, 512), color='lightgray')
+        print(f"Exception inattendue : {str(e)}")
+        error_image = Image.new('RGB', (512, 512), color='yellow')
         return error_image
 
 def update_suggestions(style):