Update app.py

app.py

@@ -24,14 +24,13 @@ import gradio as gr
 from huggingface_hub import hf_hub_download, HfApi, CommitOperationAdd
 from huggingface_hub.utils import HfHubHTTPError
 
-# ===== LOGGING =====
 logging.basicConfig(
     level=logging.INFO,
     format="%(asctime)s - %(levelname)s - %(message)s",
     handlers=[logging.StreamHandler(sys.stdout)],
 )
 logger = logging.getLogger("NextGenEngine")
-logger.info("🚀 Next-gen multi-engine Space boot (FLUX / SD3.5 / Adult SDXL)")
+logger.info("🚀 NSFW East Asian Lab boot (19+ · Adult SDXL / FLUX / SD3.5)")
 
 if hasattr(sys.stdout, "reconfigure"):
     sys.stdout.reconfigure(encoding="utf-8")
@@ -41,78 +40,51 @@ if hasattr(sys.stderr, "reconfigure"):
 BACKUP_DATASET_ID = "JustinD/generation-images"
 LAST_GEN_TIME = {}
 COOLDOWN_SECONDS = 12
-
-# Public-ready: visitor gens are NOT uploaded unless Space secret LOG_VISITOR_UPLOADS=1
 LOG_VISITOR_UPLOADS = os.environ.get("LOG_VISITOR_UPLOADS", "0").strip().lower() in {"1", "true", "yes"}
 
-# ---------------------------------------------------------------------------
-# ENGINES
-# ---------------------------------------------------------------------------
 ENGINE_FLUX = "FLUX.1 Schnell (latest fast)"
 ENGINE_SD35 = "SD 3.5 Medium (gated / optional)"
 ENGINE_SDXL = "Adult SDXL (NSFW chefs)"
-
 ENGINE_CHOICES = [ENGINE_SDXL, ENGINE_FLUX, ENGINE_SD35]
 
-# Adult SDXL single-file + Diffusers (proven Justin lane)
 HF_REPO_ID = "AriaVale2/my-safetensors-backup"
+
+# Single-file chefs (Asian realism + proven NSFW pantry)
 MODEL_REPOS = {
+    "asianRealismByStable_v20FP16.safetensors": "Stableyogi/Asian-Realism-Checkpoints",
+    "asianRealismByStable_v30FP16.safetensors": "Stableyogi/Asian-Realism-Checkpoints",
     "pornworksSexyBeauty_v04.safetensors": "AI-Porn/PornWorks-Sexy-Beauty-Aesthetic-Erotic-NSFW-SDXL-Chekpoint",
     "pornworksRealPornPhoto_v04.safetensors": "Manjushri/pornworks-real-porn-photo-realistic-nsfw-sdxl-and-pony-chekpoint",
-    "pornworksHardcoreFantasy_v04.safetensors": "AI-Porn/PornWorks-Hardcore-Fantasy-NSFW-3D-SDXL-Chekpoint",
-    "CyberRealisticXLPlay_V10.0_FP16.safetensors": "cyberdelia/CyberRealisticXL",
-    "Juggernaut-XL_v9_RunDiffusionPhoto_v2.safetensors": "saber-end/Juggernaut-XL-v9",
-    "RealVisXL_V4.0.safetensors": "SG161222/RealVisXL_V4.0",
-    "Pony Diffusion V6 XL.safetensors": "AriaVale2/my-safetensors-backup",
     "Hassaku XL (Illustrious).safetensors": "AriaVale2/my-safetensors-backup",
-    "animagine-xl-4.0.safetensors": "cagliostrolab/animagine-xl-4.0",
-    "OnomaAIResearch_Illustrious-XL-v1.0.safetensors": "AriaVale2/my-safetensors-backup",
-    "UnfilteredAI_NSFW-GEN-ANIME-v2.safetensors": "AriaVale2/my-safetensors-backup",
     "Mala Anime Mix NSFW PonyXL.safetensors": "AriaVale2/my-safetensors-backup",
     "PornWorks_Anime_Desire_NSFW_Anime_Hentai_SDXL_Pony_Chekpoint.safetensors": "AriaVale2/my-safetensors-backup",
 }
 
