feat(voices): TTS时长修复 + 提取视频配音 + AI配音标识 (#1656)

Co-authored-by: xiaoxia <dev@xiaoxiajianji.com> Co-committed-by: xiaoxia <dev@xiaoxiajianji.com>
2026-09-03 19:04:35 +08:00
parent e6e4090f3c
commit 6d2d63da7e
8 changed files with 418 additions and 3 deletions
@@ -2,7 +2,9 @@

 from __future__ import annotations

+import json
 import logging
+import subprocess
 import tempfile
 from pathlib import Path
 from typing import Any, Optional
@@ -430,6 +432,8 @@ def save_tts_job_to_library(
    storage_key = f"uploads/voice/tts/{job.id}.{audio_format}"

    tmp_path: Path | None = None
+    audio_duration: float | None = None
+    file_size = 0
    try:
        with tempfile.NamedTemporaryFile(suffix=f".{audio_format}", delete=False) as tmp:
            tmp_path = Path(tmp.name)
@@ -445,6 +449,23 @@ def save_tts_job_to_library(
            )
        file_size = tmp_path.stat().st_size
        storage_service.upload_file(tmp_path, storage_key, content_type=content_type)
+
+        # 从音频文件提取时长（ffprobe），作为 job.duration 的兜底
+        try:
+            proc = subprocess.run(
+                [
+                    "ffprobe", "-v", "quiet", "-print_format", "json",
+                    "-show_format", str(tmp_path),
+                ],
+                capture_output=True, text=True, timeout=10,
+            )
+            if proc.returncode == 0:
+                fmt = json.loads(proc.stdout).get("format", {})
+                dur = float(fmt.get("duration", 0))
+                if dur > 0:
+                    audio_duration = dur
+        except Exception:
+            logger.warning("ffprobe 提取时长失败: job_id=%s", job.id, exc_info=True)
    except HTTPException:
        raise
    except Exception as e:
@@ -482,7 +503,7 @@ def save_tts_job_to_library(
        mime_type=content_type,
        metadata=metadata_,
        file_size=file_size,
-        duration=job.duration or None,
+        duration=job.duration or audio_duration or None,
        status=AssetStatus.READY,
        classification_status=ClassificationStatus.PENDING,  # 音频不参与内容分类，保持 pending 与 ingest 链路一致
        uploaded_by_user_id=user_id,