Update app.py

app.py

@@ -1,6 +1,10 @@
 import gradio as gr
 import random
 import time
+from diffusers import StableDiffusionPipeline
+import torch
+from PIL import Image
+import io
 
 # Styles disponibles
 STYLES = {
@@ -36,13 +40,38 @@ PROMPT_SUGGESTIONS = {
 
 def generate_image(prompt, style, format_size, material):
     """
-    Fonction de génération d'image (simulation pour le moment)
+    Génération d'image avec Stable Diffusion
     """
-    # Simuler le temps de génération
-    time.sleep(2)
-    
-    # Pour le moment, retourner un message de confirmation
-    return f"Image générée avec les paramètres :\nPrompt : {prompt}\nStyle : {style}\nFormat : {format_size}\nMatériau : {material}"
+    try:
+        # Construction du prompt enrichi
+        style_prompts = {
+            "signage": "professional signage, eco-friendly cardboard material, minimalist design",
+            "poster": "artistic poster, recycled cardboard texture, creative design",
+            "silhouette": "decorative silhouette, cardboard cutout, elegant design"
+        }
+        
+        # Enrichir le prompt avec le style et le matériau
+        enhanced_prompt = f"{prompt}, {style_prompts[style]}, {material} cardboard material, sustainable design, high quality, detailed"
+        
+        # Prompts négatifs pour améliorer la qualité
+        negative_prompt = "low quality, blurry, bad anatomy, ugly, distorted, pixelated"
+        
+        # Initialiser le pipeline
+        model_id = "runwayml/stable-diffusion-v1-5"
+        pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float32)
+        
+        # Générer l'image
+        image = pipe(
+            prompt=enhanced_prompt,
+            negative_prompt=negative_prompt,
+            num_inference_steps=30,
+            guidance_scale=7.5
+        ).images[0]
+        
+        return image
+        
+    except Exception as e:
+        return f"Erreur lors de la génération : {str(e)}"
 
 def update_suggestions(style):
     """
@@ -112,7 +141,7 @@ with gr.Blocks(css="footer {display: none}") as demo:
 
         # Panneau de droite - Prévisualisation
         with gr.Column(scale=2):
-            output = gr.Textbox(label="Résultat")
+            output = gr.Image(label="Image générée")
 
     # Interactions
     style.change(