# inputType.py # from services.templateData import templateData # 引入1-4套皮肤数据 from services.style.style1to4data import sectorStyle1to4Data # 引入5套皮肤数据 from services.style.style5data import sectorStyle5Data class inputType: def __init__(self): self.del_map = [ "删除", "去掉", "移除", "去除", "不要", "删掉", "不需要", "没有" ] self.del_group_map = [ "不要显示广告", "去掉所有广告", "纯净版" , "只要新闻" ] self.sectors_name = { # 公用通栏 "adSector": "通栏广告", # 1-4套皮肤 "headLineSector": "网站头条", "bannerSector": "焦点图", "linkSector": "外链面板", "manyPictureSector": "新闻图文组合1", "commentSector": "新闻图文组合2", "listSector": "新闻图文组合3", "onlyImgSector": "带广告新闻组合", # 第5套皮肤 "headLineSector": "网站头条", "bannerSectorMerge": "焦点图", "linkCxfwSector": "外链面板-带查询服务", "NewsHyjjSector": "行业聚焦新闻通栏", "newsTabsSector": "热点新闻选项卡", "newsTextSector": "文字新闻通栏1", "newsTabsTextSector": "选项卡文字新闻通栏", "newsAllPictureSector": "图片新闻通栏1", "newsTabsAdSector": "选项卡资讯带两个广告", } self.pos_first = ["第一个", "第一", "首位", "最前面", "开头", "第一位", "顶部"] self.pos_second = ["第二个", "第二", "第二位"] self.pos_third = ["第三个", "第三", "第三位"] self.pos_last = ["最后一个", "最后", "末尾", "垫底", "最后一位", "底部"] self.pos_last_second = ["倒数第二个", "倒数第二", "倒数第二位"] self.pos_last_third = ["倒数第三个", "倒数第三", "倒数第三位"] def generate_inputType(self, user_text,matched_template_id): # 导入推理方法 from services.ai_service import ai_service import re # 按标点符号分句处理，支持多指令 segments = re.split(r'[，,。;；|]+', user_text) segments = [s.strip() for s in segments if s.strip()] templateData = None # 根据matched_template_id选择样式数据 if int(matched_template_id) == 5: print("正在使用第5套皮肤的数据！") templateData = sectorStyle5Data else: print("正在使用前4套皮肤的数据！") templateData = sectorStyle1to4Data if not segments: segments = [user_text] for segment in segments: # 1. 检查用户输入中是否包含组删除词 if any(word in segment for word in self.del_group_map): # 删除所有广告相关通栏 # ad_sectors = ["通栏广告", "带广告新闻组合"] # 找到所有名字中含有广告的通栏 ad_sectors = [sector for sector in templateData.sectors_config if "广告" in sector['CNname']] templateData.sectors_config = [ sector for sector in templateData.sectors_config if sector not in ad_sectors ] # 添加推理过程 ai_service.add_reasoning(f"另外，用户不想看到广告，因此我修改了通栏列表，只保留了新闻相关通栏") print(f"已删除所有广告通栏: {ad_sectors}") # 2. 检查用户输入中是否包含普通删除词 elif any(word in segment for word in self.del_map): # 查找用户输入中提到的通栏名称 sectors_to_delete = [] for cnname in self.sectors_name.values(): if cnname in segment: sectors_to_delete.append(cnname) # 删除匹配的通栏配置 if sectors_to_delete: templateData.sectors_config = [ sector for sector in templateData.sectors_config if sector['CNname'] not in sectors_to_delete ] # 添加推理过程 ai_service.add_reasoning(f"另外，用户想要删除{sectors_to_delete}通栏，我来把它隐藏掉") print(f"已删除通栏: {sectors_to_delete}") # 3. 检查用户输入中是否包含位置调整 # 遍历所有通栏名称 for name, cnname in self.sectors_name.items(): if cnname in segment: new_weight = None reasoning_msg = "" # 获取当前最大和最小权重 current_weights = [s['weight'] for s in templateData.sectors_config] if not current_weights: break max_weight = max(current_weights) min_weight = min(current_weights) # 检查位置关键词 force_pos_val = None if any(word in segment for word in self.pos_first): new_weight = max_weight + 2 reasoning_msg = f"用户想要把{cnname}放在第一位，我提升了它的权重" elif any(word in segment for word in self.pos_second): new_weight = max_weight - 1 reasoning_msg = f"用户想要把{cnname}放在第二位，我调整了它的权重" elif any(word in segment for word in self.pos_third): new_weight = max_weight - 2 reasoning_msg = f"用户想要把{cnname}放在第三位，我调整了它的权重" elif any(word in segment for word in self.pos_last): new_weight = min_weight - 3 force_pos_val = 1 reasoning_msg = f"用户想要把{cnname}放在最后一位，我降低了它的权重" elif any(word in segment for word in self.pos_last_second): if min_weight < 6: new_weight = min_weight + 1 else: new_weight = min_weight - 2 force_pos_val = 2 reasoning_msg = f"用户想要把{cnname}放在倒数第二位，我调整了它的权重" elif any(word in segment for word in self.pos_last_third): if min_weight < 6: new_weight = min_weight + 2 else: new_weight = min_weight - 1 force_pos_val = 3 reasoning_msg = f"用户想要把{cnname}放在倒数第三位，我调整了它的权重" # 如果确定了新权重，进行修改 if new_weight is not None: for sector in templateData.sectors_config: if sector['name'] == name: sector['weight'] = new_weight if force_pos_val is not None: sector['force_pos'] = force_pos_val # 添加推理过程 ai_service.add_reasoning(reasoning_msg) print(f"已调整通栏 {cnname} 权重为: {new_weight}") break # 全局匹配器实例 inputType_matcher = inputType()