+# Newer Diffusers Asian NSFW / realism chefs (2025–2026)
 DIFFUSERS_MODELS = {
+    "Diving Real Asian Illustrious v5": "John6666/diving-illustrious-real-asian-v50-sdxl",
+    "Nova Asian Illustrious v6": "John6666/nova-asian-xl-illustrious-v60-sdxl",
+    "WinPic Asian Realism v1.1": "John6666/winpic-asian-realism-v11-sdxl",
+    "Evalisennia East Asian Female v3": "John6666/evalisennia-realistic-east-asian-female-v30-sdxl",
+    "Hardcore Asian Cosplay Illustrious": "John6666/hardcore-asian-cosplay-porn-il-v11-sdxl",
+    "Perfection Asian Illustrious NSFW": "John6666/perfection-asian-ilxl-illustrious-xl-nsfw-sfw-checkpoint-v10-sdxl",
+    "Your Asian GF 2.8D v3": "John6666/your-asian-gf-28d-v30-sdxl",
     "OhMySister Beautiful Asian Woman": "John6666/oms-oh-my-sister-beautiful-asian-woman-v10-sdxl",
-    "OhMyDaughter Beautiful Woman (Pony)": "John6666/omd-ohmaydaughter-beautiful-woman-pony-v10-sdxl",
+    "Asian E-Girl NSFW XL": "John6666/asian-egirl-xl-realistic-uwu-egirl-tattoo-model-sdxl-nsfw-v10-sdxl",
     "Fucktastic Anime Pony NSFW": "John6666/fucktastic-anime-checkpoint-pony-pdxl-porn-nsfw-sfw-22-sdxl",
-    "Mala Anime Mix NSFW Pony v7": "John6666/mala-anime-mix-nsfw-pony-xl-v7-sdxl",
     "YAAI WAI NSFW Illustrious": "John6666/yaai-yet-another-anime-illustrious-model-full-fine-tune-based-on-wai-nsfw-v20-sdxl",
-    "Flare Diffusion NSFW Anime": "John6666/flare-diffusion-nsfw-v16-semi-realism-anime-sdxl",
-    "PornWorks Anime Desire (Diffusers)": "John6666/pornworks-anime-desire-nsfw-pony-sdxl",
+    "Mala Anime Mix NSFW Pony v7": "John6666/mala-anime-mix-nsfw-pony-xl-v7-sdxl",
 }
 
 SDXL_CHOICES = list(DIFFUSERS_MODELS.keys()) + list(MODEL_REPOS.keys())
+DEFAULT_SDXL = "Diving Real Asian Illustrious v5"
 
