Update app.py

app.py

@@ -13,7 +13,6 @@ LORA_FACE_ADAPTER = "face_detail"
 DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
 DTYPE = torch.float16 if DEVICE == "cuda" else torch.float32
 
-print("🔥 SDXL NSFW...")
 pipe = AutoPipelineForText2Image.from_pretrained(
     BASE_MODEL_ID,
     torch_dtype=DTYPE,
@@ -26,21 +25,15 @@ pipe.to(DEVICE)
 def dummy_checker(images, **kwargs):
     return images, [False] * len(images)
 pipe.safety_checker = dummy_checker
-pipe.set_progress_bar_config(disable=True)
 
-print("LoRA...")
+pipe.set_progress_bar_config(disable=True)
 pipe.load_lora_weights(LORA_FACE_REPO, adapter_name=LORA_FACE_ADAPTER)
 
 os.makedirs("outputs", exist_ok=True)
 
 @spaces.GPU()
 def generate(prompt, negative, seed_str, steps, guidance, width, height, face_weight, filename):
-    # SEED SAFE
-    try:
-        seed = int(float(seed_str)) if seed_str.strip() else -1
-    except:
-        seed = -1
-    
+    seed = int(float(seed_str)) if seed_str.strip() else -1
     generator = torch.Generator(DEVICE).manual_seed(seed) if seed >= 0 else torch.Generator(DEVICE)
 
     pipe.set_adapters([LORA_FACE_ADAPTER], adapter_weights=[float(face_weight)])
@@ -55,11 +48,11 @@ def generate(prompt, negative, seed_str, steps, guidance, width, height, face_we
         generator=generator,
     ).images[0]
 
-    safe_name = filename.strip().replace(" ", "_").replace("/", "_") or "sdxl_nsfw"
+    safe_name = filename.strip().replace(" ", "_") or "sdxl_nsfw"
     img_path = f"outputs/{safe_name}.png"
     json_path = f"outputs/{safe_name}.json"
 
-    metadata = {"prompt": prompt, "seed": seed, "steps": int(steps), "guidance": float(guidance)}
+    metadata = {"prompt": prompt, "seed": seed, "steps": int(steps)}
     
     pnginfo = PngImagePlugin.PngInfo()
     pnginfo.add_text("params", json.dumps(metadata))
@@ -70,30 +63,39 @@ def generate(prompt, negative, seed_str, steps, guidance, width, height, face_we
 
     return result, json.dumps(metadata, indent=2), json_path
 
-with gr.Blocks(title="🔥 SDXL NSFW") as demo:
-    gr.Markdown("# SDXL 1.0 **NSFW** + Face LoRA\n✅ No Errors")
+with gr.Blocks(title="🔥 SDXL NSFW + Detail LoRA") as demo:
+    gr.Markdown("# SDXL 1.0 **NSFW** ✅ Face Detail LoRA")
     
     with gr.Row():
         with gr.Column():
-            prompt = gr.Textbox("Prompt", value="1girl nude, masterpiece, detailed skin, sharp face")
-            negative = gr.Textbox("Negative", value="blurry, deformed, ugly")
-            seed = gr.Number("Seed", value=-1)
-            steps = gr.Slider(20, 50, 35, step=1)
-            guidance = gr.Slider(5, 12, 7.5, step=0.1)
-            width = gr.Slider(512, 1536, 1024, step=64)
-            height = gr.Slider(512, 1536, 1024, step=64)
-            lora_w = gr.Slider(0, 1.2, 0.7, step=0.05, label="LoRA Weight")
-            name = gr.Textbox("Filename", value="test_nsfw")
+            prompt = gr.Textbox(
+                label="Prompt", 
+                placeholder="1girl nude, masterpiece, detailed anatomy, realistic skin, sharp face",
+                lines=4,
+                value=""  # ← FIX: vide, valeur dans placeholder
+            )
+            negative = gr.Textbox(
+                label="Negative Prompt",
+                placeholder="blurry, deformed, ugly, low quality, extra limbs",
+                value=""
+            )
+            seed = gr.Number(label="Seed", value=-1)
+            steps = gr.Slider(20, 50, 35, step=1, label="Steps")
+            guidance = gr.Slider(5.0, 12.0, 7.5, step=0.1, label="Guidance")
+            width = gr.Slider(512, 1536, 1024, step=64, label="Width")
+            height = gr.Slider(512, 1536, 1024, step=64, label="Height")
+            lora_weight = gr.Slider(0.0, 1.2, 0.7, step=0.05, label="LoRA Weight")
+            filename = gr.Textbox(label="Filename", value="nsfw_test")
             
         with gr.Column():
-            img_out = gr.Image(label="Image")
-            json_out = gr.Textbox(label="Metadata", lines=8)
-            file_out = gr.File(label="Download JSON")  # ← FIX: label= au lieu de positional
+            output_image = gr.Image(label="Generated Image")
+            output_metadata = gr.Textbox(label="Metadata JSON", lines=10)
+            output_file = gr.File(label="📥 Download JSON")
             
-            gr.Button("🚀 NSFW Generate", variant="primary").click(
+            gr.Button("🚀 Generate NSFW", variant="primary", size="lg").click(
                 generate,
-                [prompt, negative, seed, steps, guidance, width, height, lora_w, name],
-                [img_out, json_out, file_out]
+                [prompt, negative, seed, steps, guidance, width, height, lora_weight, filename],
+                [output_image, output_metadata, output_file]
             )
 
 demo.launch()