feat: 色情识别结果可视化与日志；超长路径修复；旋转开关、明暗5档等

internal/api/nsfw_controller.go

@@ -0,0 +1,75 @@
+// 色情识别相关的两个接口：查记录、看图片预览。
+// 前端「色情识别」页面用它展示每次扫描判出的色情图片（检测分 + 路径 + 预览），
+// 方便用户核对识别结果准不准。只读数据库 + 读本地图片文件，不干别的活。
+package api
+
+import (
+	"net/http"
+	"os"
+	"strings"
+
+	"github.com/gin-gonic/gin"
+
+	"kenaito-media-picker/internal/store"
+)
+
+// NsfwController 色情识别相关接口的"窗口"。
+type NsfwController struct {
+	store *store.Store // 数据库（只读）
+}
+
+// NewNsfwController 创建色情识别控制器，注入数据库。
+func NewNsfwController(st *store.Store) *NsfwController {
+	return &NsfwController{store: st}
+}
+
+// nsfwView 列表里展示的一条记录：路径用"整理后位置"，没有就用源路径。
+type nsfwView struct {
+	Id        int64   `json:"id"`        // 主键
+	FileName  string  `json:"fileName"`  // 文件名
+	Score     float64 `json:"score"`     // 检测分 0~1
+	Threshold float64 `json:"threshold"` // 判定阈值
+	Path      string  `json:"path"`      // 预览/打开用的完整路径
+	CreatedAt string  `json:"createdAt"` // 检测时间
+}
+
+// List 处理 GET /api/nsfw/list：查全部色情识别记录（按检测分从高到低）。
+func (c *NsfwController) List(ctx *gin.Context) {
+	list, err := c.store.ListNsfw()
+	if err != nil {
+		Error(ctx, http.StatusInternalServerError, "查询色情识别记录失败："+err.Error())
+		return
+	}
+	views := make([]nsfwView, 0, len(list))
+	for _, r := range list {
+		path := r.MovedPath
+		if path == "" {
+			path = r.FilePath
+		}
+		views = append(views, nsfwView{
+			Id:        r.Id,
+			FileName:  r.FileName,
+			Score:     r.Score,
+			Threshold: r.Threshold,
+			Path:      path,
+			CreatedAt: r.CreatedAt.Format("2006-01-02 15:04:05"),
+		})
+	}
+	Success(ctx, map[string]interface{}{"list": views, "total": len(views)})
+}
+
+// Preview 处理 GET /api/nsfw/preview?path=xxx：把指定图片文件返回给浏览器做预览。
+// 这是本地单机工具，直接按路径读文件；文件不存在/已被移动时返回 404。
+func (c *NsfwController) Preview(ctx *gin.Context) {
+	path := strings.TrimSpace(ctx.Query("path"))
+	if path == "" {
+		Error(ctx, http.StatusBadRequest, "缺少 path 参数")
+		return
+	}
+	if _, err := os.Stat(path); err != nil {
+		Error(ctx, http.StatusNotFound, "图片不存在或已被移动: "+path)
+		return
+	}
+	// 交给 gin 读文件并带上正确的 Content-Type / 断点续传
+	ctx.File(path)
+}

internal/api/router.go

