Update app.py

app.py

@@ -6,12 +6,19 @@ from PIL import Image
 import time
 from datetime import datetime
 from optimum.intel import OVStableDiffusionPipeline
+import re
+from functools import lru_cache
 
 # Đảm bảo tính tái tạo
 torch.manual_seed(0)
 np.random.seed(0)
 random.seed(0)
 
+# Tối ưu hóa bộ nhớ và tốc độ
+torch.set_num_threads(1)  # Giảm xung đột thread
+torch.backends.mkldnn.enabled = True  # Kích hoạt MKL-DNN
+torch.backends.openmp.enabled = True
+
 print("🔧 Loading OpenVINO pipeline: HARRY07979/sd-v1-5-openvino")
 
 try:
@@ -35,60 +42,64 @@ except Exception as e:
         print(f"Error loading local model: {e}")
         raise RuntimeError("Failed to load the pipeline.")
 
+# Tối ưu hóa pipeline
 pipeline.to("cpu")
+pipeline.model.half()  # Sử dụng half precision để tăng tốc
+pipeline.compile()  # Biên dịch mô hình nếu hỗ trợ
 
-def detect_nsfw(prompt: str):
-    nsfw_keywords = {
-        "high": [
-            "nude", "naked", "sex", "porn", "xxx", "fuck", "dick", "cock", "pussy", "vagina", "penis",
-            "boobs", "tits", "breasts", "bra", "panties", "underwear", "lingerie", "orgasm", "cum",
-            "blowjob", "handjob", "masturbate", "rape", "gangbang", "incest", "hentai", "lewd",
-            "erotic", "kinky", "bondage", "bdsm", "squirt", "creampie", "threesome", "orgy", "yaoi",
-            "yuri", "futanari", "cunnilingus", "fellatio", "anal", "paizuri", "bukkake", "guro",
-            "vore", "tentacle", "netorare", "cuckold", "exhibitionism", "voyeurism", "poop", "pee",
-            "poo", "shit", "piss", "scat", "diarrhea", "vomit", "gore", "blood", "murder",
-            "torture", "suicide", "decapitation", "mutilation", "drugs", "cocaine", "heroin",
-            "lsd", "ecstasy", "vlxx"
-        ],
-        "medium": [
-            "bikini", "swimwear", "sexy", "succubus", "leather", "latex", "stockings", "miniskirt",
-            "cleavage", "thighs", "ass", "butt", "skirt", "dress", "topless", "wet", "moaning",
-            "spread", "legs apart", "tight", "revealing", "provocative", "suggestive", "flirty"
-        ],
-        "low": [
-            "girl", "woman", "man", "boy", "beach", "bedroom", "kiss", "hug", "dance", "pose",
-            "model", "photo", "art", "illustration", "anime", "cosplay", "costume", "fashion"
-        ]
-    }
+# Tối ưu hóa từ điển NSFW - sử dụng set để tìm kiếm nhanh hơn
+NSFW_HIGH = {
+    "nude", "naked", "sex", "porn", "xxx", "fuck", "dick", "cock", "pussy", "vagina", "penis",
+    "boobs", "tits", "breasts", "bra", "panties", "underwear", "lingerie", "orgasm", "cum",
+    "blowjob", "handjob", "masturbate", "rape", "gangbang", "incest", "hentai", "lewd",
+    "erotic", "kinky", "bondage", "bdsm", "squirt", "creampie", "threesome", "orgy", "yaoi",
+    "yuri", "futanari", "cunnilingus", "fellatio", "anal", "paizuri", "bukkake", "guro",
+    "vore", "tentacle", "netorare", "cuckold", "exhibitionism", "voyeurism", "poop", "pee",
+    "poo", "shit", "piss", "scat", "diarrhea", "vomit", "gore", "blood", "murder",
+    "torture", "suicide", "decapitation", "mutilation", "drugs", "cocaine", "heroin",
+    "lsd", "ecstasy", "vlxx"
+}
 
-    prompt_lower = prompt.lower()
+NSFW_MEDIUM = {
+    "bikini", "swimwear", "sexy", "succubus", "leather", "latex", "stockings", "miniskirt",
+    "cleavage", "thighs", "ass", "butt", "skirt", "dress", "topless", "wet", "moaning",
+    "spread", "legs apart", "tight", "revealing", "provocative", "suggestive", "flirty"
+}
 
-    for word in nsfw_keywords["high"]:
-        if word in prompt_lower:
-            return True
+NSFW_PHRASES = [
+    "spreading legs", "removing bra", "pulling panties", "sucking dick", "licking pussy",
+    "penetrating", "fucking scene", "hard cock", "wet pussy", "big tits", "exposed breasts",
+    "nipples visible", "ass spread", "thigh gap", "camel toe", "pussy lips", "cum on face",
+    "blowjob scene", "anal sex", "titty fuck", "gang rape", "group sex", "public sex",
+    "hidden camera", "peeing girl", "pooping girl", "covered in blood", "cutting flesh",
+    "snorting cocaine", "injecting heroin", "hallucinating", "smoking weed"
+]
 
