Update app.py

app.py

@@ -13,7 +13,7 @@ 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 loading...")
+print("🔥 SDXL NSFW loading...")
 pipe = AutoPipelineForText2Image.from_pretrained(
     BASE_MODEL_ID,
     torch_dtype=DTYPE,
@@ -23,23 +23,28 @@ pipe = AutoPipelineForText2Image.from_pretrained(
 )
 pipe.to(DEVICE)
 
-# MONKEY PATCH NSFW
+# BYPASS NSFW total
 def dummy_checker(images, **kwargs):
     return images, [False] * len(images)
 pipe.safety_checker = dummy_checker
-
 pipe.set_progress_bar_config(disable=True)
 
-print("LoRA face...")
+print("LoRA face/detail...")
 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, steps, guidance, width, height, face_weight, name):
-    generator = torch.Generator(DEVICE).manual_seed(int(seed)) if seed >= 0 else torch.Generator(DEVICE)
+def generate(prompt, negative, seed_str, steps, guidance, width, height, face_weight, filename):
+    # FIX SEED : Gradio donne str → convert safe
+    try:
+        seed = int(float(seed_str)) if seed_str and seed_str.strip() != '' else -1
+    except:
+        seed = -1
+    
+    generator = torch.Generator(DEVICE).manual_seed(seed) if seed >= 0 else torch.Generator(DEVICE)
 
-    pipe.set_adapters([LORA_FACE_ADAPTER], adapter_weights=[face_weight])
+    pipe.set_adapters([LORA_FACE_ADAPTER], adapter_weights=[float(face_weight)])
 
     result = pipe(
         prompt=prompt,
@@ -55,12 +60,12 @@ def generate(prompt, negative, seed, steps, guidance, width, height, face_weight
         "model": BASE_MODEL_ID,
         "lora": LORA_FACE_REPO,
         "prompt": prompt,
-        "seed": int(seed),
+        "seed": seed,
         "steps": int(steps),
         "guidance": float(guidance),
     }
 
-    safe_name = name.strip().replace(" ", "_") or "sdxl_nsfw"
+    safe_name = filename.strip().replace(" ", "_").replace("/", "_") or "sdxl_nsfw"
     img_path = f"outputs/{safe_name}.png"
     json_path = f"outputs/{safe_name}.json"
 
@@ -71,28 +76,32 @@ def generate(prompt, negative, seed, steps, guidance, width, height, face_weight
     with open(json_path, "w", encoding="utf-8") as f:
         json.dump(metadata, f, indent=2)
 
-    return result, json.dumps(metadata, indent=2), json_path  # ← chemin relatif OK pour gr.File
+    return result, json.dumps(metadata, indent=2), json_path
 
-with gr.Blocks(title="🔥 SDXL NSFW Unlocked") as demo:
-    gr.Markdown("# SDXL 1.0 NSFW + Face Detail LoRA\n✅ Safety checker BYPASSED")
+with gr.Blocks(title="🔥 SDXL NSFW + Detail LoRA") as demo:
+    gr.Markdown("# SDXL 1.0 **NSFW UNLOCKED** + Face Detail\n✅ Safety checker BYPASSED")
     
     with gr.Row():
-        with gr.Column(scale=1):
-            prompt = gr.Textbox("Prompt", lines=4, value="1girl nude, masterpiece, detailed anatomy, realistic skin, sharp face")
-            negative = gr.Textbox("Negative", value="blurry, deformed, ugly, low quality")
-            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)
-            face_weight = gr.Slider(0, 1.2, 0.7, step=0.05, label="LoRA Weight")
+        with gr.Column():
+            prompt = gr.Textbox("Prompt", lines=4, value="masterpiece, 1girl nude, detailed anatomy, realistic skin, sharp face, hourglass body")
+            negative = gr.Textbox("Negative", value="blurry, deformed, ugly, extra limbs, low quality")
+            seed = gr.Number("Seed (-1=random)", 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")
+            face_weight = gr.Slider(0.0, 1.2, 0.7, step=0.05, label="LoRA Face Weight")
             filename = gr.Textbox("Filename", value="nsfw_test")
             
-        with gr.Column(scale=1):
+        with gr.Column():
+            output_img = gr.Image("Generated Image")
+            output_json = gr.Textbox("Metadata", lines=8)
+            output_file = gr.File("Download JSON")
+            
             gr.Button("🚀 Generate NSFW", variant="primary", size="lg").click(
                 generate,
                 [prompt, negative, seed, steps, guidance, width, height, face_weight, filename],
-                [gr.Image(label="Generated Image"), gr.Textbox(label="Metadata JSON"), gr.File(label="Download JSON")]  # ← FIX ICI
+                [output_img, output_json, output_file]
             )
 
 if __name__ == "__main__":