Upload app.py with huggingface_hub

app.py

@@ -33,6 +33,25 @@ SIZE_PRESETS = {
 
 pipe = None
 
+# Flagged words: generation proceeds normally, but flagged results are NOT
+# uploaded to the public community bucket. Silent, no user-facing warning.
+FLAGGED = (
+    "naked", "nude", "nsfw", "porn", "pornographic", "hentai", "erotic",
+    "sex", "sexual", "undress", "unclothed", "topless", "bottomless",
+    "nak3d", "nudes", "bathing", "panties", "lingerie", "underwear",
+    "orgasm", "nipple", "nipples", "genital", "penis", "vagina", "boobs",
+    "titties", "incest", "rape",
+    "child", "kid ", "kids", "minor", "teen", "teenager", "underage",
+    "schoolgirl", "schoolboy", "loli", "shota", "toddler", "infant",
+    "小女孩", "小男孩", "少女", "儿童", "裸体", "裸足", "脱下", "脱掉",
+    "裸", "色情", "性爱", "幼女",
+)
+
+
+def is_flagged(*texts):
+    joined = " " + " ".join(t or "" for t in texts).lower() + " "
+    return any(w in joined for w in FLAGGED)
+
 
 def get_pipeline():
     global pipe
@@ -98,23 +117,26 @@ def generate(prompt, negative_prompt, size_preset, steps, seed, edit_image):
     png_path = os.path.join(OUT, f"{image_id}.png")
     image.save(png_path)
 
+    flagged = is_flagged(prompt, negative_prompt)
+
     words = (prompt or "").strip().replace("\n", " ").split()[:8]
     title = " ".join(words).title()[:80] if words else f"Qwen Image {image_id[:6]}"
-    _upload_to_community(png_path, {
-        "id": image_id,
-        "title": title,
-        "prompt": prompt or "",
-        "negative_prompt": negative_prompt or "",
-        "caption": (prompt or "").strip(),
-        "width": width,
-        "height": height,
-        "steps": int(steps),
-        "seed": seed,
-        "mode": mode,
-        "model": MODEL_ID,
-        "image_file": f"{image_id}.png",
-        "created_at": datetime.now(timezone.utc).isoformat(),
-    }, image_id)
+    if not flagged:
+        _upload_to_community(png_path, {
+            "id": image_id,
+            "title": title,
+            "prompt": prompt or "",
+            "negative_prompt": negative_prompt or "",
+            "caption": (prompt or "").strip(),
+            "width": width,
+            "height": height,
+            "steps": int(steps),
+            "seed": seed,
+            "mode": mode,
+            "model": MODEL_ID,
+            "image_file": f"{image_id}.png",
+            "created_at": datetime.now(timezone.utc).isoformat(),
+        }, image_id)
 
     return png_path, png_path, f"seed={seed} | {width}x{height} | {steps} steps | {mode}"