hzhang/MoneyPrinterTurbo
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: return combined videos in /api/v1/tasks response

Browse Source
This commit is contained in:
cathy
2024-04-04 23:01:47 +08:00
parent 87a837a345
commit 28199c93b7
1 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
app/services/task.py
View File

@@ -127,6 +127,7 @@ def start(task_id, params: VideoParams):
sm.update_task(task_id, state=const.TASK_STATE_PROCESSING, progress=50) sm.update_task(task_id, state=const.TASK_STATE_PROCESSING, progress=50)
final_video_paths = [] final_video_paths = []
combined_video_paths = []
video_concat_mode = params.video_concat_mode video_concat_mode = params.video_concat_mode
if params.video_count > 1: if params.video_count > 1:
video_concat_mode = VideoConcatMode.random video_concat_mode = VideoConcatMode.random
@@ -162,11 +163,13 @@ def start(task_id, params: VideoParams):
sm.update_task(task_id, progress=_progress) sm.update_task(task_id, progress=_progress)
final_video_paths.append(final_video_path) 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.") logger.success(f"task {task_id} finished, generated {len(final_video_paths)} videos.")
kwargs = { kwargs = {
"videos": final_video_paths, "videos": final_video_paths,
"combined_videos": combined_video_paths
} }
sm.update_task(task_id, state=const.TASK_STATE_COMPLETE, progress=100, **kwargs) sm.update_task(task_id, state=const.TASK_STATE_COMPLETE, progress=100, **kwargs)
return kwargs return kwargs