diff --git a/spiders/ai_seo/nanometer.py b/spiders/ai_seo/nanometer.py index 349a8af..cf4b2b4 100644 --- a/spiders/ai_seo/nanometer.py +++ b/spiders/ai_seo/nanometer.py @@ -35,8 +35,7 @@ class NanometerSpider(AbstractAiSeoSpider): # 开始操作 await self.browser_page.goto(self.get_home_url(), timeout=600000) #开启深度思考 - # await self.browser_page.locator('//*[@id="NMAI_SIDEBAR_MENU"]/div/div[2]').click() - chat_input_element = self.browser_page.locator('//*[@id="NM-ASSISTANT_chat_input"]') + chat_input_element = self.browser_page.locator('//*[@id="NM-ASSISTANT_chat_input"]//textarea') # 输入提问词 await chat_input_element.press_sequentially(self.prompt) await self.browser_page.keyboard.press('Enter') @@ -68,6 +67,13 @@ class NanometerSpider(AbstractAiSeoSpider): div_height = div_box['height'] if div_box else None logger.debug(f'answer_element: {div_height}') view_port_height = div_box['height']+ 500 + + # 修改标题 + title = iframe.locator("//h1[@id='message-prompt']") + title_text = await title.inner_text() + new_title = title_text.replace('(禁用智能体)', '') + await title.evaluate(f"node => node.innerHTML = '{new_title}'") + # 调整视口大小 await self.browser_page.set_viewport_size({ 'width': 1920,