diff --git a/SelectPrompt/systemPrompt.txt b/SelectPrompt/systemPrompt.txt index cc486a7..763147b 100644 --- a/SelectPrompt/systemPrompt.txt +++ b/SelectPrompt/systemPrompt.txt @@ -14,7 +14,7 @@ - `date`: 选题日期 (例如 "2024-06-15") - `logic`: 选定逻辑内容 (描述性文本) - `object`: 选定对象 (例如 "泰宁古城") - - `product`: 选定产品内容 (如果没有单独产品，则为景点本身，例如 "尚书第建筑群+古城游览区") + - `product`: 选定产品内容 (如果没有提供单独产品，则为"None") - `product_logic`: 选定产品的逻辑内容 (描述性文本) - `style`: 选题风格的文件名。**必须是从 Style 文件夹中选择的完整文件名，例如 "攻略风文案提示词.txt"。** - `style_logic`: 选题风格的逻辑内容 (描述性文本) diff --git a/core/__pycache__/contentGen.cpython-312.pyc b/core/__pycache__/contentGen.cpython-312.pyc index 62f1d66..e49a4c4 100644 Binary files a/core/__pycache__/contentGen.cpython-312.pyc and b/core/__pycache__/contentGen.cpython-312.pyc differ diff --git a/core/contentGen.py b/core/contentGen.py index 7c17e00..d8ca58b 100644 --- a/core/contentGen.py +++ b/core/contentGen.py @@ -354,17 +354,139 @@ class ContentGenerator: # 生成时间戳 print(full_response) date_time = datetime.now().strftime("%Y-%m-%d_%H-%M-%S") - full_response = self.split_content(full_response) - # 创建结果文件路径 - result_path = os.path.join(self.output_dir, f"{date_time}.json") - os.makedirs(os.path.dirname(result_path), exist_ok=True) - # 保存结果到文件 - with open(result_path, "w", encoding="utf-8") as f: - json.dump(full_response, f, ensure_ascii=False) + try: + # 解析内容为JSON格式 + parsed_data = self.split_content(full_response) + + # 验证内容格式并修复 + validated_data = self._validate_and_fix_data(parsed_data) + + # 创建结果文件路径 + result_path = os.path.join(self.output_dir, f"{date_time}.json") + os.makedirs(os.path.dirname(result_path), exist_ok=True) + + # 保存结果到文件 + with open(result_path, "w", encoding="utf-8") as f: + json.dump(validated_data, f, ensure_ascii=False, indent=4) + + print(f"结果已保存到: {result_path}") + return result_path - print(f"结果已保存到: {result_path}") - return result_path + except Exception as e: + self.logger.error(f"保存结果到文件时出错: {e}") + # 尝试创建一个简单的备用配置 + fallback_data = [{"main_title": "景点风光", "texts": ["自然美景", "人文体验"], "index": 1}] + + # 保存备用数据 + result_path = os.path.join(self.output_dir, f"{date_time}_fallback.json") + os.makedirs(os.path.dirname(result_path), exist_ok=True) + + with open(result_path, "w", encoding="utf-8") as f: + json.dump(fallback_data, f, ensure_ascii=False, indent=4) + + print(f"出错后已保存备用数据到: {result_path}") + return result_path + + def _validate_and_fix_data(self, data): + """ + 验证并修复数据格式，确保符合预期结构 + + 参数: + data: 需要验证的数据 + + 返回: + 修复后的数据 + """ + fixed_data = [] + + # 如果数据是列表 + if isinstance(data, list): + for i, item in enumerate(data): + # 检查项目是否为字典 + if isinstance(item, dict): + # 确保必需字段存在 + fixed_item = { + "index": item.get("index", i + 1), + "main_title": item.get("main_title", f"景点风光 {i+1}"), + "texts": item.get("texts", ["自然美景", "人文体验"]) + } + + # 确保texts是列表格式 + if not isinstance(fixed_item["texts"], list): + if isinstance(fixed_item["texts"], str): + fixed_item["texts"] = [fixed_item["texts"], "美景体验"] + else: + fixed_item["texts"] = ["自然美景", "人文体验"] + + # 限制texts最多包含两个元素 + if len(fixed_item["texts"]) > 2: + fixed_item["texts"] = fixed_item["texts"][:2] + elif len(fixed_item["texts"]) < 2: + while len(fixed_item["texts"]) < 2: + fixed_item["texts"].append("美景体验") + + fixed_data.append(fixed_item) + + # 如果项目是字符串（可能是错误格式的texts值） + elif isinstance(item, str): + self.logger.warning(f"配置项 {i+1} 是字符串格式，将转换为标准格式") + fixed_item = { + "index": i + 1, + "main_title": f"景点风光 {i+1}", + "texts": [item, "美景体验"] + } + fixed_data.append(fixed_item) + else: + self.logger.warning(f"配置项 {i+1} 格式不支持: {type(item)}，将使用默认值") + fixed_data.append({ + "index": i + 1, + "main_title": f"景点风光 {i+1}", + "texts": ["自然美景", "人文体验"] + }) + + # 如果数据是字典 + elif isinstance(data, dict): + fixed_item = { + "index": data.get("index", 1), + "main_title": data.get("main_title", "景点风光"), + "texts": data.get("texts", ["自然美景", "人文体验"]) + } + + # 确保texts是列表格式 + if not isinstance(fixed_item["texts"], list): + if isinstance(fixed_item["texts"], str): + fixed_item["texts"] = [fixed_item["texts"], "美景体验"] + else: + fixed_item["texts"] = ["自然美景", "人文体验"] + + # 限制texts最多包含两个元素 + if len(fixed_item["texts"]) > 2: + fixed_item["texts"] = fixed_item["texts"][:2] + elif len(fixed_item["texts"]) < 2: + while len(fixed_item["texts"]) < 2: + fixed_item["texts"].append("美景体验") + + fixed_data.append(fixed_item) + + # 如果数据是字符串或其他格式 + else: + self.logger.warning(f"数据格式不支持: {type(data)}，将使用默认值") + fixed_data.append({ + "index": 1, + "main_title": "景点风光", + "texts": ["自然美景", "人文体验"] + }) + + # 确保至少有一个配置项 + if not fixed_data: + fixed_data.append({ + "index": 1, + "main_title": "景点风光", + "texts": ["自然美景", "人文体验"] + }) + + return fixed_data def run(self, info_directory, poster_num, tweet_content): """ @@ -392,31 +514,90 @@ class ContentGenerator: try: result_data = self.split_content(full_response) # This should return the list/dict - # 验证结果数据格式 + # 验证并修复结果数据格式 + fixed_data = [] + + # 如果结果是列表，检查每个项目 if isinstance(result_data, list): for i, item in enumerate(result_data): - if not isinstance(item, dict): - logging.warning(f"配置项 {i+1} 不是字典格式: {item}") - continue - - # 检查并确保必需字段存在 - if 'main_title' not in item: - item['main_title'] = f"景点标题 {i+1}" - logging.warning(f"配置项 {i+1} 缺少 main_title 字段，已添加默认值") - - if 'texts' not in item: - item['texts'] = ["景点特色", "游玩体验"] - logging.warning(f"配置项 {i+1} 缺少 texts 字段，已添加默认值") - - logging.info(f"成功生成并解析海报配置数据，包含 {len(result_data)} 个项目") - else: - logging.warning(f"生成的配置数据不是列表格式: {type(result_data)}") + # 如果项目是字典并且有required_fields，按原样添加或修复 + if isinstance(item, dict): + # 检查并确保必需字段存在 + if 'main_title' not in item: + item['main_title'] = f"景点标题 {i+1}" + logging.warning(f"配置项 {i+1} 缺少 main_title 字段，已添加默认值") + + if 'texts' not in item: + item['texts'] = ["景点特色", "游玩体验"] + logging.warning(f"配置项 {i+1} 缺少 texts 字段，已添加默认值") + + if 'index' not in item: + item['index'] = i + 1 + logging.warning(f"配置项 {i+1} 缺少 index 字段，已添加默认值") + + fixed_data.append(item) + # 如果项目是字符串（可能是错误格式的texts值） + elif isinstance(item, str): + logging.warning(f"配置项 {i+1} 是字符串格式，将转换为标准格式") + fixed_item = { + "index": i + 1, + "main_title": f"景点风光 {i+1}", + "texts": [item, "美景体验"] + } + fixed_data.append(fixed_item) + else: + logging.warning(f"配置项 {i+1} 格式不支持: {type(item)}，将使用默认值") + fixed_data.append({ + "index": i + 1, + "main_title": f"景点风光 {i+1}", + "texts": ["自然美景", "人文体验"] + }) + + # 如果处理后的列表为空（极端情况），则使用默认值 + if not fixed_data: + logging.warning("处理后的配置列表为空，使用默认值") + for i in range(poster_num): + fixed_data.append({ + "index": i + 1, + "main_title": f"景点风光 {i+1}", + "texts": ["自然美景", "人文体验"] + }) + + logging.info(f"成功生成并修复海报配置数据，包含 {len(fixed_data)} 个项目") + return fixed_data + + # 如果结果是单个字典（不常见但可能），将其转换为列表 + elif isinstance(result_data, dict): + logging.warning(f"生成的配置数据是单个字典格式，将转换为列表") + + # 检查并确保必需字段存在 + if 'main_title' not in result_data: + result_data['main_title'] = "景点风光" + + if 'texts' not in result_data: + result_data['texts'] = ["自然美景", "人文体验"] + + if 'index' not in result_data: + result_data['index'] = 1 + + fixed_data = [result_data] + return fixed_data + + # 如果结果是其他格式（如字符串），创建默认配置 + else: + logging.warning(f"生成的配置数据格式不支持: {type(result_data)}，将使用默认值") + for i in range(poster_num): + fixed_data.append({ + "index": i + 1, + "main_title": f"景点风光 {i+1}", + "texts": ["自然美景", "人文体验"] + }) + return fixed_data - return result_data # Return the actual data except Exception as e: logging.exception(f"Failed to parse JSON from AI response in ContentGenerator: {e}\nRaw Response:\n{full_response[:500]}...") # Log error and partial response - # 失败后尝试创建一个默认配置 + # 失败后创建一个默认配置 logging.info("创建默认海报配置数据") default_configs = [] for i in range(poster_num): diff --git a/poster_gen_config.json b/poster_gen_config.json index 2bcc3c8..d786f8b 100644 --- a/poster_gen_config.json +++ b/poster_gen_config.json @@ -1,6 +1,6 @@ { "date": "5月15日, 5月16日, 5月17日, 6月1日", - "num": 30, + "num": 2, "model": "qwenQWQ", "api_url": "http://localhost:8000/v1/", "api_key": "EMPTY", diff --git a/utils/__pycache__/tweet_generator.cpython-312.pyc b/utils/__pycache__/tweet_generator.cpython-312.pyc index cbb8a89..8826b2b 100644 Binary files a/utils/__pycache__/tweet_generator.cpython-312.pyc and b/utils/__pycache__/tweet_generator.cpython-312.pyc differ diff --git a/utils/tweet_generator.py b/utils/tweet_generator.py index 7a25fc5..7697a70 100644 --- a/utils/tweet_generator.py +++ b/utils/tweet_generator.py @@ -57,7 +57,7 @@ class tweetContent: self.variant_index = variant_index try: - self.title, self.content = self.split_content(result) + self.title, self.content = 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}")