Update app.py

app.py

@@ -15,37 +15,31 @@ if hf_token:
 
 model_id = "black-forest-labs/FLUX.1-schnell"
 
-# Repository LoRA par défaut (à adapter selon votre repository)
-DEFAULT_LORA_REPO = "XLabs-AI/flux-RealismLora"  # Exemple de repo avec LoRA photoréaliste
+# 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"
+
+# Note: FLUX.1-schnell n'utilise PAS de LoRA par défaut
+# Le modèle de base est déjà optimisé pour la génération rapide
+# Si vous voulez ajouter des LoRA, utilisez des repositories communautaires comme:
+# - XLabs-AI/flux-RealismLora
+# - Shakker-Labs/FLUX.1-dev-LoRA-collections
+# etc.
 
 lora_repo = None
 lora_path = None
 
-# Styles disponibles
-STYLES = {
-    "Photoréaliste": "lora",  # Le fichier s'appelle souvent "lora.safetensors"
-    "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):
+def load_lora(repo_id, filename):
+    """Charge un LoRA depuis un repository HuggingFace"""
     global lora_repo, lora_path
+    if not repo_id or not filename:
+        return "⚠️ Veuillez renseigner le repository ET le nom du fichier"
+    
     try:
-        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} - Style: {style_key} ({filename})"
+        return f"✅ LoRA chargé : {repo_id}/{filename}"
     except Exception as e:
-        return f"❌ Erreur : {str(e)}"
+        return f"❌ Erreur : {str(e)}\n\nVérifiez que le repository contient bien ce fichier."
 
 @spaces.GPU(duration=120)
 def generate(prompt, negative_prompt="", width=1024, height=1024, steps=4, seed=-1, lora_scale=0.8):
@@ -100,17 +94,24 @@ with gr.Blocks(title="Flux Schnell + LoRA") as demo:
     with gr.Row():
         with gr.Column():
             # LoRA
-            gr.Markdown("### Charger un LoRA (optionnel)")
+            gr.Markdown("### 🎨 Ajouter un LoRA (optionnel)")
+            gr.Markdown("""
+            ⚠️ **Important** : Le modèle de base FLUX.1-schnell fonctionne **sans LoRA**.
+            
+            Si vous voulez ajouter un style avec un LoRA, utilisez des repositories communautaires :
+            - `XLabs-AI/flux-RealismLora` (fichier: `lora.safetensors`)
+            - `Shakker-Labs/FLUX.1-dev-LoRA-add-details` (fichier: `FLUX-dev-lora-add_details.safetensors`)
+            """)
+            
             lora_repo_input = gr.Textbox(
                 label="Repository LoRA",
-                placeholder="username/repo-name",
-                value=DEFAULT_LORA_REPO
+                placeholder="Ex: XLabs-AI/flux-RealismLora",
+                value=""
             )
-            lora_style_dropdown = gr.Dropdown(
-                choices=list(STYLES.keys()),
-                label="Style",
-                value="Photoréaliste",
-                interactive=True
+            lora_filename = gr.Textbox(
+                label="Nom du fichier LoRA",
+                placeholder="Ex: lora.safetensors",
+                value=""
             )
             load_btn = gr.Button("Charger LoRA")
             lora_status = gr.Textbox(label="Status", interactive=False)
@@ -142,21 +143,22 @@ 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("### 💡 Conseils")
             gr.Markdown("""
-            - **Repository LoRA par défaut** : XLabs-AI/flux-RealismLora
-            - **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
+            - **Modèle de base** : FLUX.1-schnell (rapide, sans LoRA nécessaire)
+            - **Steps recommandés** : 4 pour Schnell
+            - **Résolution** : 1024x1024 par défaut
+            - **LoRA Scale** : 0.8 pour un effet équilibré (ajustez entre 0-1)
             
-            💡 **Note** : Le repository `black-forest-labs/FLUX.1-schnell` est le modèle de base, 
-            pas un repository LoRA. Pour utiliser des LoRA, trouvez des repositories dédiés sur HuggingFace.
+            **Exemples de repositories LoRA compatibles :**
+            - [XLabs-AI/flux-RealismLora](https://huggingface.co/XLabs-AI/flux-RealismLora)
+            - [Shakker-Labs collections](https://huggingface.co/Shakker-Labs)
             """)
     
     # Actions
     load_btn.click(
         fn=load_lora,
-        inputs=[lora_repo_input, lora_style_dropdown],
+        inputs=[lora_repo_input, lora_filename],
         outputs=lora_status
     )