fix(gradio): fix OpenAPI bool schema crash and add Base NSFW Adapter slot with Civitai 1307155

app.py

@@ -33,7 +33,31 @@ LORA_MAX_BYTES = 2 * 1024**3
 DEFAULT_FPS = 16
 
 # =========================================================================
-# Preset Catalog for Wan 2.1 / 2.2 LoRAs
+# Base NSFW Adapters (Unlocks adult generation on Wan Base Models)
+# =========================================================================
+NSFW_BASE_ADAPTERS = {
+    "WAN General NSFW Model (Civitai 1307155 - Recomendado)": {
+        "source": "https://civitai.com/models/1307155",
+        "trigger_words": "nsfwsks, uncensored, naked, explicit",
+        "default_strength": 1.0,
+        "description": "Desbloqueia anatomia e geração explícita no modelo base Wan (Civitai Model 1307155).",
+    },
+    "HMNSFW AIO V2 Base (Civitai 2834417)": {
+        "source": "https://civitai.com/api/download/models/3206518",
+        "trigger_words": "hmmotion",
+        "default_strength": 0.5,
+        "description": "Adaptador anatômico geral HMNSFW V2 para Wan 2.1/2.2.",
+    },
+    "Desativado / Apenas Base SFW": {
+        "source": "",
+        "trigger_words": "",
+        "default_strength": 0.0,
+        "description": "Usa apenas o modelo base padrão sem adaptador NSFW adicional.",
+    },
+}
+
+# =========================================================================
+# Preset Catalog for Wan 2.1 / 2.2 LoRAs (Actions, Motions & Styles)
 # =========================================================================
 LORA_PRESETS = {
     "None / Desativado": {
@@ -42,6 +66,13 @@ LORA_PRESETS = {
         "default_strength": 1.0,
         "description": "Nenhum LoRA selecionado neste slot.",
     },
+    "WAN General NSFW model (Civitai 1307155)": {
+        "type": "civitai",
+        "source": "https://civitai.com/models/1307155",
+        "trigger_words": "nsfwsks, uncensored, naked, explicit",
+        "default_strength": 1.0,
+        "description": "LoRA Geral NSFW para Wan 2.1 / 2.2 (Civitai 1307155).",
+    },
     "HMNSFW AIO V2 / hmmotion (Wan 2.1 / 2.2)": {
         "type": "civitai",
         "source": "https://civitai.com/api/download/models/3206518",
@@ -402,10 +433,10 @@ def _fit_keyframe(image_input, target_width: int, target_height: int) -> Image.I
     return img
 
 
-def get_duration(prompt, negative_prompt, input_image, canvas, num_frames, fps, steps, *a, **k):
+def get_duration(prompt, negative_prompt, input_image, nsfw_base_choice, nsfw_base_strength, canvas, num_frames, fps, steps, *a, **k):
     steps = int(steps)
     num_frames = int(num_frames)
-    h, w = CANVACES_GET = CANVASES.get(canvas, (480, 832))
+    h, w = CANVASES.get(canvas, (480, 832))
     pixels_per_frame = h * w
     total_tokens = (pixels_per_frame / 256) * (num_frames / 4)
     estimated = int(steps * (total_tokens * 0.00012) + 20)
@@ -488,6 +519,8 @@ def generate_video(
     prompt: str,
     negative_prompt: str,
     input_image: Image.Image | str | None,
+    nsfw_base_choice: str,
+    nsfw_base_strength: float,
     canvas: str,
     num_frames: int,
     fps: int,
@@ -527,6 +560,17 @@ def generate_video(
     lora_configs = []
     active_labels = []
 
+    # 1. Base NSFW Adapter
+    if nsfw_base_choice in NSFW_BASE_ADAPTERS and nsfw_base_strength > 0:
+        base_spec = NSFW_BASE_ADAPTERS[nsfw_base_choice]
+        if base_spec.get("source"):
+            b_url = normalize_civitai_url(base_spec["source"])
+            if b_url:
+                b_path, b_label = _download_lora_url(b_url, civitai_api_key)
+                lora_configs.append((b_path, float(nsfw_base_strength)))
+                active_labels.append(f"🔞 NSFW Base: {nsfw_base_choice.split('(')[0].strip()} (@ {nsfw_base_strength:g})")
+
+    # 2. Slot 1 LoRA
     l1_path, l1_label, l1_scale = resolve_single_lora(
         lora1_preset, lora1_custom_url, lora1_hf_repo, lora1_hf_file, lora1_local_file, lora1_strength, civitai_api_key
     )
@@ -534,6 +578,7 @@ def generate_video(
         lora_configs.append((l1_path, l1_scale))
         active_labels.append(f"{l1_label} (@ {lora1_strength:g})")
 
+    # 3. Slot 2 LoRA
     l2_path, l2_label, l2_scale = resolve_single_lora(
         lora2_preset, lora2_custom_url, lora2_hf_repo, lora2_hf_file, lora2_local_file, lora2_strength, civitai_api_key
     )
@@ -661,11 +706,11 @@ with gr.Blocks(css=custom_css, title="Wan 2.1 / 2.2 Video Studio") as app:
         <div class="header-card">
             <div style="margin-bottom: 12px;">
                 <span class="badge badge-zerogpu">⚡ ZeroGPU</span>
-                <span class="badge badge-model">🎬 Wan 2.1 (14B Diffusers)</span>
-                <span class="badge badge-multilora">🧩 Multi-LoRA Engine</span>
+                <span class="badge badge-model">🎬 Wan 2.1 / 2.2 (Diffusers)</span>
+                <span class="badge badge-multilora">🧩 Base NSFW + Multi-LoRA Engine</span>
             </div>
             <h1>Wan 2.1 / 2.2 AI Video Studio</h1>
-            <p>Image-to-Video (I2V) & Text-to-Video (T2V) com suporte a múltiplos LoRAs e resoluções otimizadas (Múltiplos de 16).</p>
+            <p>Image-to-Video (I2V) & Text-to-Video (T2V) com Adaptador NSFW Base, Multi-LoRA e Resoluções Múltiplas de 16.</p>
         </div>
         """
     )
@@ -676,14 +721,14 @@ with gr.Blocks(css=custom_css, title="Wan 2.1 / 2.2 Video Studio") as app:
                 label="Prompt",
                 placeholder="Descreva a cena detalhada do vídeo...",
                 lines=4,
-                value="A cinematic shot of a beautiful woman with glowing eyes, dramatic lighting, 8k masterpiece",
+                value="nsfwsks, a cinematic shot of a beautiful woman with glowing eyes, dramatic lighting, 8k masterpiece",
             )
 
             negative_prompt = gr.Textbox(
                 label="Negative Prompt",
                 placeholder="Elementos indesejados (distorções, membros extras, baixa qualidade)...",
                 lines=2,
-                value="deformed, bad anatomy, extra limbs, blurry, low resolution, bad quality, distortion",
+                value="deformed, bad anatomy, extra limbs, blurry, low resolution, bad quality, distortion, censorship",
             )
 
             with gr.Accordion("📖 Guia & Modelos de Prompt HMNSFW / hmmotion (Wan 2.1 / 2.2)", open=False):
@@ -702,15 +747,32 @@ with gr.Blocks(css=custom_css, title="Wan 2.1 / 2.2 Video Studio") as app:
                 hmnsfw_example_btn = gr.Button("💡 Inserir Exemplo Completo de Prompt HMNSFW", elem_classes=["trigger-btn"])
 
             with gr.Tabs():
+                with gr.TabItem("🔞 Adaptador Base NSFW (Essencial)"):
+                    gr.Markdown("Como o modelo base do Wan é treinado sem conteúdo explícito, este adaptador é carregado como base para desbloquear anatomia sem gastar seus slots de LoRA!")
+                    with gr.Row():
+                        nsfw_base_choice = gr.Dropdown(
+                            label="Adaptador Base NSFW",
+                            choices=list(NSFW_BASE_ADAPTERS.keys()),
+                            value="WAN General NSFW Model (Civitai 1307155 - Recomendado)",
+                        )
+                        nsfw_base_strength = gr.Slider(
+                            label="Força do Adaptador Base NSFW",
+                            minimum=0.0,
+                            maximum=1.5,
+                            step=0.05,
+                            value=1.0,
+                        )
+                    nsfw_base_trigger_btn = gr.Button("📋 Inserir Trigger `nsfwsks` no Prompt", elem_classes=["trigger-btn"])
+
                 with gr.TabItem("🖼️ Image-to-Video (I2V)"):
                     gr.Markdown("Faça upload de uma imagem inicial (`First Frame`). O sistema ajustará o corte automaticamente para a proporção múltipla de 16 selecionada.")
-                    input_image = gr.Image(label="First Frame (Imagem Inicial)", type="filepath")
+                    input_image = gr.Image(label="First Frame (Imagem Inicial)", type="pil")
 
-                with gr.TabItem("🎨 LoRA Slot 1 (Estilo / Ação)"):
+                with gr.TabItem("🎨 LoRA Slot 1 (Ação / Pose)"):
                     lora1_preset = gr.Dropdown(
                         label="LoRA Slot 1 Preset / Fonte",
                         choices=all_preset_choices,
-                        value="None / Desativado",
+                        value="HMNSFW AIO V2 / hmmotion (Wan 2.1 / 2.2)",
                     )
                     with gr.Group(visible=False) as lora1_custom_url_grp:
                         lora1_custom_url = gr.Textbox(
@@ -726,7 +788,7 @@ with gr.Blocks(css=custom_css, title="Wan 2.1 / 2.2 Video Studio") as app:
 
                     lora1_trigger_display = gr.Textbox(
                         label="Trigger Words",
-                        value="",
+                        value=LORA_PRESETS.get("HMNSFW AIO V2 / hmmotion (Wan 2.1 / 2.2)", {}).get("trigger_words", ""),
                         interactive=False,
                     )
                     add_lora1_triggers_btn = gr.Button("📋 Inserir Trigger Words no Prompt", elem_classes=["trigger-btn"])
@@ -736,7 +798,7 @@ with gr.Blocks(css=custom_css, title="Wan 2.1 / 2.2 Video Studio") as app:
                         minimum=0.0,
                         maximum=2.0,
                         step=0.05,
-                        value=1.0,
+                        value=0.5,
                     )
 
                 with gr.TabItem("🎬 LoRA Slot 2 (Movimento / Câmera)"):
@@ -855,6 +917,7 @@ with gr.Blocks(css=custom_css, title="Wan 2.1 / 2.2 Video Studio") as app:
         fn=on_lora1_change,
         inputs=[lora1_preset],
         outputs=[lora1_custom_url_grp, lora1_hf_grp, lora1_local_grp, lora1_trigger_display, lora1_strength],
+        api_name=False,
     )
 
     def on_lora2_change(preset_val):
@@ -873,6 +936,7 @@ with gr.Blocks(css=custom_css, title="Wan 2.1 / 2.2 Video Studio") as app:
         fn=on_lora2_change,
         inputs=[lora2_preset],
         outputs=[lora2_custom_url_grp, lora2_hf_grp, lora2_local_grp, lora2_trigger_display, lora2_strength],
+        api_name=False,
     )
 
     def append_triggers(curr_prompt, triggers):
@@ -904,9 +968,10 @@ with gr.Blocks(css=custom_css, title="Wan 2.1 / 2.2 Video Studio") as app:
             "The setting is a bed with dark grey sheets under warm, low indoor lighting."
         )
 
-    hmnsfw_example_btn.click(fn=load_hmnsfw_example, outputs=[prompt])
-    add_lora1_triggers_btn.click(fn=append_triggers, inputs=[prompt, lora1_trigger_display], outputs=[prompt])
-    add_lora2_triggers_btn.click(fn=append_triggers, inputs=[prompt, lora2_trigger_display], outputs=[prompt])
+    hmnsfw_example_btn.click(fn=load_hmnsfw_example, outputs=[prompt], api_name=False)
+    nsfw_base_trigger_btn.click(fn=lambda p: append_triggers(p, "nsfwsks, uncensored, explicit"), inputs=[prompt], outputs=[prompt], api_name=False)
+    add_lora1_triggers_btn.click(fn=append_triggers, inputs=[prompt, lora1_trigger_display], outputs=[prompt], api_name=False)
+    add_lora2_triggers_btn.click(fn=append_triggers, inputs=[prompt, lora2_trigger_display], outputs=[prompt], api_name=False)
 
     generate_btn.click(
         fn=generate_video,
@@ -914,6 +979,8 @@ with gr.Blocks(css=custom_css, title="Wan 2.1 / 2.2 Video Studio") as app:
             prompt,
             negative_prompt,
             input_image,
+            nsfw_base_choice,
+            nsfw_base_strength,
             canvas,
             num_frames,
             fps,
@@ -937,7 +1004,8 @@ with gr.Blocks(css=custom_css, title="Wan 2.1 / 2.2 Video Studio") as app:
             civitai_api_key,
         ],
         outputs=[output_video, output_report, seed],
+        api_name="generate_video",
     )
 
 if __name__ == "__main__":
-    app.launch(show_error=True, allowed_paths=[OUTPUT_DIR])
+    app.launch(show_api=False, show_error=True, allowed_paths=[OUTPUT_DIR])