+# B + C: East Asian slim/tall/big tits lane; Testing Tits is the only style pack
 STYLE_PACKS = {
-    "Men Only": {
-        "prefix": "for men's eyes only, hyper sexual pinup, hungry male gaze",
-        "suffix": "deep cleavage, arched back, bedroom eyes, wet parted lips",
-    },
-    "Art Nude": {
-        "prefix": "classical art nude, museum erotic nude",
-        "suffix": "bare breasts, soft weight, masterpiece lighting",
-    },
-    "Cartoon Slut": {
-        "prefix": "score_9, score_8_up, source_anime, rating_explicit, anime cartoon cel shading",
-        "suffix": "slutty pinup, exaggerated hourglass, heavy cleavage",
-    },
-    "Rembrandt Oil": {
-        "prefix": "Rembrandt oil painting, warm side light, visible brushstrokes",
-        "suffix": "classical academic nude energy",
-    },
-    "4K TV Wide": {
-        "prefix": "16:9 ultra wide cinematic framing for 4K TV",
-        "suffix": "razor sharp when blown up large",
-    },
-    "Photo Gloss": {
-        "prefix": "photorealistic, wet shiny skin, editorial flash",
-        "suffix": "ultra detailed skin pores and fabric",
-    },
     "Testing Tits": {
-        "prefix": "Nice tits, big tits, fun bags",
-        "suffix": "No bra",
-    }
+        "prefix": "nice tits, big tits, fun bags, heavy cleavage focus",
+        "suffix": "no bra, bare breasts energy, chest forward",
+    },
 }
 
 DEFAULT_QUALITY = (
@@ -126,23 +98,23 @@ DEFAULT_QUALITY = (
 )
 
 BASE_NEGATIVES = (
-    "child, loli, underage, teen, "
+    "child, loli, underage, teen, 18, eighteen, "
     "man, men, male, boy, guy, guys, male focus, masculine, "
     "old, elderly, aged, grandma, wrinkly, "
-    "fat, chubby, obese, overweight, belly fat, "
+    "fat, chubby, obese, overweight, belly fat, thick thighs, "
     "beard, mustache, penis, dick, balls, "
     "small breasts, flat chest, bad hands, extra fingers, deformed, "
     "blurry, lowres, watermark, text, censored, ugly"
 )
 
 RANDOM_PROMPTS = [
-    "asian slut",
-    "nude woman"
+    "adult Japanese woman late 20s, sheer lace lingerie, bedroom eyes",
+    "adult Chinese woman late 20s, nude, glossy skin, heavy cleavage",
+    "adult Taiwanese woman late 20s, soft editorial heat, wet parted lips",
+    "adult Vietnamese woman late 20s, no bra, chest forward",
+    "adult Korean woman late 20s, nude woman, large breasts",
 ]
 
-# ---------------------------------------------------------------------------
-# SAFETY
-# ---------------------------------------------------------------------------
 FORBIDDEN_KEYWORDS = []
 if os.path.exists("restricted_words.txt"):
     try:
@@ -155,10 +127,11 @@ if os.path.exists("restricted_words.txt"):
     except Exception as e:
         logger.error(f"restricted words load failed: {e}")
 
+# 19+: block ages under 19 in prompt text
 AGE_PATTERN = re.compile(
     r"\b(one|two|three|four|five|six|seven|eight|nine|ten|"
-    r"eleven|twelve|thirteen|fourteen|fifteen|sixteen|seventeen|"
-    r"1|2|3|4|5|6|7|8|9|10|11|12|13|14|15|16|17)"
+    r"eleven|twelve|thirteen|fourteen|fifteen|sixteen|seventeen|eighteen|"
+    r"1|2|3|4|5|6|7|8|9|10|11|12|13|14|15|16|17|18)"
     r"\s*[-\s]?\s*(year|yr|y\/o|yo)\b"
 )
 
@@ -191,12 +164,7 @@ def build_styles(selected):
     return ", ".join(prefixes), ", ".join(suffixes)
 
 
-# ---------------------------------------------------------------------------
-# LOGGING UPLOAD (same pantry)
-# ---------------------------------------------------------------------------
-
 def should_log_upload(username) -> bool:
-    """Default off for public visitors. Owner can set Space secret LOG_VISITOR_UPLOADS=1."""
     if not LOG_VISITOR_UPLOADS:
         return False
     return True
@@ -206,7 +174,6 @@ def async_log_upload(pil_img, seed, model_name, prompt, username, width, height,
     if not should_log_upload(username):
         logger.info("⏭️ visitor log upload skipped (LOG_VISITOR_UPLOADS not enabled)")
         return
-
     hf_token = os.environ.get("HF_TOKEN")
     if not username or not str(username).strip():
         username = "Unknown_Visitor"
@@ -253,35 +220,32 @@ def async_log_upload(pil_img, seed, model_name, prompt, username, width, height,
             time.sleep(1.5 * attempt)
 
 
-# ---------------------------------------------------------------------------
-# PIPELINE CACHE
-# ---------------------------------------------------------------------------
 _pipes = {}
-_current_key = None
 
 
 def _dtype():
     if torch.cuda.is_available():
-        # Flux prefers bfloat16 when available
         return torch.bfloat16 if torch.cuda.is_bf16_supported() else torch.float16
     return torch.float32
 
 
 def unload_all():
-    global _pipes, _current_key
+    global _pipes
     _pipes.clear()
-    _current_key = None
     gc.collect()
     if torch.cuda.is_available():
         torch.cuda.empty_cache()
 
 
+def _hf_token():
+    return os.environ.get("HF_TOKEN") or os.environ.get("HUGGINGFACE_HUB_TOKEN")
+
+
 def get_flux_pipe():
     key = "flux-schnell"
     if key in _pipes:
         return _pipes[key]
     from diffusers import FluxPipeline
-
     unload_all()
     logger.info("Loading FLUX.1 Schnell…")
     token = _hf_token()
@@ -296,28 +260,22 @@ def get_flux_pipe():
     return pipe
 
 
-def _hf_token():
-    return os.environ.get("HF_TOKEN") or os.environ.get("HUGGINGFACE_HUB_TOKEN")
-
-
 def get_sd35_pipe():
     key = "sd35-medium"
     if key in _pipes:
         return _pipes[key]
     from diffusers import StableDiffusion3Pipeline
     from huggingface_hub import login as hf_login
-
     token = _hf_token()
     if not token:
         raise RuntimeError(
-            "SD 3.5 is gated. Add a Write/Read HF token as Space secret named HF_TOKEN, "
-            "and accept the license at https://huggingface.co/stabilityai/stable-diffusion-3.5-medium"
+            "SD 3.5 is gated. Add HF_TOKEN secret and accept "
+            "https://huggingface.co/stabilityai/stable-diffusion-3.5-medium"
         )
     try:
         hf_login(token=token, add_to_git_credential=False)
     except Exception as e:
         logger.warning(f"hf_login warning: {e}")
-
     unload_all()
     logger.info("Loading SD 3.5 Medium (authenticated)…")
     pipe = StableDiffusion3Pipeline.from_pretrained(
@@ -336,8 +294,6 @@ def get_sdxl_pipe(model_name: str):
     if key in _pipes:
         return _pipes[key]
     from diffusers import StableDiffusionXLPipeline, EulerDiscreteScheduler
-
-    # Keep only one heavy pipe resident
     unload_all()
     dtype = torch.float16 if torch.cuda.is_available() else torch.float32
     logger.info(f"Loading Adult SDXL chef: {model_name}")
@@ -375,9 +331,6 @@ def enhance_prompt(prompt, styles):
     return ", ".join([p for p in parts if p])
 
 
-# ---------------------------------------------------------------------------
-# GENERATE
-# ---------------------------------------------------------------------------
 @spaces.GPU(duration=180) if SPACES_AVAILABLE else (lambda f: f)
 def generate(
     engine,
@@ -395,19 +348,16 @@ def generate(
 ):
     if not prompt or not str(prompt).strip():
         return None, "❌ Prompt missing", "", []
-
     now = time.time()
     user_key = username or "anonymous"
     if user_key in LAST_GEN_TIME and now - LAST_GEN_TIME[user_key] < COOLDOWN_SECONDS:
         left = int(COOLDOWN_SECONDS - (now - LAST_GEN_TIME[user_key]))
         gr.Warning(f"Cooldown {left}s")
         return None, f"⏳ Cooldown {left}s", "", []
-
     ok, bad = check_prompt_safety(prompt)
     if not ok:
         gr.Warning("Restricted content")
         return None, "❌ Restricted content", "", []
-
     batch = int(max(1, min(3, int(batch))))
     seeds = [
         random.randint(0, 2**31 - 1) if int(seed) < 0 else int(seed) + i
@@ -417,22 +367,18 @@ def generate(
     neg = BASE_NEGATIVES
     if negative and str(negative).strip():
         neg = f"{BASE_NEGATIVES}, {negative.strip()}"
-
     LAST_GEN_TIME[user_key] = now
     gallery = []
     jobs = []
-
     try:
         device = "cuda" if torch.cuda.is_available() else "cpu"
-
         if engine == ENGINE_FLUX:
             pipe = get_flux_pipe()
-            # Schnell: low steps, guidance often ~0
             use_steps = int(max(1, min(8, int(steps) if int(steps) <= 8 else 4)))
-            # FLUX ignores negatives — shove woman-only steer into the prompt
             flux_prompt = (
                 enhanced
-                + ", 1girl, solo, adult woman only, no men, no males, no old people, slim fit body"
+                + ", 1girl, solo, adult East Asian woman only, no men, no males, "
+                "no old people, tall slim body, large breasts"
             )
             for s in seeds:
                 gen = torch.Generator(device=device).manual_seed(int(s))
@@ -448,7 +394,6 @@ def generate(
                 label = f"FLUX · seed {s} · {datetime.now().strftime('%I:%M:%S %p')}"
                 gallery.append((out, label))
                 jobs.append((out, s, "FLUX.1-schnell", prompt, username, width, height, use_steps, 0.0))
-
         elif engine == ENGINE_SD35:
             pipe = get_sd35_pipe()
             use_steps = int(max(15, min(40, int(steps))))
@@ -466,7 +411,6 @@ def generate(
                 label = f"SD3.5 · seed {s} · {datetime.now().strftime('%I:%M:%S %p')}"
                 gallery.append((out, label))
                 jobs.append((out, s, "SD3.5-medium", prompt, username, width, height, use_steps, cfg))
-
         else:
             pipe = get_sdxl_pipe(sdxl_model)
             use_steps = int(max(10, min(40, int(steps))))
@@ -484,21 +428,16 @@ def generate(
                 label = f"SDXL · {sdxl_model[:28]} · seed {s}"
                 gallery.append((out, label))
                 jobs.append((out, s, sdxl_model, prompt, username, width, height, use_steps, cfg))
-
         info = f"{engine} | seeds {seeds} | {width}x{height} | enhanced prompt ready"
         return gallery, info, enhanced, jobs
-
     except Exception as e:
         logger.exception("generate failed")
         msg = str(e)
         if "gated" in msg.lower() or "401" in msg or "403" in msg:
-            msg += " — open the model page on HF, accept the license, ensure Space secret HF_TOKEN can read it (or use HF login)."
+            msg += " — accept the model license and ensure Space secret HF_TOKEN can read it."
         return None, f"❌ {msg}", "", []
 
 
-# ---------------------------------------------------------------------------
-# UI
-# ---------------------------------------------------------------------------
 js_visitor = """
 function() {
   let id = localStorage.getItem('adult_ai_visitor_id');
@@ -521,35 +460,39 @@ THEME = gr.themes.Default(primary_hue="purple", secondary_hue="indigo", neutral_
 def create_ui():
     with gr.Blocks() as demo:
         user_id = gr.Textbox(visible=False)
-        gr.Markdown("# ⚡ Next-Gen Image Lab")
+        gr.Markdown("# ⚡ NSFW East Asian Lab")
         gr.Markdown(
-            """### 18+ · ZeroGPU · first model load is slow · queues happen
+            """### 19+ only · ZeroGPU · first model load is slow · queues happen
 
-**Defaults for this lane:** Adult SDXL first (NSFW chefs, anti-man/old/fat locked).  
-FLUX is optional tech demo (weaker NSFW control). SD 3.5 is gated/optional.  
-Visitor gens are **not** saved unless secret `LOG_VISITOR_UPLOADS=1`."""
+**This Space is adult NSFW East Asian women only** (slim / tall / big tits).
+Default engine: Adult SDXL with newer Asian Illustrious / Pony chefs.
+Style pack: **Testing Tits** only.
+FLUX and SD 3.5 stay optional. Visitor gens are **not** saved unless secret `LOG_VISITOR_UPLOADS=1`."""
         )
-
         with gr.Row():
             with gr.Column(scale=2):
                 prompt = gr.Textbox(label="Prompt", value=random.choice(RANDOM_PROMPTS), lines=7)
-                negative = gr.Textbox(label="Negatives (SD3.5 / SDXL; ignored by FLUX)", value=BASE_NEGATIVES, lines=4)
+                negative = gr.Textbox(
+                    label="Negatives (SD3.5 / SDXL; ignored by FLUX)",
+                    value=BASE_NEGATIVES,
+                    lines=4,
+                )
                 styles = gr.CheckboxGroup(
-                    label="Style packs (stackable)",
+                    label="Style packs",
                     choices=list(STYLE_PACKS.keys()),
-                    value=["Men Only", "Photo Gloss"],
+                    value=["Testing Tits"],
                 )
                 go = gr.Button("GENERATE", variant="primary", size="lg")
             with gr.Column(scale=1):
                 engine = gr.Radio(
-                    label="Engine (Adult SDXL = default for this lane)",
+                    label="Engine (Adult SDXL = default)",
                     choices=ENGINE_CHOICES,
                     value=ENGINE_SDXL,
                 )
                 sdxl_model = gr.Dropdown(
                     label="Adult SDXL chef (only used on Adult SDXL engine)",
                     choices=SDXL_CHOICES,
-                    value="OhMySister Beautiful Asian Woman",
+                    value=DEFAULT_SDXL,
                 )
                 seed = gr.Number(label="Seed (-1 random)", value=-1, precision=0)
                 width = gr.Slider(512, 1536, value=1024, step=64, label="Width")
@@ -557,11 +500,9 @@ Visitor gens are **not** saved unless secret `LOG_VISITOR_UPLOADS=1`."""
                 steps = gr.Slider(1, 40, value=28, step=1, label="Steps (SDXL~28, FLUX~4, SD3.5~28)")
                 cfg = gr.Slider(0.0, 12.0, value=6.0, step=0.5, label="CFG (SDXL~6, FLUX uses 0)")
                 batch = gr.Slider(1, 3, value=1, step=1, label="Batch (max 3)")
-
         gallery = gr.Gallery(label="Outputs", columns=3, height="auto")
         info = gr.Textbox(label="Run info", visible=True)
         meta = gr.Textbox(label="Enhanced prompt", visible=True)
-
         demo.load(fn=None, inputs=None, outputs=user_id, js=js_visitor)
 
         def on_go(*args):