coosld/MoneyPrinterTurbo
1
0
Fork 0
mirror of https://github.com/harry0703/MoneyPrinterTurbo.git synced 2026-03-06 23:07:22 +08:00
Code Issues 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
Show Stats 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)
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