Update app.py

app.py

@@ -1,291 +1,198 @@
 import sys, subprocess
 
-
-
 # 確保依賴套件安裝
-
 subprocess.run(
-
     [sys.executable, "-m", "pip", "install", "--quiet", "--no-deps", "--upgrade", "huggingface-hub==0.25.2"],
-
     check=False
-
 )
 
-
-
 try:
-
     import huggingface_hub
-
     print("huggingface-hub pinned to:", huggingface_hub.__version__)
-
 except Exception as e:
-
     print("check hub version failed:", e)
 
-
-
 import os, shutil, pathlib, time
 
-
-
 # ===== 1) 環境設定 =====
-
 os.environ.setdefault("TMPDIR", "/tmp")
-
 os.environ.setdefault("HF_HOME", "/tmp/.huggingface")
-
 os.environ.setdefault("HF_HUB_CACHE", os.path.join(os.environ["HF_HOME"], "hub"))
-
 os.environ.setdefault("HF_HUB_ENABLE_HF_TRANSFER", "1")
-
 os.environ.setdefault("GRADIO_ANALYTICS_ENABLED", "false")
 
-
-
 for p in ["/home/user/.cache/pip", "/root/.cache/pip"]:
-
     shutil.rmtree(p, ignore_errors=True)
 
-
-
 pathlib.Path(os.environ["HF_HOME"]).mkdir(parents=True, exist_ok=True)
 
-
-
 # ===== 2) 載入模型 =====
-
 import gradio as gr
-
 import torch
-
 from diffusers import StableDiffusionPipeline
 
-
-
 device = "cuda" if torch.cuda.is_available() else "cpu"
-
 torch_dtype = torch.float16 if device == "cuda" else torch.float32
 
-
-
 pipe = StableDiffusionPipeline.from_pretrained(
-
     "hakurei/waifu-diffusion",
-
     torch_dtype=torch_dtype,
-
     safety_checker=None,
-
     cache_dir=os.environ["HF_HOME"],
-
     use_safetensors=True
-
 ).to(device)
 
-
-
 # 省顯存設定
-
 if torch.cuda.is_available():
-
     for fn in ("enable_attention_slicing", "enable_vae_tiling", "enable_model_cpu_offload"):
-
         try:
-
             getattr(pipe, fn)()
-
         except Exception:
-
             pass
-
 else:
-
     try:
-
         pipe.enable_attention_slicing()
-
     except Exception:
-
         pass
-
     pipe.to("cpu")
 
-
-
 print(f"✅ Model loaded on {device}")
 
-
-
-# ===== 3) 生成函式 (先定義基礎函式) =====
-
+# ===== 3) 生成函式 =====
 def generate_image(prompt: str):
-
     prompt = (prompt or "").strip()
-
     t0 = time.time()
-
     
-
     negative_prompt = (
-
         "low quality, worst quality, blurry, deformed, extra limbs, "
-
         "extra fingers, missing fingers, bad anatomy, bad hands, arms, "
-
         "text, watermark, username"
-
     )
 
-
-
     result = pipe(
-
         prompt=prompt,
-
         negative_prompt=negative_prompt,
-
         width=512, height=512,
-
         num_inference_steps=28,
-
         guidance_scale=7.0
-
     )
-
     img = result.images[0]
-
     del result
-
     
-
     try:
-
         import gc
-
         gc.collect()
-
         if device == "cuda":
-
             torch.cuda.empty_cache()
-
     except Exception:
-
         pass
-
     
-
     elapsed = time.time() - t0
-
     print(f"✅ Image generated in {elapsed:.1f}s")
-
     return img
 
-
-
-# ===== 4) 包裝函式 (在基礎函式之後定義) =====
-
 def generate_with_timeout_warning(prompt):
-
     """包裝生成函式，加入錯誤處理"""
-
     try:
-
         return generate_image(prompt)
-
     except Exception as e:
-
         if "timeout" in str(e).lower():
-
             raise gr.Error("⏱️ 生成超時。")
-
         raise gr.Error(f"生成失敗: {str(e)}")
 
