Update app.py

app.py

@@ -1,9 +1,7 @@
 import gradio as gr
-import random
 from PIL import Image
 import requests
 from io import BytesIO
-import os
 
 # Token Hugging Face - Assurez-vous que ce token est correct et a les bonnes permissions
 HF_TOKEN = "hf_..."  # Remplacez par votre vrai token
@@ -42,59 +40,56 @@ PROMPT_SUGGESTIONS = {
 
 def generate_image(prompt, style, format_size, material):
     """
-    Génération d'image via l'API Hugging Face
+    Génération d'image via l'API Hugging Face avec gestion des erreurs.
     """
     API_URL = "https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-xl-base-1.0"
     headers = {"Authorization": f"Bearer {HF_TOKEN}"}
-    
-    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"
-    }
-    
-    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}")
-    
-    payload = {
-        "inputs": enhanced_prompt,
-        "negative_prompt": "low quality, blurry, bad anatomy, distorted, disfigured, pixelated",
-        "num_inference_steps": 40,
-        "guidance_scale": 7.5,
-        "width": 512,
-        "height": 512
-    }
 
     try:
-        print("Envoi de la requête à l'API...")
+        # Validation des inputs
+        if not prompt or len(prompt.strip()) < 5:
+            return None, "Le prompt est trop court. Veuillez décrire plus en détail."
+
+        # Préparation du prompt
+        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"
+        }
+
+        enhanced_prompt = f"a {material} cardboard {style} design showing {prompt}, {style_prompts.get(style, '')}, professional photography, 8k uhd, detailed"
+
+        # Payload de la requête
+        payload = {
+            "inputs": enhanced_prompt,
+            "negative_prompt": "low quality, blurry, bad anatomy, distorted, disfigured, pixelated",
+            "num_inference_steps": 40,
+            "guidance_scale": 7.5,
+            "width": 512,
+            "height": 512
+        }
+
+        # Envoi de la requête
         response = requests.post(API_URL, headers=headers, json=payload)
-        print(f"Status code: {response.status_code}")
-        
+
+        # Gestion des différents codes de statut
         if response.status_code == 200:
-            print("Image générée avec succès")
-            return Image.open(BytesIO(response.content))
+            return Image.open(BytesIO(response.content)), None  # Pas d'erreur
+        elif response.status_code == 401:
+            return None, "Erreur d'authentification. Vérifiez votre token."
+        elif response.status_code == 429:
+            return None, "Quota API dépassé. Réessayez plus tard."
         else:
-            print(f"Erreur API - Response: {response.text}")
-            error_message = f"Erreur {response.status_code}: {response.text}"
-            error_image = Image.new('RGB', (512, 512), color='red')
-            return gr.Image.update(value=error_image, label=error_message)
-            
-    except requests.exceptions.RequestException as e:
-        print(f"Erreur de requête : {str(e)}")
-        error_message = f"Erreur de requête : {str(e)}"
-        error_image = Image.new('RGB', (512, 512), color='orange')
-        return gr.Image.update(value=error_image, label=error_message)
+            return None, f"Erreur inattendue. Code {response.status_code} - {response.text}"
+
+    except requests.exceptions.RequestException as req_err:
+        return None, f"Erreur de connexion : {req_err}"
+
     except Exception as e:
-        print(f"Exception inattendue : {str(e)}")
-        error_message = f"Erreur inattendue : {str(e)}"
-        error_image = Image.new('RGB', (512, 512), color='yellow')
-        return gr.Image.update(value=error_image, label=error_message)
+        return None, f"Erreur système : {e}"
 
 def update_suggestions(style):
-    """
-    Mise à jour des suggestions de prompts selon le style
-    """
+    """Mise à jour des suggestions de prompts selon le style."""
     return gr.Dropdown(choices=PROMPT_SUGGESTIONS.get(style, []))
 
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
@@ -102,73 +97,12 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
         """
         # 🎨 Equity Creation Studio
         ### Studio de Design Éco-responsable
-        Transformez vos idées en créations durables avec notre générateur d'images IA
+        Transformez vos idées en créations durables avec notre générateur d'images IA.
         """
     )
-    
+
     with gr.Row():
         with gr.Column(scale=1):
             style = gr.Radio(
                 choices=list(STYLES.keys()),
-                value="signage",
-                label="Style de Création",
-                info="Choisissez le type de design souhaité"
-            )
-            
-            format_size = gr.Radio(
-                choices=FORMATS,
-                value="A4",
-                label="Format",
-                info="Sélectionnez la taille de votre création"
-            )
-            
-            material = gr.Radio(
-                choices=MATERIALS,
-                value="Lisse",
-                label="Type de Carton",
-                info="Choisissez votre matériau"
-            )
-            
-            prompt = gr.Textbox(
-                lines=3,
-                label="Description",
-                placeholder="Décrivez votre vision..."
-            )
-            
-            suggestions = gr.Dropdown(
-                choices=PROMPT_SUGGESTIONS["signage"],
-                label="Suggestions de prompts",
-                info="Sélectionnez une suggestion ou écrivez votre propre description"
-            )
-            
-            generate_btn = gr.Button("🪄 Générer", variant="primary")
-
-        with gr.Column(scale=2):
-            output = gr.Image(label="Image générée")
-
-    style.change(
-        fn=update_suggestions,
-        inputs=[style],
-        outputs=[suggestions]
-    )
-    
-    suggestions.change(
-        fn=lambda x: x,
-        inputs=[suggestions],
-        outputs=[prompt]
-    )
-    
-    generate_btn.click(
-        fn=generate_image,
-        inputs=[prompt, style, format_size, material],
-        outputs=[output]
-    )
-
-    gr.Markdown(
-        """
-        ### 🌿 Design Éco-responsable
-        Equity Creation Studio transforme vos déchets carton en œuvres d'art
-        """
-    )
-
-demo.launch()
+                value="signage",