optimize code

This commit is contained in:
harry
2024-04-22 16:25:13 +08:00
parent 6a5c4e9e73
commit c7c7b4847e
3 changed files with 11 additions and 5 deletions

View File

@@ -213,7 +213,7 @@ Generate a script for a video, depending on the subject of the video.
final_script = ""
logger.info(f"subject: {video_subject}")
logger.debug(f"prompt: \n{prompt}")
# logger.debug(f"prompt: \n{prompt}")
response = _generate_response(prompt=prompt)
# Return the generated script
@@ -241,6 +241,10 @@ Generate a script for a video, depending on the subject of the video.
else:
logging.error("gpt returned an empty response")
# g4f may return an error message
if final_script and "当日额度已消耗完" in final_script:
raise ValueError(final_script)
logger.success(f"completed: \n{final_script}")
return final_script
@@ -273,7 +277,7 @@ Please note that you must use English for generating video search terms; Chinese
""".strip()
logger.info(f"subject: {video_subject}")
logger.debug(f"prompt: \n{prompt}")
# logger.debug(f"prompt: \n{prompt}")
response = _generate_response(prompt)
search_terms = []

View File

@@ -78,7 +78,11 @@ def start(task_id, params: VideoParams):
if sub_maker is None:
sm.state.update_task(task_id, state=const.TASK_STATE_FAILED)
logger.error(
"failed to generate audio, maybe the network is not available. if you are in China, please use a VPN.")
"""failed to generate audio:
1. check if the language of the voice matches the language of the video script.
2. check if the network is available. If you are in China, it is recommended to use a VPN and enable the global traffic mode.
""".strip()
)
return
audio_duration = voice.get_audio_duration(sub_maker)
@@ -174,7 +178,6 @@ def start(task_id, params: VideoParams):
sm.state.update_task(task_id, state=const.TASK_STATE_COMPLETE, progress=100, **kwargs)
return kwargs
# def start_test(task_id, params: VideoParams):
# print(f"start task {task_id} \n")
# time.sleep(5)

View File

@@ -1023,7 +1023,6 @@ def parse_voice_name(name: str):
def is_azure_v2_voice(voice_name: str):
voice_name = parse_voice_name(voice_name)
print(voice_name)
if voice_name.endswith("-V2"):
return voice_name.replace("-V2", "").strip()
return ""