optimize code
This commit is contained in:
@@ -213,7 +213,7 @@ Generate a script for a video, depending on the subject of the video.
|
|||||||
|
|
||||||
final_script = ""
|
final_script = ""
|
||||||
logger.info(f"subject: {video_subject}")
|
logger.info(f"subject: {video_subject}")
|
||||||
logger.debug(f"prompt: \n{prompt}")
|
# logger.debug(f"prompt: \n{prompt}")
|
||||||
response = _generate_response(prompt=prompt)
|
response = _generate_response(prompt=prompt)
|
||||||
|
|
||||||
# Return the generated script
|
# Return the generated script
|
||||||
@@ -241,6 +241,10 @@ Generate a script for a video, depending on the subject of the video.
|
|||||||
else:
|
else:
|
||||||
logging.error("gpt returned an empty response")
|
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}")
|
logger.success(f"completed: \n{final_script}")
|
||||||
return final_script
|
return final_script
|
||||||
|
|
||||||
@@ -273,7 +277,7 @@ Please note that you must use English for generating video search terms; Chinese
|
|||||||
""".strip()
|
""".strip()
|
||||||
|
|
||||||
logger.info(f"subject: {video_subject}")
|
logger.info(f"subject: {video_subject}")
|
||||||
logger.debug(f"prompt: \n{prompt}")
|
# logger.debug(f"prompt: \n{prompt}")
|
||||||
response = _generate_response(prompt)
|
response = _generate_response(prompt)
|
||||||
search_terms = []
|
search_terms = []
|
||||||
|
|
||||||
|
|||||||
@@ -78,7 +78,11 @@ def start(task_id, params: VideoParams):
|
|||||||
if sub_maker is None:
|
if sub_maker is None:
|
||||||
sm.state.update_task(task_id, state=const.TASK_STATE_FAILED)
|
sm.state.update_task(task_id, state=const.TASK_STATE_FAILED)
|
||||||
logger.error(
|
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
|
return
|
||||||
|
|
||||||
audio_duration = voice.get_audio_duration(sub_maker)
|
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)
|
sm.state.update_task(task_id, state=const.TASK_STATE_COMPLETE, progress=100, **kwargs)
|
||||||
return kwargs
|
return kwargs
|
||||||
|
|
||||||
|
|
||||||
# def start_test(task_id, params: VideoParams):
|
# def start_test(task_id, params: VideoParams):
|
||||||
# print(f"start task {task_id} \n")
|
# print(f"start task {task_id} \n")
|
||||||
# time.sleep(5)
|
# time.sleep(5)
|
||||||
|
|||||||
@@ -1023,7 +1023,6 @@ def parse_voice_name(name: str):
|
|||||||
|
|
||||||
def is_azure_v2_voice(voice_name: str):
|
def is_azure_v2_voice(voice_name: str):
|
||||||
voice_name = parse_voice_name(voice_name)
|
voice_name = parse_voice_name(voice_name)
|
||||||
print(voice_name)
|
|
||||||
if voice_name.endswith("-V2"):
|
if voice_name.endswith("-V2"):
|
||||||
return voice_name.replace("-V2", "").strip()
|
return voice_name.replace("-V2", "").strip()
|
||||||
return ""
|
return ""
|
||||||
|
|||||||
Reference in New Issue
Block a user