-
-
-# ===== 5) 提示詞模板 =====
-
+# ===== 4) 提示詞模板 =====
 default_prompt = "(masterpiece, best quality:1.3), 1girl, cute, long silver hair, red eyes, smiling, gentle sunlight, (headshot:1.6), (extreme close-up:1.5), (face focus:1.5), face only portrait, shoulders up only, cropped image, highly detailed face and eyes, (sakura trees, japanese town blurred background), soft lighting, intricate details"
 
-
-
 macro_prompt = "(masterpiece, best quality:1.3), 1girl, cute, long silver hair, red eyes, smiling, gentle sunlight, (macro lens portrait:1.4), (shallow depth of field:1.5), focus on eyes, sharp focus face, bokeh background, (sakura trees, japanese town), soft lighting, intricate details, head and shoulders portrait"
 
-
-
-# ===== 6) Gradio 介面 (只定義一次!) =====
-
-# 根據硬體動態調整警告訊息
-
+# ===== 5) 自訂 CSS 和 JavaScript (禁用右鍵和開發者工具) =====
+custom_js = """
+function() {
+    // 禁用右鍵選單
+    document.addEventListener('contextmenu', function(e) {
+        e.preventDefault();
+        return false;
+    }, false);
+    
+    // 禁用 F12、Ctrl+Shift+I、Ctrl+U 等快捷鍵
+    document.addEventListener('keydown', function(e) {
+        if (e.keyCode == 123 || // F12
+            (e.ctrlKey && e.shiftKey && e.keyCode == 73) || // Ctrl+Shift+I
+            (e.ctrlKey && e.shiftKey && e.keyCode == 74) || // Ctrl+Shift+J
+            (e.ctrlKey && e.keyCode == 85)) { // Ctrl+U
+            e.preventDefault();
+            return false;
+        }
+    });
+    
+    // 禁用文字選取和複製
+    document.addEventListener('selectstart', function(e) {
+        e.preventDefault();
+        return false;
+    });
+    
+    document.addEventListener('copy', function(e) {
+        e.preventDefault();
+        return false;
+    });
+}
+"""
+
+custom_css = """
+/* 禁用使用者選取文字 */
+* {
+    -webkit-user-select: none;
+    -moz-user-select: none;
+    -ms-user-select: none;
+    user-select: none;
+}
+
+/* 保留輸入框可以選取文字 */
+input, textarea {
+    -webkit-user-select: text !important;
+    -moz-user-select: text !important;
+    -ms-user-select: text !important;
+    user-select: text !important;
+}
+"""
+
+# ===== 6) Gradio 介面 =====
 if device == "cpu":
-
     title = "Waifu Diffusion 1:1 動漫風大頭照生成器 (CPU 模式)"
-
     description = """
-
-    **提醒**: 當前使用 CPU 運算，單張圖片需要 **5-15 分鐘**
+    **提醒**: 當前使用雲端 CPU 運算,單張圖片需要 **5-15 分鐘**
     
-    點擊下方 Examples 可載入預設模板，記得按下【Submit】開始生成
-
+    點擊下方 Examples 可載入預設模板,記得按下【Submit】開始生成
     """
-
 else:
-
     title = "Waifu Diffusion 1:1 動漫風大頭照生成器 (GPU 模式)"
-
-    description = "點擊下方 Examples 可載入預設模板，記得按下【Submit】開始生成。預計 10-30 秒完成。"
-
-
+    description = "點擊下方 Examples 可載入預設模板,記得按下【Submit】開始生成。預計 10-30 秒完成。"
 
 demo = gr.Interface(
-
     fn=generate_with_timeout_warning,  # 使用包裝後的函式
-
     inputs=gr.Textbox(
-
         label="輸入提示詞 (Prompt)", 
-
         value=default_prompt,
-
         lines=5,
-
         placeholder="在此輸入英文提示詞..."
-
     ),
-
     outputs=gr.Image(label="生成結果 (Image)", type="pil", format="png"),
-
     title=title,
-
     description=description,
-
     examples=[
-
         [default_prompt], 
-
         [macro_prompt],   
-
     ],
-
-    cache_examples=False
-
+    cache_examples=False,
+    css=custom_css,
+    js=custom_js
 ).queue()
 
-
-
 # ===== 7) 啟動服務 =====
-
 if __name__ == "__main__":
-
     demo.launch(
-
         server_name="0.0.0.0", 
-
         server_port=7860, 
-
         share=False, 
-
         max_threads=1
-
     )