Update app.py

app.py

@@ -1,9 +1,12 @@
 import gradio as gr
 import random
-import time
 from PIL import Image
 import requests
 from io import BytesIO
+import os
+
+# Token Hugging Face - À remplacer par votre token
+HF_TOKEN = "hf_..." # Mettez votre token ici
 
 # Styles disponibles
 STYLES = {
@@ -37,16 +40,39 @@ PROMPT_SUGGESTIONS = {
     ]
 }
 
-def generate_test_image(prompt, style, format_size, material):
+def generate_image(prompt, style, format_size, material):
     """
-    Fonction de simulation de génération d'image
+    Génération d'image via l'API Hugging Face
     """
-    # Simuler un délai de génération
-    time.sleep(2)
+    API_URL = "https://api-inference.huggingface.co/models/runwayml/stable-diffusion-v1-5"
+    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"
+    }
     
-    # Pour le moment, retourner une image de test
-    test_image = Image.new('RGB', (512, 512), color = 'white')
-    return test_image
+    # Enrichir le prompt
+    enhanced_prompt = f"{prompt}, {style_prompts[style]}, made of {material} cardboard, eco-friendly design, high quality, professional, detailed"
+    
+    payload = {
+        "inputs": enhanced_prompt,
+        "negative_prompt": "low quality, blurry, bad anatomy, distorted, pixelated",
+    }
+
+    try:
+        response = requests.post(API_URL, headers=headers, json=payload)
+        if response.status_code == 200:
+            image_bytes = response.content
+            return Image.open(BytesIO(image_bytes))
+        else:
+            print(f"Erreur API : {response.status_code}")
+            return Image.new('RGB', (512, 512), color='lightgray')
+    except Exception as e:
+        print(f"Erreur de génération : {e}")
+        return Image.new('RGB', (512, 512), color='lightgray')
 
 def update_suggestions(style):
     """
@@ -54,13 +80,6 @@ def update_suggestions(style):
     """
     return gr.Dropdown(choices=PROMPT_SUGGESTIONS.get(style, []))
 
-def get_example_prompt(style):
-    """
-    Obtenir un prompt exemple du style sélectionné
-    """
-    suggestions = PROMPT_SUGGESTIONS.get(style, [])
-    return random.choice(suggestions) if suggestions else ""
-
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
     gr.Markdown(
         """
@@ -71,9 +90,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
     )
     
     with gr.Row():
-        # Panneau de gauche - Contrôles
         with gr.Column(scale=1):
-            # Section Style
             style = gr.Radio(
                 choices=list(STYLES.keys()),
                 value="signage",
@@ -81,7 +98,6 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
                 info="Choisissez le type de design souhaité"
             )
             
-            # Section Format
             format_size = gr.Radio(
                 choices=FORMATS,
                 value="A4",
@@ -89,7 +105,6 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
                 info="Sélectionnez la taille de votre création"
             )
             
-            # Section Matériau
             material = gr.Radio(
                 choices=MATERIALS,
                 value="Lisse",
@@ -97,28 +112,23 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
                 info="Choisissez votre matériau"
             )
             
-            # Section Prompt
             prompt = gr.Textbox(
                 lines=3,
                 label="Description",
                 placeholder="Décrivez votre vision..."
             )
             
-            # Suggestions de prompts
             suggestions = gr.Dropdown(
                 choices=PROMPT_SUGGESTIONS["signage"],
                 label="Suggestions de prompts",
                 info="Sélectionnez une suggestion ou écrivez votre propre description"
             )
             
-            # Bouton de génération
             generate_btn = gr.Button("🪄 Générer", variant="primary")
 
-        # Panneau de droite - Prévisualisation
         with gr.Column(scale=2):
             output = gr.Image(label="Image générée")
 
-    # Interactions
     style.change(
         fn=update_suggestions,
         inputs=[style],
@@ -132,7 +142,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
     )
     
     generate_btn.click(
-        fn=generate_test_image,
+        fn=generate_image,
         inputs=[prompt, style, format_size, material],
         outputs=[output]
     )
@@ -144,5 +154,4 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
         """
     )
 
-# Lancement de l'interface
 demo.launch()