feat(task): 优化脚本时间戳更新逻辑并添加音频路径- 修改 update_script_timestamps 函数，支持同时更新视频和音频路径

- 在任务处理流程中生成 tts_clip_result 字典，用于更新脚本中的音频路径
-优化脚本列表的更新逻辑，确保所有时间戳都计算持续时间

app/services/task.py

@@ -248,7 +248,8 @@ def start_subclip(task_id: str, params: VideoClipParams, subclip_path_videos: di
     clip_result = clip_video.clip_video(params.video_origin_path, tts_results)
     subclip_path_videos.update(clip_result)
     # 更新 list_script 中的时间戳
-    list_script = update_script.update_script_timestamps(list_script, clip_result)
+    tts_clip_result = {tts_result['timestamp']: tts_result['audio_file'] for tts_result in tts_results}
+    list_script = update_script.update_script_timestamps(list_script, clip_result, tts_clip_result)
     subclip_videos = [x for x in subclip_path_videos.values()]
 
     sm.state.update_task(task_id, state=const.TASK_STATE_PROCESSING, progress=60)
@@ -257,7 +258,6 @@ def start_subclip(task_id: str, params: VideoClipParams, subclip_path_videos: di
     4. 合并音频和字幕
     """
     logger.info("\n\n## 4. 合并音频和字幕")
-
     subtitle_files = [
         tts_result["subtitle_file"] for tts_result in tts_results
     ]

app/services/update_script.py

@@ -63,13 +63,14 @@ def calculate_duration(timestamp: str) -> float:
         return 0.0
 
 
-def update_script_timestamps(script_list: List[Dict[str, Any]], tts_result: Dict[str, str]) -> List[Dict[str, Any]]:
+def update_script_timestamps(script_list: List[Dict[str, Any]], video_result: Dict[str, str], audio_result: Dict[str, str] = None) -> List[Dict[str, Any]]:
     """
-    根据 tts_res 中的视频文件更新 list_script 中的时间戳，并添加持续时间
+    根据 video_result 中的视频文件更新 script_list 中的时间戳，添加持续时间，并根据 audio_result 添加音频路径
     
     Args:
         script_list: 原始脚本列表
-        tts_result: TTS 结果字典，键为原时间戳，值为视频文件路径
+        video_result: 视频结果字典，键为原时间戳，值为视频文件路径
+        audio_result: 音频结果字典，键为原时间戳，值为音频文件路径
     
     Returns:
         更新后的脚本列表
@@ -77,26 +78,36 @@ def update_script_timestamps(script_list: List[Dict[str, Any]], tts_result: Dict
     # 创建副本，避免修改原始数据
     updated_script = []
 
-    # 建立原始时间戳到新时间戳的映射
+    # 建立原始时间戳到视频路径和新时间戳的映射
     timestamp_mapping = {}
-    for orig_timestamp, video_path in tts_result.items():
+    for orig_timestamp, video_path in video_result.items():
         new_timestamp = extract_timestamp_from_video_path(video_path)
         if new_timestamp:
-            timestamp_mapping[orig_timestamp] = new_timestamp
+            timestamp_mapping[orig_timestamp] = {
+                'new_timestamp': new_timestamp,
+                'video_path': video_path
+            }
 
     # 更新脚本中的时间戳
     for item in script_list:
         item_copy = item.copy()
-        if item_copy.get('timestamp') in timestamp_mapping:
+        orig_timestamp = item_copy.get('timestamp', '')
-            # 更新时间戳
-            new_timestamp = timestamp_mapping[item_copy['timestamp']]
-            item_copy['timestamp'] = new_timestamp
 
-            # 计算并添加持续时间
+        # 初始化音频路径为空字符串
-            item_copy['duration'] = calculate_duration(new_timestamp)
+        item_copy['audio'] = ""
-        elif 'timestamp' in item_copy:
+
-            # 对于未更新的时间戳，也计算并添加持续时间
+        # 如果提供了音频结果字典且时间戳存在于音频结果中，直接使用对应的音频路径
+        if audio_result and orig_timestamp in audio_result:
+            item_copy['audio'] = audio_result[orig_timestamp]
+
+        if orig_timestamp in timestamp_mapping:
+            # 更新时间戳
+            item_copy['timestamp'] = timestamp_mapping[orig_timestamp]['new_timestamp']
+            # 计算持续时间
             item_copy['duration'] = calculate_duration(item_copy['timestamp'])
+        elif orig_timestamp:
+            # 对于未更新的时间戳，也计算并添加持续时间
+            item_copy['duration'] = calculate_duration(orig_timestamp)
 
         updated_script.append(item_copy)
 
@@ -122,13 +133,19 @@ if __name__ == '__main__':
                      'narration': '但想见庆帝，哪有那么容易？范闲艺高人胆大，竟然选择了最激进的方式——闯宫！', 'OST': 0},
         {'picture': '画面切换到范闲蒙面闯入皇宫，被侍卫包围的场景。', 'timestamp': '00:06:00-00:06:03',
          'narration': '抓刺客', 'OST': 1}]
-    tts_res = {
+    video_res = {
         '00:00:00-00:01:15': '/Users/apple/Desktop/home/NarratoAI/storage/temp/clip_video/0ac14d474144b54d614c26a5c87cffe7/vid-00-00-00-00-00-26.mp4',
         '00:01:15-00:04:40': '/Users/apple/Desktop/home/NarratoAI/storage/temp/clip_video/0ac14d474144b54d614c26a5c87cffe7/vid-00-01-15-00-01-29.mp4',
         '00:04:58-00:05:45': '/Users/apple/Desktop/home/NarratoAI/storage/temp/clip_video/0ac14d474144b54d614c26a5c87cffe7/vid-00-04-58-00-05-20.mp4',
         '00:05:45-00:06:00': '/Users/apple/Desktop/home/NarratoAI/storage/temp/clip_video/0ac14d474144b54d614c26a5c87cffe7/vid-00-05-45-00-05-53.mp4'}
+    audio_res = {
+        '00:00:00-00:01:15': '/Users/apple/Desktop/home/NarratoAI/storage/tasks/qyn2-2-demo/audio_00_00_00-00_01_15.mp3',
+        '00:01:15-00:04:40': '/Users/apple/Desktop/home/NarratoAI/storage/tasks/qyn2-2-demo/audio_00_01_15-00_04_40.mp3',
+        '00:04:58-00:05:45': '/Users/apple/Desktop/home/NarratoAI/storage/tasks/qyn2-2-demo/audio_00_04_58-00_05_45.mp3',
+        '00:05:45-00:06:00': '/Users/apple/Desktop/home/NarratoAI/storage/tasks/qyn2-2-demo/audio_00_05_45-00_06_00.mp3'}
     
     # 更新并打印结果
-    updated_list_script = update_script_timestamps(list_script, tts_res)
+    updated_list_script = update_script_timestamps(list_script, video_res, audio_res)
     for item in updated_list_script:
-        print(f"Picture: {item['picture'][:20]}... | Timestamp: {item['timestamp']} | Duration: {item['duration']} 秒")
+        print(
+            f"Picture: {item['picture'][:20]}... | Timestamp: {item['timestamp']} | Duration: {item['duration']} 秒 | Audio: {item['audio']}")