-    nsfw_phrases = [
-        "spreading legs", "removing bra", "pulling panties", "sucking dick", "licking pussy",
-        "penetrating", "fucking scene", "hard cock", "wet pussy", "big tits", "exposed breasts",
-        "nipples visible", "ass spread", "thigh gap", "camel toe", "pussy lips", "cum on face",
-        "blowjob scene", "anal sex", "titty fuck", "gang rape", "group sex", "public sex",
-        "hidden camera", "peeing girl", "pooping girl", "covered in blood", "cutting flesh",
-        "snorting cocaine", "injecting heroin", "hallucinating", "smoking weed"
-    ]
-    for phrase in nsfw_phrases:
-        if phrase in prompt_lower:
-            return True
+SENSITIVE_CONTEXT = {
+    "spread", "removing", "pulling", "sucking", "licking", "penetrating",
+    "fucking", "hard", "wet", "exposed", "visible", "ass", "tight", "revealing"
+}
 
-    medium_matches = [word for word in nsfw_keywords["medium"] if word in prompt_lower]
-    if medium_matches:
-        sensitive_context = any(
-            word in prompt_lower
-            for word in ["spread", "removing", "pulling", "sucking", "licking", "penetrating",
-                         "fucking", "hard", "wet", "exposed", "visible", "ass", "tight", "revealing"]
-        )
-        if sensitive_context:
+# Tối ưu hóa hàm NSFW detection
+@lru_cache(maxsize=1024)
+def detect_nsfw(prompt: str):
+    prompt_lower = prompt.lower()
+    
+    # Kiểm tra từ khóa cao
+    words = set(re.findall(r'\b\w+\b', prompt_lower))
+    if words & NSFW_HIGH:
+        return True
+    
+    # Kiểm tra cụm từ NSFW
+    for phrase in NSFW_PHRASES:
+        if phrase in prompt_lower:
             return True
-
+    
+    # Kiểm tra từ khóa trung bình + ngữ cảnh nhạy cảm
+    medium_matches = words & NSFW_MEDIUM
+    if medium_matches and (words & SENSITIVE_CONTEXT):
+        return True
+    
     return False
 
 def infer(
@@ -108,28 +119,44 @@ def infer(
     print(f"⏰ Run at: {current_time}")
     print(f"📝 Prompt: {prompt}")
     print(f"🚫 Negative Prompt: {negative_prompt or '[None]'}")
+    
+    # Tối ưu hóa kiểm tra NSFW
     if detect_nsfw(prompt):
         raise gr.Error("⚠️ Prompt contains NSFW content. Please use a safe prompt.")
+    
     if randomize_seed:
         seed = random.randint(0, np.iinfo(np.int32).max)
         print(f"🎲 Random Seed generated: {seed}")
     else:
         print(f"🔢 Using fixed Seed: {seed}")
+    
+    # Tối ưu hóa kích thước ảnh
     width = (width // 8) * 8
     height = (height // 8) * 8
     print(f"🖼️ Image Size: {width}x{height}")
     print(f"🎯 Guidance Scale: {guidance_scale}")
     print(f"📈 Inference Steps: {num_inference_steps}")
+    
+    # Tối ưu hóa generator
     generator = torch.Generator("cpu").manual_seed(seed)
-    result = pipeline(
-        prompt=prompt,
-        negative_prompt=negative_prompt,
-        guidance_scale=guidance_scale,
-        num_inference_steps=num_inference_steps,
-        width=width,
-        height=height,
-        generator=generator,
-    )
+    
+    # Sử dụng inference_mode để tăng tốc
+    with torch.inference_mode():
+        result = pipeline(
+            prompt=prompt,
+            negative_prompt=negative_prompt,
+            guidance_scale=guidance_scale,
+            num_inference_steps=num_inference_steps,
+            width=width,
+            height=height,
+            generator=generator,
+            # Tối ưu hóa batch
+            batch_size=1,
+            # Tăng chất lượng
+            eta=0.0,  # Giảm nhiễu
+            use_karras_sigmas=True,  # Cải thiện chất lượng
+        )
+    
     image = result.images[0]
     return image, seed
 
@@ -138,11 +165,26 @@ css = """
     margin: 0 auto;
     max-width: 640px;
 }
+.donate-button {
+    background-color: #FFDD00 !important;
+    color: #000000 !important;
+}
+"""
+
+# Hàm JavaScript để mở trang donate khi nhấn nút
+donate_js = """
+function() {
+    window.open('https://buymeacoffee.com/harry07?status=1', '_blank');
+}
 """
 
 with gr.Blocks(css=css) as demo:
     with gr.Column(elem_id="col-container"):
-        gr.Markdown("## Stable Diffusion v1.5 DEMO")
+        with gr.Row():
+            gr.Markdown("## Stable Diffusion v1.5 DEMO")
+            # Thêm nút Donate với biểu tượng ☕
+            donate_btn = gr.Button("☕ Donate", elem_classes="donate-button")
+        
         with gr.Row():
             prompt = gr.Text(label="Prompt", placeholder="Enter your prompt", show_label=False)
             run_button = gr.Button("Generate", variant="primary")
@@ -180,6 +222,9 @@ with gr.Blocks(css=css) as demo:
         ],
         outputs=[result, seed],
     )
+    
+    # Gán sự kiện click cho nút Donate
+    donate_btn.click(None, js=donate_js)
 
 if __name__ == "__main__":
     demo.launch()