Update app.py

app.py

@@ -146,11 +146,10 @@ def generate(
 # -----------------------
 # 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 de style animé, et sauvegarde des paramètres de génération."
+        "Génération SDXL 1.0 avec LoRA photorealistic + face-helper."
     )
 
     with gr.Row():
@@ -163,7 +162,7 @@ with gr.Blocks(title="SDXL + LoRA (Diffusers)") as demo:
             negative = gr.Textbox(
                 label="Negative prompt",
                 lines=3,
-                value="Poor quality, irregular face structure, ugly, disfigured, misaligned features, blurry, watermark, signature",
+                value="",
             )
 
             seed = gr.Number(
@@ -173,7 +172,7 @@ with gr.Blocks(title="SDXL + LoRA (Diffusers)") as demo:
             steps = gr.Slider(
                 minimum=10,
                 maximum=60,
-                value=10,
+                value=30,
                 step=1,
                 label="Steps",
             )
@@ -188,14 +187,14 @@ with gr.Blocks(title="SDXL + LoRA (Diffusers)") as demo:
             width = gr.Slider(
                 minimum=512,
                 maximum=1536,
-                value=512,
+                value=1024,
                 step=64,
                 label="Width (SDXL natif 1024)",
             )
             height = gr.Slider(
                 minimum=512,
                 maximum=1536,
-                value=512,
+                value=1024,
                 step=64,
                 label="Height (SDXL natif 1024)",
             )
@@ -212,6 +211,18 @@ with gr.Blocks(title="SDXL + LoRA (Diffusers)") as demo:
                 label="Force LoRA photorealistic",
             )
 
+            face_enabled = gr.Checkbox(
+                label="Activer LoRA face-helper (visages)",
+                value=True,
+            )
+            face_weight = gr.Slider(
+                minimum=0.0,
+                maximum=2.0,
+                value=1.0,
+                step=0.05,
+                label="Force LoRA face-helper",
+            )
+
             script_name = gr.Textbox(
                 label="Nom base pour l'image / script",
                 value="sdxl_lora_generation",
@@ -250,5 +261,24 @@ with gr.Blocks(title="SDXL + LoRA (Diffusers)") as demo:
         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],
+    )
+
 if __name__ == "__main__":
     demo.launch()