Flatten skill category directory structure

audio-creation/asmr-ambient/scripts/render_mix_preview.py

@@ -0,0 +1,221 @@
+#!/usr/bin/env python3
+"""启动助眠音频混合预览服务器。
+
+用法：
+  python3 render_mix_preview.py <preview_data.json> [--port PORT]
+
+流程：
+  1. 读取 JSON 数据文件（包含各轨道路径、参数配置）
+  2. 注入 HTML 模板，启动本地 HTTP 服务器
+  3. 在浏览器中打开预览页面
+  4. 用户提交反馈后，写入 feedback.md + mix_settings.json，服务器自动退出
+"""
+
+import argparse
+import base64
+import json
+import mimetypes
+import os
+import sys
+import threading
+import webbrowser
+from http.server import HTTPServer, BaseHTTPRequestHandler
+from pathlib import Path
+from socketserver import ThreadingMixIn
+
+SKILL_DIR = Path(__file__).resolve().parent.parent
+TEMPLATE_DIR = SKILL_DIR / "html"
+
+
+class ThreadingHTTPServer(ThreadingMixIn, HTTPServer):
+    daemon_threads = True
+
+
+class PreviewHandler(BaseHTTPRequestHandler):
+    html_content: str = ""
+    feedback_path: Path = Path("feedback.md")
+    settings_path: Path = Path("mix_settings.json")
+    server_ref: HTTPServer | None = None
+
+    def do_GET(self):
+        if self.path.startswith("/audio/"):
+            self._serve_audio()
+            return
+        self.send_response(200)
+        self.send_header("Content-Type", "text/html; charset=utf-8")
+        self.end_headers()
+        self.wfile.write(self.html_content.encode("utf-8"))
+
+    def _serve_audio(self):
+        encoded = self.path[len("/audio/"):]
+        try:
+            audio_path = Path(base64.urlsafe_b64decode(encoded).decode("utf-8"))
+        except Exception:
+            self.send_error(400, "invalid audio path")
+            return
+        if not audio_path.exists():
+            self.send_error(404, "audio not found")
+            return
+        mime = mimetypes.guess_type(str(audio_path))[0] or "audio/mpeg"
+        file_size = audio_path.stat().st_size
+
+        try:
+            self._send_audio(audio_path, mime, file_size)
+        except BrokenPipeError:
+            pass
+
+    def _send_audio(self, audio_path, mime, file_size):
+        range_header = self.headers.get("Range")
+        if range_header:
+            # Parse Range: bytes=start-end
+            range_spec = range_header.replace("bytes=", "")
+            parts = range_spec.split("-")
+            start = int(parts[0]) if parts[0] else 0
+            end = int(parts[1]) if parts[1] else file_size - 1
+            end = min(end, file_size - 1)
+            length = end - start + 1
+
+            self.send_response(206)
+            self.send_header("Content-Type", mime)
+            self.send_header("Content-Range", f"bytes {start}-{end}/{file_size}")
+            self.send_header("Content-Length", str(length))
+            self.send_header("Accept-Ranges", "bytes")
+            self.send_header("Cache-Control", "max-age=3600")
+            self.end_headers()
+
+            with open(audio_path, "rb") as f:
+                f.seek(start)
+                self.wfile.write(f.read(length))
+        else:
+            self.send_response(200)
+            self.send_header("Content-Type", mime)
+            self.send_header("Content-Length", str(file_size))
+            self.send_header("Accept-Ranges", "bytes")
+            self.send_header("Cache-Control", "max-age=3600")
+            self.end_headers()
+
+            with open(audio_path, "rb") as f:
+                while True:
+                    chunk = f.read(65536)
+                    if not chunk:
+                        break
+                    self.wfile.write(chunk)
+
+    def do_POST(self):
+        if self.path == "/feedback":
+            self._handle_feedback()
+        else:
+            self.send_error(404)
+
+    def _handle_feedback(self):
+        length = int(self.headers.get("Content-Length", 0))
+        body = self.rfile.read(length).decode("utf-8")
+        data = json.loads(body)
+        feedback = data.get("feedback", "").strip()
+        settings = data.get("settings", {})
+
+        self.send_response(200)
+        self.send_header("Content-Type", "application/json")
+        self.end_headers()
+
+        if feedback:
+            self.feedback_path.write_text(feedback, encoding="utf-8")
+            if settings:
+                self.settings_path.write_text(
+                    json.dumps(settings, ensure_ascii=False, indent=2),
+                    encoding="utf-8",
+                )
+            self.wfile.write(json.dumps({"ok": True}).encode())
+            print(f"FEEDBACK_FILE={self.feedback_path}")
+            if settings:
+                print(f"SETTINGS_FILE={self.settings_path}")
+            threading.Timer(0.5, self._shutdown).start()
+        else:
+            self.wfile.write(json.dumps({"ok": False, "msg": "empty"}).encode())
+
+    def _shutdown(self):
+        if self.server_ref:
+            self.server_ref.shutdown()
+
+    def log_message(self, format, *args):
+        pass
+
+
+def to_audio_url(abs_path: str) -> str:
+    """将绝对路径转为 /audio/<base64> URL"""
+    encoded = base64.urlsafe_b64encode(abs_path.encode()).decode()
+    return f"/audio/{encoded}"
+
+
+def prepare_data(raw_data: dict) -> dict:
+    """将各轨道的绝对路径转为可访问的 /audio/ URL。
+
+    数据结构：
+    - voice.path → voice.audioUrl
+    - nature.path → nature.audioUrl
+    - music.path → music.audioUrl (可选)
+    - mixed.path → mixed.audioUrl
+    """
+    if raw_data.get("voice") and raw_data["voice"].get("path"):
+        raw_data["voice"]["audioUrl"] = to_audio_url(raw_data["voice"]["path"])
+
+    if raw_data.get("nature") and raw_data["nature"].get("path"):
+        raw_data["nature"]["audioUrl"] = to_audio_url(raw_data["nature"]["path"])
+
+    if raw_data.get("music") and raw_data["music"].get("path"):
+        raw_data["music"]["audioUrl"] = to_audio_url(raw_data["music"]["path"])
+
+    if raw_data.get("mixed") and raw_data["mixed"].get("path"):
+        raw_data["mixed"]["audioUrl"] = to_audio_url(raw_data["mixed"]["path"])
+
+    return raw_data
+
+
+def main():
+    parser = argparse.ArgumentParser(description="启动助眠音频混合预览服务器")
+    parser.add_argument("data_file", help="JSON 数据文件路径")
+    parser.add_argument("--port", type=int, default=0, help="端口号，默认自动分配")
+    parsed = parser.parse_args()
+
+    data_path = Path(parsed.data_file).resolve()
+    if not data_path.exists():
+        print(f"ERROR: 数据文件不存在: {data_path}")
+        sys.exit(1)
+
+    raw_data = json.loads(data_path.read_text(encoding="utf-8"))
+    preview_data = prepare_data(raw_data)
+
+    template_path = TEMPLATE_DIR / "preview-mix_feedback.html"
+    if not template_path.exists():
+        print(f"ERROR: 模板不存在: {template_path}")
+        sys.exit(1)
+
+    template = template_path.read_text(encoding="utf-8")
+    injected = template.replace(
+        "const PREVIEW_TYPE = null;  /* __PREVIEW_TYPE__ */",
+        'const PREVIEW_TYPE = "mix_feedback";',
+    ).replace(
+        "const PREVIEW_DATA = null;  /* __PREVIEW_DATA__ */",
+        f"const PREVIEW_DATA = {json.dumps(json.dumps(preview_data, ensure_ascii=False))};",
+    )
+
+    feedback_dir = data_path.parent
+    server = ThreadingHTTPServer(("127.0.0.1", parsed.port), PreviewHandler)
+    port = server.server_address[1]
+
+    PreviewHandler.html_content = injected
+    PreviewHandler.feedback_path = feedback_dir / "feedback.md"
+    PreviewHandler.settings_path = feedback_dir / "mix_settings.json"
+    PreviewHandler.server_ref = server
+
+    url = f"http://127.0.0.1:{port}"
+    print(f"PREVIEW_URL={url}")
+    print("等待用户提交...")
+
+    webbrowser.open(url)
+    server.serve_forever()
+    print("服务器已停止")
+
+
+if __name__ == "__main__":
+    main()

