hyper-animator-codex 0.5.0 → 0.7.0

package/skills/hyper-animator-codex/scripts/generate_subtitles.mjs

@@ -0,0 +1,181 @@
+#!/usr/bin/env node
+import { mkdir, readFile, writeFile } from "node:fs/promises";
+import { join } from "node:path";
+
+const TRAILING_PUNCTUATION = /[，。！？；：,.!?;:、]+$/u;
+
+function requireValue(args, index, flag) {
+  const value = args[index + 1];
+  if (!value || value.startsWith("--")) {
+    throw new Error(`${flag} requires a value`);
+  }
+  return value;
+}
+
+function parseArgs(args) {
+  const parsed = {
+    narration: undefined,
+    output_dir: join(process.cwd(), "hyper-animator-output", "subtitles"),
+    style: "default",
+  };
+
+  for (let index = 0; index < args.length; index += 1) {
+    const arg = args[index];
+    if (arg === "--narration") {
+      parsed.narration = requireValue(args, index, arg);
+      index += 1;
+    } else if (arg === "--output-dir") {
+      parsed.output_dir = requireValue(args, index, arg);
+      index += 1;
+    } else if (arg === "--style") {
+      parsed.style = requireValue(args, index, arg);
+      index += 1;
+    } else if (arg === "--help" || arg === "-h") {
+      parsed.help = true;
+    } else {
+      throw new Error(`Unknown option: ${arg}`);
+    }
+  }
+
+  return parsed;
+}
+
+function validateNarration(narration) {
+  if (!narration || typeof narration !== "object" || !Array.isArray(narration.scenes)) {
+    throw new Error("narration JSON must include scenes[]");
+  }
+
+  narration.scenes.forEach((scene, index) => {
+    if (typeof scene?.narration !== "string" || !scene.narration.trim()) {
+      throw new Error(`scenes[${index}].narration is required`);
+    }
+
+    if (scene.duration_estimate !== undefined && (!Number.isFinite(Number(scene.duration_estimate)) || Number(scene.duration_estimate) <= 0)) {
+      throw new Error(`scenes[${index}].duration_estimate must be a positive number`);
+    }
+  });
+}
+
+function cleanCueText(text) {
+  return text.trim().replace(TRAILING_PUNCTUATION, "").trim();
+}
+
+function splitNarration(text) {
+  return text
+    .split(/[\r\n]+|(?<=[，。！？；：,.!?;:、])/u)
+    .map(cleanCueText)
+    .filter(Boolean);
+}
+
+function roundTime(value) {
+  return Number(value.toFixed(3));
+}
+
+function buildCues(narration) {
+  const cues = [];
+  let cursor = 0;
+
+  narration.scenes.forEach((scene, sceneIndex) => {
+    const parts = splitNarration(scene.narration);
+    const duration = Number(scene.duration_estimate) || Math.max(parts.length * 2, 2);
+    const cueDuration = duration / Math.max(parts.length, 1);
+
+    parts.forEach((text) => {
+      const start = cursor;
+      const end = cursor + cueDuration;
+      cues.push({
+        index: cues.length + 1,
+        scene: scene.scene ?? sceneIndex + 1,
+        start: roundTime(start),
+        end: roundTime(end),
+        text,
+      });
+      cursor = end;
+    });
+  });
+
+  return cues;
+}
+
+function formatSrtTime(seconds) {
+  const totalMs = Math.round(seconds * 1000);
+  const ms = totalMs % 1000;
+  const totalSeconds = Math.floor(totalMs / 1000);
+  const sec = totalSeconds % 60;
+  const totalMinutes = Math.floor(totalSeconds / 60);
+  const min = totalMinutes % 60;
+  const hour = Math.floor(totalMinutes / 60);
+
+  return `${String(hour).padStart(2, "0")}:${String(min).padStart(2, "0")}:${String(sec).padStart(2, "0")},${String(ms).padStart(3, "0")}`;
+}
+
+function renderSrt(cues) {
+  return `${cues.map((cue) => [
+    cue.index,
+    `${formatSrtTime(cue.start)} --> ${formatSrtTime(cue.end)}`,
+    cue.text,
+  ].join("\n")).join("\n\n")}\n`;
+}
+
+async function generateSubtitles(options) {
+  if (!options.narration) {
+    throw new Error("--narration is required");
+  }
+
+  const narration = JSON.parse(await readFile(options.narration, "utf8"));
+  validateNarration(narration);
+
+  const cues = buildCues(narration);
+  await mkdir(options.output_dir, { recursive: true });
+
+  const jsonPath = join(options.output_dir, "subtitles.json");
+  const srtPath = join(options.output_dir, "subtitles.srt");
+  const payload = {
+    ok: true,
+    style: options.style,
+    source: options.narration,
+    cues,
+  };
+
+  await writeFile(jsonPath, `${JSON.stringify(payload, null, 2)}\n`, "utf8");
+  await writeFile(srtPath, renderSrt(cues), "utf8");
+
+  return {
+    ...payload,
+    json_path: jsonPath,
+    srt_path: srtPath,
+  };
+}
+
+function printHelp() {
+  console.log(`Usage:
+  node scripts/generate_subtitles.mjs --narration <file> [options]
+
+Options:
+  --narration <file>    Narration JSON file
+  --output-dir <dir>    Output directory, default hyper-animator-output/subtitles
+  --style <name>        Subtitle style label for downstream HTML/render steps
+`);
+}
+
+async function main() {
+  try {
+    const options = parseArgs(process.argv.slice(2));
+    if (options.help) {
+      printHelp();
+      return;
+    }
+
+    console.log(JSON.stringify(await generateSubtitles(options), null, 2));
+  } catch (error) {
+    console.log(JSON.stringify({
+      ok: false,
+      error: {
+        message: error.message,
+      },
+    }, null, 2));
+    process.exitCode = 1;
+  }
+}
+
+main();