Update app.py

app.py

@@ -17,13 +17,29 @@ model_id = "black-forest-labs/FLUX.1-schnell"
 lora_repo = None
 lora_path = None
 
-def load_lora(repo_id, style):
+# Styles disponibles
+STYLES = {
+    "Photoréaliste": "photorealistic",
+    "Artistique": "artistic",
+    "Anime": "anime",
+    "3D Render": "3d_render",
+    "Peinture à l'huile": "oil_painting",
+    "Aquarelle": "watercolor",
+    "Cyberpunk": "cyberpunk",
+    "Fantasy": "fantasy"
+}
+
+# Prompt par défaut
+DEFAULT_PROMPT = "A stunning portrait of a woman with flowing red hair, piercing green eyes, natural lighting, ultra detailed, 8k quality, professional photography"
+
+def load_lora(repo_id, style_key):
     global lora_repo, lora_path
     try:
-        filename = f"{style}_lora.safetensors"
+        style_filename = STYLES.get(style_key, style_key)
+        filename = f"{style_filename}_lora.safetensors"
         lora_path = hf_hub_download(repo_id=repo_id, filename=filename)
         lora_repo = repo_id
-        return f"✅ LoRA chargé : {repo_id} ({filename})"
+        return f"✅ LoRA chargé : {repo_id} - Style: {style_key} ({filename})"
     except Exception as e:
         return f"❌ Erreur : {str(e)}"
 
@@ -75,6 +91,7 @@ def generate(prompt, negative_prompt="", width=1024, height=1024, steps=4, seed=
 # Interface Gradio
 with gr.Blocks(title="Flux Schnell + LoRA") as demo:
     gr.Markdown("# 🎨 Générateur Flux Schnell + LoRA")
+    gr.Markdown(f"**Modèle :** `{model_id}`")
     
     with gr.Row():
         with gr.Column():
@@ -84,10 +101,11 @@ with gr.Blocks(title="Flux Schnell + LoRA") as demo:
                 label="Repository LoRA",
                 placeholder="username/repo-name"
             )
-            lora_style = gr.Textbox(
-                label="Style LoRA",
-                placeholder="style_name",
-                value="style"
+            lora_style_dropdown = gr.Dropdown(
+                choices=list(STYLES.keys()),
+                label="Style",
+                value="Photoréaliste",
+                interactive=True
             )
             load_btn = gr.Button("Charger LoRA")
             lora_status = gr.Textbox(label="Status", interactive=False)
@@ -97,11 +115,12 @@ with gr.Blocks(title="Flux Schnell + LoRA") as demo:
             prompt = gr.Textbox(
                 label="Prompt",
                 placeholder="Décrivez votre image...",
-                lines=3
+                lines=3,
+                value=DEFAULT_PROMPT
             )
             negative_prompt = gr.Textbox(
                 label="Negative Prompt (optionnel)",
-                placeholder="Ce que vous ne voulez pas...",
+                placeholder="blurry, low quality, distorted, ugly",
                 lines=2
             )
             
@@ -118,11 +137,17 @@ with gr.Blocks(title="Flux Schnell + LoRA") as demo:
         
         with gr.Column():
             output_image = gr.Image(label="Image générée", type="pil")
+            gr.Markdown("### Conseils")
+            gr.Markdown("""
+            - **Photoréaliste** : Idéal pour portraits et photos réalistes
+            - **Steps recommandés** : 4 pour Schnell (rapide)
+            - **Résolution** : 1024x1024 par défaut, ajustez selon besoin
+            """)
     
     # Actions
     load_btn.click(
         fn=load_lora,
-        inputs=[lora_repo_input, lora_style],
+        inputs=[lora_repo_input, lora_style_dropdown],
         outputs=lora_status
     )