Update app.py

app.py

@@ -10,31 +10,12 @@ from huggingface_hub import hf_hub_download
 
 BASE_MODEL_ID = "stabilityai/stable-diffusion-xl-base-1.0"
 
-# LoRA réalisme global
-#LORA_PHOTO_REPO = "ostris/photorealistic-slider-sdxl-lora"
-#LORA_PHOTO_REPO = "Muapi/hourglass-body-shape-sd1.5-sdxl-pony-flux-olaz"
-#LORA_PHOTO_ADAPTER = "photo_realism"
-#LORA_PHOTO_ADAPTER = "hourglass_body"
-
+# LoRA cuerpo hourglass (Muapi)
 LORA_HOURGLASS_REPO = "Muapi/hourglass-body-shape-sd1.5-sdxl-pony-flux-olaz"
 LORA_HOURGLASS_FILE = "hourglass-body-shape-sd1.5-sdxl-pony-flux-olaz.safetensors"
 LORA_HOURGLASS_ADAPTER = "hourglass_body"
 
-
-# Teste le téléchargement
-try:
-    local_path = hf_hub_download(
-        repo_id=LORA_HOURGLASS_REPO,
-        filename=LORA_HOURGLASS_FILE,
-        local_dir="./lora_cache"
-    )
-    print(f"Fichier OK : {local_path}")
-except Exception as e:
-    print(f"Erreur repo/fichier : {e}")
-
-
-# LoRA amélioration de visages
-#LORA_FACE_REPO = "ostris/face-helper-sdxl-lora"
+# LoRA visages / détails (Stable Yogi)
 LORA_FACE_REPO = "akash-guptag/Detailers_By_Stable_Yogi"
 LORA_FACE_ADAPTER = "face_helper"
 
