jinye_huang/TravelContentCreator
Template
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

增加了结果中tag解析

Browse Source
This commit is contained in:
jinye_huang 2025-05-08 22:39:49 +08:00
parent f59b5ae76b
commit 97be17303f
1 changed files with 11 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
utils/tweet_generator.py
View File

@ -58,7 +58,7 @@ class tweetContent:
self.variant_index = variant_index
try:
self.title, self.content = self.split_content(result)
self.title, self.content, self.tag = self.split_content(result)
self.json_data = self.gen_result_json()
except Exception as e:
logging.error(f"Failed to parse AI result for {article_index}_{variant_index}: {e}")
@ -84,11 +84,14 @@ class tweetContent:
processed_result = result
if "</think>" in result:
processed_result = result.split("</think>", 1)[1] # Take part after </think>
title = processed_result.split("title>", 1)[1].split("</title>", 1)[0]
content = processed_result.split("content>", 1)[1].split("</content>", 1)[0]
# 以json 格式输出
json_data = json.loads(processed_result)
title = json_data.get("title")
content = json_data.get("content")
tag = json_data.get("tag")
# --- End Existing Logic ---
return title.strip(), content.strip()
return title.strip(), content.strip(), tag.strip()
except Exception as e:
logging.warning(f"解析内容时出错: {e}, 返回空字符串")
return "", ""
@ -96,12 +99,14 @@ class tweetContent:
def gen_result_json(self):
json_file = {
"title": self.title,
"content": self.content
"content": self.content,
"tag": self.tag
}
# Add error flag if it exists
if hasattr(self, 'json_data') and self.json_data.get('error'):
json_file['error'] = True
json_file['raw_result'] = self.json_data.get('raw_result')
logging.info(f"Generated JSON: {json_file}")
return json_file
def get_json_data(self):