@@ -33,6 +33,11 @@ func NewRouter(st *store.Store) *gin.Engine {
 		apiGroup.POST("/media/list", mc.List)
 		apiGroup.POST("/media/stats", mc.Stats)
 
+		// 色情识别（查看检测分 + 图片预览）
+		nc := NewNsfwController(st)
+		apiGroup.GET("/nsfw/list", nc.List)
+		apiGroup.GET("/nsfw/preview", nc.Preview)
+
 		// 扫描整理（开始 / 中止 / 查进度 / 实时进度）
 		sc := NewScanController(service.NewScanService(st))
 		apiGroup.POST("/scan/start", sc.Start)

internal/config/config.go

@@ -17,7 +17,7 @@ import (
 //   - Dsn：数据库文件叫什么（默认 media-info.db，放在程序目录下）
 //   - AutoOpen：启动后要不要自动打开浏览器
 //   - FFmpegPath：ffmpeg 程序路径（用于把带 EXIF 旋转标记的图片转正）；留空自动在 PATH 里找
-//   - RotateImages：整理时是否自动转正带 EXIF 旋转标记的图片（默认 true）
+//   - RotateImages：是否处理图片 EXIF 旋转（默认 true）；false 则完全不处理，按存储方向/尺寸归类
 //   - Workers：探测/MIME/移动阶段的并发 worker 数（默认 4；设 0 表示用满全部 CPU 核数）
 //   - NSFWEnabled：是否开启色情图片识别（本地 python + NudeNet，默认 false）
 //   - NSFWPython：python 可执行文件路径；留空自动在 PATH 里找

internal/nsfw/detector.go

@@ -216,6 +216,11 @@ func (d *Detector) Stop() {
 	}
 }
 
+// Threshold 返回当前使用的判定阈值（0~1）。给上层记录结果用。
+func (d *Detector) Threshold() float64 {
+	return d.threshold
+}
+
 // resolvePython 在配置值 / PATH 里找可用的 python 解释器。
 // Windows 上常见 python / python3 / py；Linux/mac 常见 python3 / python。
 func resolvePython(configured string) string {

internal/nsfw/server.py

@@ -2,14 +2,29 @@
 """NudeNet 本地检测服务：由 Go 侧启动，常驻提供 /health 和 /detect。
 
 只依赖 Python 标准库 + nudenet 包，模型在进程里只加载一次。
-收到 /detect 就把图片路径丢给 NudeDetector，返回结果 JSON。
+收到 /detect 就把图片喂给 NudeDetector，返回结果 JSON。
+
+Windows 超长路径说明：OpenCV 的 imread 走老式 C API，路径超过 260 字符就
+打不开（报 can't open/read file）。所以这里改用 Python 的 open() 读字节，
+再把字节直接交给 NudeNet（v3 的 detect 支持 bytes）——从内存解码，彻底
+绕开路径限制，同时兼容老版 v2（只收路径时落到临时短路径文件再试）。
 """
 import json
+import os
 import sys
+import tempfile
 from http.server import BaseHTTPRequestHandler, HTTPServer
 
 from nudenet import NudeDetector
 
+# 只报 OpenCV 致命错误，别让 imread 的 WARN 刷屏（也避免误导排查）
+try:
+    import cv2
+
+    cv2.setLogLevel(8)  # LOG_LEVEL_ERROR
+except Exception:
+    pass
+
 # 只有这些"暴露部位"才算色情；其它如肚子/腋窝/脚踝暴露不判。
 SENSITIVE = {
     "FEMALE_GENITALIA_EXPOSED",
@@ -28,6 +43,51 @@ def get_detector():
     return _detector
 
 
+def _longpath(path):
+    """Windows 下给绝对路径加 \\\\?\\ 前缀，让 Python 原生 open 也能处理超长路径。"""
+    if os.name != "nt":
+        return path
+    if path.startswith("\\\\?\\"):
+        return path
+    if path.startswith("\\\\"):
+        # UNC 网络路径：\\server\share → \\?\UNC\server\share
+        return "\\\\?\\UNC\\" + path.lstrip("\\")
+    if os.path.isabs(path):
+        return "\\\\?\\" + os.path.normpath(path)
+    return path
+
+
+def _read_bytes(path):
+    """用 Python 读文件字节（不经过 OpenCV），读不了就抛异常由上层兜底。"""
+    lp = _longpath(path)
+    if not os.path.exists(lp):
+        raise OSError("文件不存在: %s" % path)
+    with open(lp, "rb") as f:
+        data = f.read()
+    if not data:
+        raise OSError("空文件: %s" % path)
+    return data
+
+
+def _detect_bytes(det, data, path):
+    """把图片字节喂给 NudeNet。v3 直接吃 bytes；老版 v2 只收路径时落到临时短路径重试。"""
+    try:
+        return det.detect(data)
+    except (TypeError, AttributeError):
+        tmp = None
+        try:
+            fd, tmp = tempfile.mkstemp(suffix=os.path.splitext(path)[1] or ".jpg")
+            with os.fdopen(fd, "wb") as f:
+                f.write(data)
+            return det.detect(tmp)
+        finally:
+            if tmp:
+                try:
+                    os.unlink(tmp)
+                except OSError:
+                    pass
+
+
 class Handler(BaseHTTPRequestHandler):
     def _send(self, obj, status=200):
         data = json.dumps(obj, ensure_ascii=False).encode("utf-8")
@@ -51,7 +111,10 @@ class Handler(BaseHTTPRequestHandler):
             body = json.loads(self.rfile.read(length) or b"{}")
             path = body.get("path", "")
             threshold = float(body.get("threshold", 0.5))
-            detections = get_detector().detect(path)
+            if not path:
+                return self._send({"ok": False, "error": "缺少图片路径"})
+            data = _read_bytes(path)
+            detections = _detect_bytes(get_detector(), data, path)
         except Exception as exc:  # noqa: BLE001 - 服务端任何异常都要回给 Go，别让进程崩
             return self._send({"ok": False, "error": str(exc)})
 
@@ -73,4 +136,4 @@ class Handler(BaseHTTPRequestHandler):
 
 if __name__ == "__main__":
     port = int(sys.argv[1]) if len(sys.argv) > 1 else 0
-    HTTPServer(("127.0.0.1", port), Handler).serve_forever()
+    HTTPServer(("127.0.0.1", port), Handler).serve_forever()

internal/organize/organize.go

@@ -56,30 +56,33 @@ const (
 //
 // 这样设计是为了避免"边分类边搬"导致目录反复开关、文件搬来搬去，效率更高。
 type Organizer struct {
-	pickerDir   string              // 整理后的文件都放到这个根目录下
-	ffprobePath string              // ffprobe 程序路径；为空表示没有可用解码器
-	ffmpegPath  string              // ffmpeg 程序路径；为空表示没有转正 EXIF 旋转图片的能力
-	store       *store.Store        // 数据库；为 nil 表示不记录媒体信息
-	mover       *mover.Mover        // 负责实际搬文件的家伙
-	ctx         context.Context     // 中止信号；ctx 被取消时探测/归类立即停下
-	nsfw        *nsfw.Detector      // 色情检测器；为 nil 表示没启用，图片不检测
-	buckets     map[string][]string // 目标目录 → 里面该放哪些文件（只记源路径）
-	order       []string            // 目标目录的先后顺序（保证输出稳定、不来回跳）
-	onProgress  func(phase string, done, total int)
-	pending     []*store.MediaInfo // 待写入数据库的媒体记录（攒一批再批量写）
+	pickerDir    string              // 整理后的文件都放到这个根目录下
+	ffprobePath  string              // ffprobe 程序路径；为空表示没有可用解码器
+	ffmpegPath   string              // ffmpeg 程序路径；为空表示没有转正 EXIF 旋转图片的能力
+	store        *store.Store        // 数据库；为 nil 表示不记录媒体信息
+	mover        *mover.Mover        // 负责实际搬文件的家伙
+	ctx          context.Context     // 中止信号；ctx 被取消时探测/归类立即停下
+	nsfw         *nsfw.Detector      // 色情检测器；为 nil 表示没启用，图片不检测
+	rotateImages bool                // 是否处理 EXIF 旋转（转正/按显示方向归类）；false 则完全不动，按存储方向归类
+	buckets      map[string][]string // 目标目录 → 里面该放哪些文件（只记源路径）
+	order        []string            // 目标目录的先后顺序（保证输出稳定、不来回跳）
+	onProgress   func(phase string, done, total int)
+	pending      []*store.MediaInfo // 待写入数据库的媒体记录（攒一批再批量写）
+	pendingNsfw  []store.NsfwRecord // 待写入数据库的色情识别记录（攒一批再批量写）
 }
 
 // New 创建整理器。store 可以为 nil——数据库打不开时，整理照常进行，只是不记入库信息。
 // ctx 可以为 nil——不提供中止能力，任务跑到完为止。
 func New(pickerDir, ffprobePath, ffmpegPath string, st *store.Store, ctx context.Context) *Organizer {
 	return &Organizer{
-		pickerDir:   pickerDir,
-		ffprobePath: ffprobePath,
-		ffmpegPath:  ffmpegPath,
-		store:       st,
-		mover:       &mover.Mover{},
-		ctx:         ctx,
-		buckets:     make(map[string][]string),
+		pickerDir:    pickerDir,
+		ffprobePath:  ffprobePath,
+		ffmpegPath:   ffmpegPath,
+		store:        st,
+		mover:        &mover.Mover{},
+		ctx:          ctx,
+		rotateImages: true, // 默认处理 EXIF 旋转；关掉用 SetRotateImages(false)
+		buckets:      make(map[string][]string),
 	}
 }
 
@@ -96,6 +99,13 @@ func (o *Organizer) SetNSFWDetector(d *nsfw.Detector) {
 	o.nsfw = d
 }
 
+// SetRotateImages 设置是否处理图片 EXIF 旋转（默认 true）。
+//   - true：带旋转标记的图片做转正处理（有 ffmpeg 就烧进像素，没有则按显示方向归类）；
+//   - false：完全不动，图片按文件里存储的方向/尺寸归类，不做任何旋转相关处理。
+func (o *Organizer) SetRotateImages(b bool) {
+	o.rotateImages = b
+}
+
 // report 上报进度。total 为 0 表示这一步没活干，就不用报了。
 func (o *Organizer) report(phase string, done, total int) {
 	if o.onProgress != nil && total > 0 {
@@ -337,6 +347,10 @@ func (o *Organizer) probeImage(f scanner.File) imageProbeResult {
 		} else {
 			r.nsfwScore = score
 			r.nsfw = hit
+			// 命中的图片把"文件名 + 检测分"打进运行日志，方便用户盯着看是否判对
+			if hit {
+				fmt.Printf("=== 色情命中: %s, 检测分=%.2f (阈值=%.2f)\n", f.Path, score, o.nsfw.Threshold())
+			}
 		}
 	}
 	return r
@@ -353,6 +367,9 @@ func (o *Organizer) probeImage(f scanner.File) imageProbeResult {
 //
 // 只有方向 5/6/7/8（90°/270° 旋转）才需要宽高对调；2/3/4 是镜像或 180°，横竖不变。
 func (o *Organizer) correctImageRotation(f scanner.File, res *probe.ProbeResult) {
+	if !o.rotateImages {
+		return // 配置关闭：完全不处理旋转，按存储方向/尺寸归类
+	}
 	ori := probe.ReadExifOrientation(f.Path)
 	if ori == 0 {
 		return // 没有旋转标记，不用管
@@ -380,6 +397,7 @@ func (o *Organizer) correctImageRotation(f scanner.File, res *probe.ProbeResult)
 func (o *Organizer) ProcessImages(images []scanner.File) {
 	// 统计结果，最后打一行总结给用户看
 	success, fail, ignore, dup := 0, 0, 0, 0
+	nsfwCount := 0 // 命中的色情图片数
 	total := len(images)
 	results := runParallel(o.ctx, images, o.probeImage, func(done int) {
 		o.report(PhaseImages, done, total)
@@ -415,6 +433,14 @@ func (o *Organizer) ProcessImages(images []scanner.File) {
 			o.record(f, "image", r.res, r.md5Str, r.headMd5, r.tailMd5, r.size)
 			if r.nsfw {
 				o.add(f, classify.ImageDirSensitive(f.Ext, r.res.Width, r.res.Height, r.darkScore, r.lightScore))
+				// 攒一条色情识别记录，最后连同"搬去哪了"一起写进数据库
+				o.pendingNsfw = append(o.pendingNsfw, store.NsfwRecord{
+					FilePath:  f.Path,
+					FileName:  filepath.Base(f.Path),
+					Score:     r.nsfwScore,
+					Threshold: o.nsfw.Threshold(),
+				})
+				nsfwCount++
 			} else {
 				o.add(f, classify.ImageDir(f.Ext, r.res.Width, r.res.Height, r.darkScore, r.lightScore))
 			}
@@ -422,6 +448,9 @@ func (o *Organizer) ProcessImages(images []scanner.File) {
 		}
 	}
 	fmt.Printf("=== 图片处理完毕，成功: %d, 读取异常: %d, 忽略: %d, 重复: %d \n\n", success, fail, ignore, dup)
+	if o.nsfw != nil && nsfwCount > 0 {
+		fmt.Printf("=== 色情识别: 共命中 %d 张，已归入「图片_敏感_*」目录，可在【色情识别】页面查看并人工复核\n", nsfwCount)
+	}
 }
 
 // ============================ 视频/音频整理 ============================
@@ -544,7 +573,27 @@ func (o *Organizer) Flush() {
 
 	// 全局并行搬完：不管文件分散在几个目录，所有文件同时开搬。
 	// onProgress 收到的是全局累计已搬数，正好和总数对得上。
-	o.mover.MoveAll(jobs, func(done int) {
+	// moved 是"源路径 → 新位置"的对照表，色情记录要用它补上整理后的实际路径。
+	moved := o.mover.MoveAll(jobs, func(done int) {
 		o.report(PhaseMove, done, len(jobs))
 	})
+
+	// 把色情识别记录补上"整理后位置"（预览就用它），再批量写进数据库
+	if len(o.pendingNsfw) > 0 {
+		records := make([]*store.NsfwRecord, 0, len(o.pendingNsfw))
+		for i := range o.pendingNsfw {
+			if target, ok := moved[o.pendingNsfw[i].FilePath]; ok {
+				o.pendingNsfw[i].MovedPath = target
+			} else {
+				o.pendingNsfw[i].MovedPath = o.pendingNsfw[i].FilePath // 没搬走就用源路径
+			}
+			records = append(records, &o.pendingNsfw[i])
+		}
+		if o.store != nil {
+			if err := o.store.SaveNsfwBatch(records); err != nil {
+				fmt.Printf("=== 保存色情识别记录失败: %v\n", err)
+			}
+		}
+		o.pendingNsfw = nil
+	}
 }

internal/service/scan.go

@@ -505,11 +505,18 @@ func (s *ScanService) runScan(ctx context.Context, rootDir, destDir string) erro
 			nsfwDet = d
 			defer nsfwDet.Stop()
 			fmt.Println("=== 色情识别已启用(NudeNet)")
+			// 清掉上一次的色情识别记录，保证「色情识别」页面显示的是本次扫描的结果
+			if s.store != nil {
+				if err := s.store.ClearNsfw(); err != nil {
+					fmt.Printf("=== 清空旧色情识别记录失败: %v\n", err)
+				}
+			}
 		}
 	}
 
 	org := organize.New(pickerDir, ffprobePath, ffmpegPath, s.store, ctx)
 	org.SetNSFWDetector(nsfwDet)
+	org.SetRotateImages(config.GetAppConfig().RotateImages)
 
 	// 整个"整理"分两个阶段，界面上共用一条进度条：
 	//   归类阶段：图片、视频/音频分两批处理。进度 = 已处理文件数 / 全部待整理文件数

internal/store/store.go

@@ -63,6 +63,18 @@ type AppConfig struct {
 	Value string `xorm:"text 'value'"`         // 配置值
 }
 
+// NsfwRecord 一条"色情图片识别"的记录：哪张图、检测分多少、判了没有、现在在哪。
+// 前端「色情识别」页面就是查这张表来展示每张图片的检测结果，方便人工复核准确性。
+type NsfwRecord struct {
+	Id        int64     `xorm:"pk autoincr 'id'" json:"id"`               // 自增主键
+	FilePath  string    `xorm:"text 'file_path'" json:"filePath"`         // 检测时的源路径（在哪发现的）
+	FileName  string    `xorm:"varchar(255) 'file_name'" json:"fileName"` // 文件名（含扩展名）
+	Score     float64   `xorm:"real 'score'" json:"score"`                // 命中的敏感部位最高分 0~1
+	Threshold float64   `xorm:"real 'threshold'" json:"threshold"`        // 判定阈值（>= 它才算色情）
+	MovedPath string    `xorm:"text 'moved_path'" json:"movedPath"`       // 整理后实际位置（预览用）；没搬走时同源路径
+	CreatedAt time.Time `xorm:"created 'created_at'" json:"createdAt"`    // 检测时间
+}
+
 // Open 打开（或第一次时创建）数据库，并保证表结构齐全。
 func Open(dbPath string) (*Store, error) {
 	e, err := xorm.NewEngine("sqlite3", dbPath)
@@ -72,8 +84,8 @@ func Open(dbPath string) (*Store, error) {
 	e.ShowSQL(false)
 	// SQLite 同一时刻只允许一个写连接，这里限制成 1 个，避免并发写冲突
 	e.SetMaxOpenConns(1)
-	// 自动创建/补全两张表（媒体记录表、配置表）
-	if err := e.Sync(new(MediaInfo), new(AppConfig)); err != nil {
+	// 自动创建/补全三张表（媒体记录表、配置表、色情识别记录表）
+	if err := e.Sync(new(MediaInfo), new(AppConfig), new(NsfwRecord)); err != nil {
 		_ = e.Close()
 		return nil, err
 	}
@@ -123,6 +135,37 @@ func (s *Store) GetConfig(key string) string {
 	return cfg.Value
 }
 
+// ============================ 色情识别记录 ============================
+
+// SaveNsfwBatch 批量保存色情识别记录（一次扫描攒一批，最后统一写入）。
+func (s *Store) SaveNsfwBatch(records []*NsfwRecord) error {
+	if s == nil || s.engine == nil || len(records) == 0 {
+		return nil
+	}
+	_, err := s.engine.Insert(records)
+	return err
+}
+
+// ClearNsfw 清空色情识别记录。每次开启色情识别的扫描开始时调用，
+// 保证「色情识别」页面看到的是最近一次扫描的结果。
+func (s *Store) ClearNsfw() error {
+	if s == nil || s.engine == nil {
+		return nil
+	}
+	_, err := s.engine.Where("1 = 1").Delete(new(NsfwRecord))
+	return err
+}
+
+// ListNsfw 查全部色情识别记录，按检测分从高到低排（方便先看最"实锤"的）。
+func (s *Store) ListNsfw() ([]NsfwRecord, error) {
+	if s == nil || s.engine == nil {
+		return nil, nil
+	}
+	list := make([]NsfwRecord, 0)
+	err := s.engine.Desc("score").Find(&list)
+	return list, err
+}
+
 // ============================ 媒体记录的新增/更新 ============================
 
 // SaveMedia 保存/更新一条媒体记录：按 MD5 找，有就更新，没有就新增。