新增：NSFW 开关

Author: c1921

model_manager.py

@@ -227,21 +227,26 @@ def get_model_display_info(self, model_path: str) -> dict:
                 "preview_url": None,
                 "description": "未找到模型信息",
                 "baseModel": "未知",
+                "nsfw": False,
+                "nsfwLevel": 0,
             }
 
         info = model_info.get("info", {})
         model_data = info.get("model", {})
-        preview_url = info.get("images", [{}])[0].get("url") if info.get("images") else None
+        preview_image = info.get("images", [{}])[0] if info.get("images") else {}
+        preview_url = preview_image.get("url")
 
         # 添加本地图片路径
         local_preview = info.get("local_preview")
 
         return {
             "name": model_data.get("name", Path(model_path).name),
             "type": model_data.get("type", "未知"),
-            "preview_url": local_preview or preview_url,  # 优先使用本地路径
+            "preview_url": local_preview or preview_url,
             "baseModel": info.get("baseModel", "未知"),
-            "url": f"https://civitai.com/models/{info['modelId']}?modelVersionId={info['id']}"
+            "url": f"https://civitai.com/models/{info['modelId']}?modelVersionId={info['id']}",
+            "nsfw": model_data.get("nsfw", False),
+            "nsfwLevel": preview_image.get("nsfwLevel", 0),
         }
 
     def get_all_models_info(self) -> list:

requirements.txt

@@ -1,8 +1,6 @@
 fastapi
 uvicorn
 requests
-python-multipart
-pydantic
 aiohttp
 aiofiles
 pyinstaller

templates/index.html

@@ -44,6 +44,10 @@
     <nav class="navbar navbar-expand-lg">
         <div class="container-fluid">
             <span class="navbar-brand">Stable Diffusion 模型管理器</span>
+            <div class="form-check form-switch me-3">
+                <input class="form-check-input" type="checkbox" role="switch" id="nsfwToggle">
+                <label class="form-check-label" for="nsfwToggle">显示 NSFW</label>
+            </div>
             <button class="btn btn-outline ms-auto" id="darkModeToggle">
                 <i class="bi bi-sun-fill" id="lightIcon"></i>
                 <i class="bi bi-moon-stars-fill d-none" id="darkIcon"></i>
@@ -148,6 +152,16 @@
             type: new Set()
         };
 
+        // NSFW 显示状态
+        let showNSFW = false;
+        const nsfwToggle = document.getElementById('nsfwToggle');
+        
+        // 监听 NSFW 开关
+        nsfwToggle.addEventListener('change', (e) => {
+            showNSFW = e.target.checked;
+            displayModels(allModels); // 重新显示模型列表
+        });
+
         // 检查系统暗黑模式
         function updateDarkModeUI(isDark) {
             document.documentElement.setAttribute('data-bs-theme', isDark ? 'dark' : 'light');
@@ -276,8 +290,11 @@
 
         // 显示模型列表
         function displayModels(models) {
+            // 根据 NSFW 开关过滤模型
+            const filteredModels = showNSFW ? models : models.filter(model => !model.nsfw && model.nsfwLevel < 3);
+            
             // 按类型分组模型
-            const modelsByType = models.reduce((acc, model) => {
+            const modelsByType = filteredModels.reduce((acc, model) => {
                 const type = model.type || '未知';
                 if (!acc[type]) {
                     acc[type] = [];
@@ -349,12 +366,15 @@
                         <div class="card h-100">
                             ${imageHtml}
                             <div class="card-body">
-                                <h5 class="card-title">${model.name}</h5>
+                                <div class="d-flex justify-content-between align-items-start mb-2">
+                                    <h5 class="card-title mb-0">${model.name}</h5>
+                                    ${model.nsfw ? '<span class="badge text-bg-danger">NSFW</span>' : ''}
+                                </div>
                                 <p class="card-text">
                                     <small class="d-block">类型: ${model.type}</small>
                                     <small class="d-block">基础模型: ${model.baseModel || '未知'}</small>
                                 </p>
-                                ${model.url ? `<a href="${model.url}" class="btn btn-primary btn-sm" target="_blank">查看详情</a>` : ''}
+                                ${model.url ? `<a href="${model.url}" class="btn btn-outline-primary btn-sm" target="_blank">查看详情</a>` : ''}
                             </div>
                         </div>
                     `;
@@ -501,6 +521,11 @@ <h5 class="card-title">${model.name}</h5>
                 );
             }
 
+            // 应用 NSFW 过滤
+            if (!showNSFW) {
+                filteredModels = filteredModels.filter(model => !model.nsfw && model.nsfwLevel < 3);
+            }
+
             displayModels(filteredModels);
         }