From 28199c93b78f67e9a6bf50f290f1591078f63da8 Mon Sep 17 00:00:00 2001
From: cathy <1574829575@qq.com>
Date: Thu, 4 Apr 2024 23:01:47 +0800
Subject: [PATCH] feat: return combined videos in /api/v1/tasks response

---
 app/services/task.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/app/services/task.py b/app/services/task.py
index 3adf44d..d5a6218 100644
--- a/app/services/task.py
+++ b/app/services/task.py
@@ -127,6 +127,7 @@ def start(task_id, params: VideoParams):
     sm.update_task(task_id, state=const.TASK_STATE_PROCESSING, progress=50)
 
     final_video_paths = []
+    combined_video_paths = []
     video_concat_mode = params.video_concat_mode
     if params.video_count > 1:
         video_concat_mode = VideoConcatMode.random
@@ -162,11 +163,13 @@ def start(task_id, params: VideoParams):
         sm.update_task(task_id, progress=_progress)
 
         final_video_paths.append(final_video_path)
+        combined_video_paths.append(combined_video_path)
 
     logger.success(f"task {task_id} finished, generated {len(final_video_paths)} videos.")
 
     kwargs = {
         "videos": final_video_paths,
+        "combined_videos": combined_video_paths
     }
     sm.update_task(task_id, state=const.TASK_STATE_COMPLETE, progress=100, **kwargs)
     return kwargs