fix: 跟随前端更新

Author: Mag1cFall

src/browser/initialization.py

@@ -305,7 +305,7 @@ async def _initialize_page_logic(browser: AsyncBrowser):
         logger.info(f'✅ 确认位于 AI Studio: {current_url}')
         await found_page.bring_to_front()
         try:
-            input_wrapper_locator = found_page.locator('ms-prompt-input-wrapper')
+            input_wrapper_locator = found_page.locator('ms-prompt-box')
             await expect_async(input_wrapper_locator).to_be_visible(timeout=35000)
             await expect_async(found_page.locator(INPUT_SELECTOR)).to_be_visible(timeout=10000)
             logger.info('✅ 核心输入区域可见。')

src/browser/page_controller.py

@@ -834,7 +834,7 @@ async def _paste_images_via_event(self, images: List[Dict[str, str]], target_loc
     async def submit_prompt(self, prompt: str, image_list: List, check_client_disconnected: Callable):
         self.logger.info(f'[{self.req_id}] 📤 提交提示 ({len(prompt)} chars)...')
         prompt_textarea_locator = self.page.locator(PROMPT_TEXTAREA_SELECTOR)
-        autosize_wrapper_locator = self.page.locator('ms-prompt-input-wrapper ms-autosize-textarea')
+        autosize_wrapper_locator = self.page.locator('ms-prompt-box .text-wrapper')
         submit_button_locator = self.page.locator(SUBMIT_BUTTON_SELECTOR)
         try:
             await expect_async(prompt_textarea_locator).to_be_visible(timeout=5000)
@@ -923,7 +923,7 @@ async def _verify_images_uploaded(self, expected_count: int, check_client_discon
                     except Exception:
                         continue
                 uploaded_images = 0
-                priority_selectors = ['ms-prompt-input-wrapper img', '.prompt-input img', 'textarea[data-test-ms-prompt-textarea] ~ * img', '[data-testid="prompt-input"] img']
+                priority_selectors = ['ms-prompt-box img', '.prompt-input img', 'textarea[data-test-ms-prompt-textarea] ~ * img', '[data-testid="prompt-input"] img']
                 for selector in priority_selectors:
                     try:
                         locator = self.page.locator(selector)

src/config/imagen_selectors.py

@@ -7,8 +7,8 @@
 IMAGEN_TOOLBAR_SELECTOR = 'ms-toolbar'
 IMAGEN_TOOLBAR_TITLE_SELECTOR = 'ms-toolbar h1.mode-title'
 
-IMAGEN_PROMPT_INPUT_SELECTOR = 'ms-prompt-input-wrapper textarea[aria-label="Enter a prompt to generate an image"]'
-IMAGEN_RUN_BUTTON_SELECTOR = 'ms-prompt-input-wrapper ms-run-button button[aria-label="Run"]'
+IMAGEN_PROMPT_INPUT_SELECTOR = 'ms-prompt-box textarea[aria-label="Enter a prompt to generate an image"]'
+IMAGEN_RUN_BUTTON_SELECTOR = 'ms-prompt-box ms-run-button button[aria-label="Run"]'
 
 IMAGEN_GALLERY_CONTAINER_SELECTOR = 'ms-image-generation-gallery'
 IMAGEN_GALLERY_ITEM_SELECTOR = 'ms-image-generation-gallery-image'

src/config/selectors.py

@@ -1,4 +1,4 @@
-PROMPT_TEXTAREA_SELECTOR = 'ms-prompt-input-wrapper ms-autosize-textarea textarea'
+PROMPT_TEXTAREA_SELECTOR = 'ms-prompt-box textarea'
 INPUT_SELECTOR = PROMPT_TEXTAREA_SELECTOR
 INPUT_SELECTOR2 = PROMPT_TEXTAREA_SELECTOR
 SUBMIT_BUTTON_SELECTOR = 'button[aria-label="Run"].run-button'