|
@@ -75,15 +75,18 @@ class MatchArticlesTask(object):
|
|
|
async def find_material(task_tuple, mysql_client):
|
|
|
task_id, title, text = task_tuple
|
|
|
# 先用视频标题作为query, 后续可逐步优化
|
|
|
- question_dict = question_fission(title)
|
|
|
- material_dict = {}
|
|
|
- for question_key in question_dict:
|
|
|
- question = question_dict[question_key]
|
|
|
- material = generate_text(question)
|
|
|
- cleand_question = question.replace("\n", "").replace("*", "").replace("#", "").replace(":", "").replace('"', "").replace("'", "")
|
|
|
- material_dict[cleand_question] = material
|
|
|
+ try:
|
|
|
+ question_dict = question_fission(title)
|
|
|
+ material_dict = {}
|
|
|
+ for question_key in question_dict:
|
|
|
+ question = question_dict[question_key]
|
|
|
+ material = generate_text(question)
|
|
|
+ cleand_question = question.replace("\n", "").replace("*", "").replace("#", "").replace(":", "").replace('"', "").replace("'", "")
|
|
|
+ material_dict[cleand_question] = material
|
|
|
|
|
|
- material_result = json.dumps(material_dict, ensure_ascii=False)
|
|
|
+ material_result = json.dumps(material_dict, ensure_ascii=False)
|
|
|
+ except:
|
|
|
+ material_result=json.dumps({title: text}, ensure_ascii=False)
|
|
|
update_sql = f"""
|
|
|
UPDATE {db_config}
|
|
|
SET materials = %s, status_code = %s
|
|
@@ -105,7 +108,10 @@ class MatchArticlesTask(object):
|
|
|
|
|
|
async def ai_generate_text(task_tuple, mysql_client):
|
|
|
task_id, video_title, materials = task_tuple
|
|
|
- ai_title, ai_text = summary_articles(materials)
|
|
|
+ try:
|
|
|
+ ai_title, ai_text = summary_articles(materials)
|
|
|
+ except:
|
|
|
+ ai_title, ai_text = video_title, "文章生成失败"
|
|
|
imgs = get_img_list(video_title)
|
|
|
update_sql = f"""
|
|
|
UPDATE {db_config}
|