audio-creation/asmr-ambient/scripts/validate_nature.py

@@ -0,0 +1,89 @@
+#!/usr/bin/env python3
+"""
+验证 nature/ 目录中的素材文件，输出类别识别结果和元数据。
+用法：python validate_nature.py <nature_dir>
+"""
+import os, subprocess, sys, json
+
+CATEGORY_KEYWORDS = {
+    "rain": ["rain", "雨"],
+    "thunder": ["thunder", "雷"],
+    "fireplace": ["fireplace", "fire", "壁炉", "篝火"],
+    "ocean": ["ocean", "wave", "sea", "海浪", "海"],
+    "river": ["river", "stream", "creek", "brook", "waterfall", "溪流", "河", "瀑布"],
+    "forest": ["forest", "森林"],
+    "bird": ["bird", "鸟"],
+    "cricket": ["cricket", "cicada", "蟋蟀", "蝉"],
+    "wind": ["wind", "风"],
+    "snow": ["snow", "雪"],
+    "cafe": ["cafe", "coffee", "咖啡"],
+    "clock": ["clock", "tick", "钟"],
+    "keyboard": ["keyboard", "typing", "键盘"],
+    "vinyl": ["vinyl", "record", "黑胶"],
+    "train": ["train", "rail", "火车", "铁轨"],
+    "city": ["city", "traffic", "street", "城市", "街道"],
+    "subway": ["subway", "metro", "地铁"],
+    "whitenoise": ["whitenoise", "white_noise", "白噪音"],
+    "underwater": ["underwater", "bubble", "水下"],
+    "space": ["space", "cosmic", "太空"],
+}
+
+AUDIO_EXTS = {".mp3", ".wav", ".m4a", ".flac", ".ogg", ".aac"}
+
+
+def get_duration(path):
+    try:
+        r = subprocess.run(
+            ["ffprobe", "-v", "quiet", "-show_entries", "format=duration", "-of", "csv=p=0", path],
+            capture_output=True, text=True,
+        )
+        return float(r.stdout.strip())
+    except Exception:
+        return 0.0
+
+
+def detect_category(filename):
+    name = filename.lower()
+    for cat, keywords in CATEGORY_KEYWORDS.items():
+        for kw in keywords:
+            if kw in name:
+                return cat
+    return None
+
+
+def main():
+    nature_dir = sys.argv[1] if len(sys.argv) > 1 else "."
+
+    files = sorted([
+        f for f in os.listdir(nature_dir)
+        if os.path.splitext(f)[1].lower() in AUDIO_EXTS
+    ])
+
+    if not files:
+        print(json.dumps({"error": "no audio files found", "dir": nature_dir}))
+        sys.exit(1)
+
+    results = []
+    for f in files:
+        path = os.path.join(nature_dir, f)
+        cat = detect_category(f)
+        dur = get_duration(path)
+        results.append({
+            "filename": f,
+            "category": cat,
+            "duration_s": round(dur, 1),
+            "size_mb": round(os.path.getsize(path) / 1024 / 1024, 1),
+            "needs_category": cat is None,
+        })
+
+    summary = {
+        "total": len(results),
+        "categorized": sum(1 for r in results if r["category"]),
+        "uncategorized": sum(1 for r in results if not r["category"]),
+        "files": results,
+    }
+    print(json.dumps(summary, ensure_ascii=False, indent=2))
+
+
+if __name__ == "__main__":
+    main()