@@ -50,19 +31,26 @@ pipe = AutoPipelineForText2Image.from_pretrained(
 pipe.to(DEVICE)
 pipe.set_progress_bar_config(disable=True)
 
-#print("Chargement LoRA photorealistic...")
-#pipe.load_lora_weights(
-#    LORA_PHOTO_REPO,
-#    adapter_name=LORA_PHOTO_ADAPTER,
-#)
+# Téléchargement explicite du LoRA hourglass (debug + cache)
+try:
+    local_path = hf_hub_download(
+        repo_id=LORA_HOURGLASS_REPO,
+        filename=LORA_HOURGLASS_FILE,
+        local_dir="./lora_cache"
+    )
+    print(f"Fichier LoRA hourglass OK : {local_path}")
+except Exception as e:
+    print(f"Erreur repo/fichier hourglass : {e}")
+
 print("Chargement LoRA hourglass body...")
 pipe.load_lora_weights(
-    "Muapi/hourglass-body-shape-sd1.5-sdxl-pony-flux-olaz",  # repo
-    weight_name="hourglass-body-shape-sd1.5-sdxl-pony-flux-olaz.safetensors",  # fichier précis
+    LORA_HOURGLASS_REPO,
+    weight_name=LORA_HOURGLASS_FILE,
     adapter_name=LORA_HOURGLASS_ADAPTER,
+    low_cpu_mem_usage=False,  # important pour certains LoRA hybrides
 )
 
-print("Chargement LoRA face-helper...")
+print("Chargement LoRA face/detail (Stable Yogi)...")
 pipe.load_lora_weights(
     LORA_FACE_REPO,
     adapter_name=LORA_FACE_ADAPTER,
@@ -83,10 +71,10 @@ def generate(
     guidance: float,
     width: float,
     height: float,
-    photo_enabled: bool,
-    photo_weight: float,
     face_enabled: bool,
     face_weight: float,
+    hourglass_enabled: bool,
+    hourglass_weight: float,
     script_name: str,
 ):
     if not prompt:
@@ -103,26 +91,14 @@ def generate(
     adapters = []
     weights = []
 
-    if photo_enabled:
-        adapters.append(LORA_PHOTO_ADAPTER)
-        weights.append(float(photo_weight))
-    else:
-        adapters.append(LORA_PHOTO_ADAPTER)
-        weights.append(0.0)
-
-    if face_enabled:
-        adapters.append(LORA_FACE_ADAPTER)
-        weights.append(float(face_weight))
-    else:
-        adapters.append(LORA_FACE_ADAPTER)
-        weights.append(0.0)
+    # LoRA visages / détails
+    adapters.append(LORA_FACE_ADAPTER)
+    weights.append(float(face_weight) if face_enabled else 0.0)
 
-    # NOUVEAU : Hourglass body
-    hourglass_enabled = True  # ou passe en param UI
-    hourglass_weight = 0.9   # recommandé par l'auteur [web:19]
+    # LoRA hourglass
     adapters.append(LORA_HOURGLASS_ADAPTER)
     weights.append(float(hourglass_weight) if hourglass_enabled else 0.0)
-    
+
     pipe.set_adapters(adapters, adapter_weights=weights)
 
     try:
@@ -143,14 +119,15 @@ def generate(
 
     metadata = {
         "base_model": BASE_MODEL_ID,
-        "photo_lora_repo": LORA_PHOTO_REPO,
-        "photo_lora_adapter": LORA_PHOTO_ADAPTER,
-        "photo_enabled": bool(photo_enabled),
-        "photo_weight": float(photo_weight) if photo_enabled else 0.0,
         "face_lora_repo": LORA_FACE_REPO,
         "face_lora_adapter": LORA_FACE_ADAPTER,
         "face_enabled": bool(face_enabled),
         "face_weight": float(face_weight) if face_enabled else 0.0,
+        "hourglass_lora_repo": LORA_HOURGLASS_REPO,
+        "hourglass_lora_file": LORA_HOURGLASS_FILE,
+        "hourglass_lora_adapter": LORA_HOURGLASS_ADAPTER,
+        "hourglass_enabled": bool(hourglass_enabled),
+        "hourglass_weight": float(hourglass_weight) if hourglass_enabled else 0.0,
         "prompt": prompt,
         "negative_prompt": negative_prompt,
         "seed": seed_int,
@@ -176,13 +153,14 @@ def generate(
 
     return image, script_txt, json_path
 
+
 # -----------------------
 # UI Gradio
 # -----------------------
 with gr.Blocks(title="SDXL + LoRA (Diffusers)") as demo:
     gr.Markdown(
-        "## SDXL base + LoRA SDXL (Diffusers)  \n"
-        "Génération SDXL 1.0 avec LoRA photorealistic + face-helper."
+        "## SDXL base + LoRA SDXL (Diffusers)\n"
+        "Génération SDXL 1.0 avec LoRA detailer + hourglass body."
     )
 
     with gr.Row():
@@ -191,12 +169,15 @@ with gr.Blocks(title="SDXL + LoRA (Diffusers)") as demo:
                 label="Prompt",
                 lines=4,
                 placeholder="Décris la scène à générer avec SDXL...",
-                value="sharp face, detailed skin, realistic eyes, masterpiece, best quality, 1girl, looking at viewer",
+                value=(
+                    "masterpiece, best quality, 1girl, looking at viewer, "
+                    "sharp face, detailed skin, realistic eyes, hourglass figure, slim waist, wide hips"
+                ),
             )
             negative = gr.Textbox(
                 label="Negative prompt",
                 lines=3,
-                value="blurry face, deformed face, bad anatomy",
+                value="blurry face, deformed face, bad anatomy, bad proportions, low quality",
             )
 
             seed = gr.Number(
@@ -233,34 +214,16 @@ with gr.Blocks(title="SDXL + LoRA (Diffusers)") as demo:
                 label="Height (SDXL natif 1024)",
             )
 
-            photo_enabled = gr.Checkbox(
-                label="Activer LoRA photorealistic (rendu global)",
-                value=True,
-            )
-            photo_weight = gr.Slider(
-                minimum=0.0,
-                maximum=2.0,
-                value=1.3,
-                step=0.05,
-                label="Force LoRA photorealistic",
-            )
-
-            
             face_enabled = gr.Checkbox(
-                label="Activer LoRA face-helper (visages)",
+                label="Activer LoRA face/detail (Stable Yogi)",
                 value=True,
             )
             face_weight = gr.Slider(
                 minimum=0.0,
-                maximum=2.0,
-                value=1.0,
+                maximum=1.5,
+                value=0.8,  # recommandé <= 0.8 pour ce LoRA[web:60][web:63]
                 step=0.05,
-                label="Force LoRA face-helper",
-            )
-
-            script_name = gr.Textbox(
-                label="Nom base pour l'image / script",
-                value="sdxl_lora_generation",
+                label="Force LoRA face/detail",
             )
 
             hourglass_enabled = gr.Checkbox(
@@ -271,11 +234,16 @@ with gr.Blocks(title="SDXL + LoRA (Diffusers)") as demo:
             hourglass_weight = gr.Slider(
                 minimum=0.0,
                 maximum=2.0,
-                value=0.9,  # valeur recommandée [web:19]
+                value=0.9,
                 step=0.05,
                 label="Force LoRA hourglass body",
             )
-            
+
+            script_name = gr.Textbox(
+                label="Nom base pour l'image / script",
+                value="sdxl_lora_generation",
+            )
+
             run_btn = gr.Button("Générer", variant="primary")
 
         with gr.Column():
@@ -300,31 +268,10 @@ with gr.Blocks(title="SDXL + LoRA (Diffusers)") as demo:
             guidance,
             width,
             height,
-            photo_enabled,
-            photo_weight,
             face_enabled,
             face_weight,
-            script_name,
             hourglass_enabled,
             hourglass_weight,
-        ],
-        outputs=[out_img, out_script, out_file],
-    )
-
-    run_btn.click(
-        fn=generate,
-        inputs=[
-            prompt,
-            negative,
-            seed,
-            steps,
-            guidance,
-            width,
-            height,
-            photo_enabled,
-            photo_weight,
-            face_enabled,
-            face_weight,
             script_name,
         ],
         outputs=[out_img, out_script, out_file],