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

support pixabay

Browse Source
This commit is contained in:
harry
2024-05-15 22:39:01 +08:00
parent bdf7af0a12
commit afb4eff3e5
10 changed files with 194 additions and 127 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
app/services/task.py
View File

@@ -44,6 +44,11 @@ def start(task_id, params: VideoParams):
else:
logger.debug(f"video script: \n{video_script}")
if not video_script:
sm.state.update_task(task_id, state=const.TASK_STATE_FAILED)
logger.error("failed to generate video script.")
return
sm.state.update_task(task_id, state=const.TASK_STATE_PROCESSING, progress=10)
logger.info("\n\n## generating video terms")
@@ -60,6 +65,11 @@ def start(task_id, params: VideoParams):
logger.debug(f"video terms: {utils.to_json(video_terms)}")
if not video_terms:
sm.state.update_task(task_id, state=const.TASK_STATE_FAILED)
logger.error("failed to generate video terms.")
return
script_file = path.join(utils.task_dir(task_id), f"script.json")
script_data = {
"script": video_script,
@@ -128,9 +138,10 @@ def start(task_id, params: VideoParams):
print(material_info)
downloaded_videos.append(material_info.url)
else:
logger.info("\n\n## downloading videos")
logger.info(f"\n\n## downloading videos from {params.video_source}")
downloaded_videos = material.download_videos(task_id=task_id,
search_terms=video_terms,
source=params.video_source,
video_aspect=params.video_aspect,
video_contact_mode=params.video_concat_mode,
audio_duration=audio_duration * params.video_count,