使用新增的临时聊天模式

Author: MAX-TAB

api_utils/app.py

@@ -30,7 +30,8 @@
     _initialize_page_logic,
     _close_page_logic,
     load_excluded_models,
-    _handle_initial_model_state_and_storage
+    _handle_initial_model_state_and_storage,
+    enable_temporary_chat_mode
 )
 
 import stream
@@ -141,6 +142,7 @@ async def _initialize_browser_and_page():
         server.page_instance, server.is_page_ready = await _initialize_page_logic(server.browser_instance)
         if server.is_page_ready:
             await _handle_initial_model_state_and_storage(server.page_instance)
+            await enable_temporary_chat_mode(server.page_instance)
             server.logger.info("Page initialized successfully.")
         else:
             server.logger.error("Page initialization failed.")

browser_utils/__init__.py

@@ -1,6 +1,6 @@
 # --- browser_utils/__init__.py ---
 # 浏览器操作工具模块
-from .initialization import _initialize_page_logic, _close_page_logic, signal_camoufox_shutdown
+from .initialization import _initialize_page_logic, _close_page_logic, signal_camoufox_shutdown, enable_temporary_chat_mode
 from .operations import (
     _handle_model_list_response,
     detect_and_extract_page_error,
@@ -28,6 +28,7 @@
     '_initialize_page_logic',
     '_close_page_logic', 
     'signal_camoufox_shutdown',
+    'enable_temporary_chat_mode',
 
     # 页面操作相关
     '_handle_model_list_response',

browser_utils/initialization.py

@@ -596,4 +596,34 @@ async def _handle_auth_file_save_auto(temp_context):
         logger.info(f"   自动保存认证状态成功: {auth_save_path}")
     except Exception as save_state_err:
         logger.error(f"   ❌ 自动保存认证状态失败: {save_state_err}", exc_info=True)
-        print(f"   ❌ 自动保存认证状态失败: {save_state_err}", flush=True)
+        print(f"   ❌ 自动保存认证状态失败: {save_state_err}", flush=True)
+
+async def enable_temporary_chat_mode(page: AsyncPage):
+    """
+    检查并启用 AI Studio 界面的“临时聊天”模式。
+    这是一个独立的UI操作，应该在页面完全稳定后调用。
+    """
+    try:
+        logger.info("-> (UI Op) 正在检查并启用 '临时聊天' 模式...")
+        
+        incognito_button_locator = page.locator('button[aria-label="Temporary chat toggle"]')
+        
+        await incognito_button_locator.wait_for(state="visible", timeout=10000)
+        
+        button_classes = await incognito_button_locator.get_attribute("class")
+        
+        if button_classes and 'ms-button-active' in button_classes:
+            logger.info("-> (UI Op) '临时聊天' 模式已激活。")
+        else:
+            logger.info("-> (UI Op) '临时聊天' 模式未激活，正在点击...")
+            await incognito_button_locator.click(timeout=5000, force=True)
+            await asyncio.sleep(1)
+            
+            updated_classes = await incognito_button_locator.get_attribute("class")
+            if updated_classes and 'ms-button-active' in updated_classes:
+                logger.info("✅ (UI Op) '临时聊天' 模式已成功启用。")
+            else:
+                logger.warning("⚠️ (UI Op) 点击后 '临时聊天' 模式状态验证失败。")
+
+    except Exception as e:
+        logger.warning(f"⚠️ (UI Op) 启用 '临时聊天' 模式时出错: {e}")

browser_utils/model_management.py

@@ -294,6 +294,29 @@ async def switch_ai_studio_model(page: AsyncPage, model_id: str, req_id: str) ->
                 logger.warning(f"[{req_id}] 读取页面显示的当前模型ID时出错: {e_disp}。将无法验证页面显示。")
 
             if page_display_match:
+                try:
+                    logger.info(f"[{req_id}] 模型切换成功，重新启用 '临时聊天' 模式...")
+                    incognito_button_locator = page.locator('button[aria-label="Temporary chat toggle"]')
+                    
+                    await incognito_button_locator.wait_for(state="visible", timeout=5000)
+                    
+                    button_classes = await incognito_button_locator.get_attribute("class")
+                    
+                    if button_classes and 'ms-button-active' in button_classes:
+                        logger.info(f"[{req_id}] '临时聊天' 模式已处于激活状态。")
+                    else:
+                        logger.info(f"[{req_id}] '临时聊天' 模式未激活，正在点击以开启...")
+                        await incognito_button_locator.click(timeout=3000)
+                        await asyncio.sleep(0.5)
+                        
+                        updated_classes = await incognito_button_locator.get_attribute("class")
+                        if updated_classes and 'ms-button-active' in updated_classes:
+                             logger.info(f"[{req_id}] ✅ '临时聊天' 模式已成功重新启用。")
+                        else:
+                             logger.warning(f"[{req_id}] ⚠️ 点击后 '临时聊天' 模式状态验证失败，可能未成功重新开启。")
+                
+                except Exception as e:
+                    logger.warning(f"[{req_id}] ⚠️ 模型切换后重新启用 '临时聊天' 模式失败: {e}")
                 return True
             else:
                 logger.error(f"[{req_id}] ❌ 模型切换失败，因为页面显示的模型与期望不符 (即使localStorage可能已更改)。")

browser_utils/page_controller.py

@@ -23,6 +23,7 @@
 )
 from models import ClientDisconnectedError
 from .operations import save_error_snapshot, _wait_for_response_completion, _get_final_response_content
+from .initialization import enable_temporary_chat_mode
 
 class PageController:
     """封装了与AI Studio页面交互的所有操作。"""
@@ -572,6 +573,8 @@ async def clear_chat_history(self, check_client_disconnected: Callable):
             if can_attempt_clear:
                 await self._execute_chat_clear(clear_chat_button_locator, confirm_button_locator, overlay_locator, check_client_disconnected)
                 await self._verify_chat_cleared(check_client_disconnected)
+                self.logger.info(f"[{self.req_id}] 聊天已清空，重新启用 '临时聊天' 模式...")
+                await enable_temporary_chat_mode(self.page)
 
         except Exception as e_clear:
             self.logger.error(f"[{self.req_id}] 清空聊天过程中发生错误: {e_clear}")
@@ -580,21 +583,78 @@ async def clear_chat_history(self, check_client_disconnected: Callable):
             raise
 
     async def _execute_chat_clear(self, clear_chat_button_locator, confirm_button_locator, overlay_locator, check_client_disconnected: Callable):
-        """执行清空聊天操作 (适配新版UI，无确认对话框)"""
-        self.logger.info(f"[{self.req_id}] 点击\"清空聊天\"按钮: {CLEAR_CHAT_BUTTON_SELECTOR}")
-        
+        """执行清空聊天操作"""
+        overlay_initially_visible = False
         try:
+            if await overlay_locator.is_visible(timeout=1000):
+                overlay_initially_visible = True
+                self.logger.info(f"[{self.req_id}] 清空聊天确认遮罩层已可见。直接点击\"继续\"。")
+        except TimeoutError:
+            self.logger.info(f"[{self.req_id}] 清空聊天确认遮罩层初始不可见 (检查超时或未找到)。")
+            overlay_initially_visible = False
+        except Exception as e_vis_check:
+            self.logger.warning(f"[{self.req_id}] 检查遮罩层可见性时发生错误: {e_vis_check}。假定不可见。")
+            overlay_initially_visible = False
+
+        await self._check_disconnect(check_client_disconnected, "清空聊天 - 初始遮罩层检查后")
+
+        if overlay_initially_visible:
+            self.logger.info(f"[{self.req_id}] 点击\"继续\"按钮 (遮罩层已存在): {CLEAR_CHAT_CONFIRM_BUTTON_SELECTOR}")
+            await confirm_button_locator.click(timeout=CLICK_TIMEOUT_MS)
+        else:
+            self.logger.info(f"[{self.req_id}] 点击\"清空聊天\"按钮: {CLEAR_CHAT_BUTTON_SELECTOR}")
             await clear_chat_button_locator.click(timeout=CLICK_TIMEOUT_MS)
             await self._check_disconnect(check_client_disconnected, "清空聊天 - 点击\"清空聊天\"后")
-            
-            await asyncio.sleep(0.5) 
-            
-            self.logger.info(f"[{self.req_id}] ✅ '清空聊天'按钮已点击。新版UI无确认步骤，操作完成。")
 
-        except Exception as e:
-            self.logger.error(f"[{self.req_id}] 点击'清空聊天'按钮时发生错误: {e}")
-            raise
-        
+            try:
+                self.logger.info(f"[{self.req_id}] 等待清空聊天确认遮罩层出现: {OVERLAY_SELECTOR}")
+                await expect_async(overlay_locator).to_be_visible(timeout=WAIT_FOR_ELEMENT_TIMEOUT_MS)
+                self.logger.info(f"[{self.req_id}] 清空聊天确认遮罩层已出现。")
+            except TimeoutError:
+                error_msg = f"等待清空聊天确认遮罩层超时 (点击清空按钮后)。请求 ID: {self.req_id}"
+                self.logger.error(error_msg)
+                await save_error_snapshot(f"clear_chat_overlay_timeout_{self.req_id}")
+                raise Exception(error_msg)
+
+            await self._check_disconnect(check_client_disconnected, "清空聊天 - 遮罩层出现后")
+            self.logger.info(f"[{self.req_id}] 点击\"继续\"按钮 (在对话框中): {CLEAR_CHAT_CONFIRM_BUTTON_SELECTOR}")
+            await confirm_button_locator.click(timeout=CLICK_TIMEOUT_MS)
+
+        await self._check_disconnect(check_client_disconnected, "清空聊天 - 点击\"继续\"后")
+
+        # 等待对话框消失
+        max_retries_disappear = 3
+        for attempt_disappear in range(max_retries_disappear):
+            try:
+                self.logger.info(f"[{self.req_id}] 等待清空聊天确认按钮/对话框消失 (尝试 {attempt_disappear + 1}/{max_retries_disappear})...")
+                await expect_async(confirm_button_locator).to_be_hidden(timeout=CLEAR_CHAT_VERIFY_TIMEOUT_MS)
+                await expect_async(overlay_locator).to_be_hidden(timeout=1000)
+                self.logger.info(f"[{self.req_id}] ✅ 清空聊天确认对话框已成功消失。")
+                break
+            except TimeoutError:
+                self.logger.warning(f"[{self.req_id}] ⚠️ 等待清空聊天确认对话框消失超时 (尝试 {attempt_disappear + 1}/{max_retries_disappear})。")
+                if attempt_disappear < max_retries_disappear - 1:
+                    await asyncio.sleep(1.0)
+                    await self._check_disconnect(check_client_disconnected, f"清空聊天 - 重试消失检查 {attempt_disappear + 1} 前")
+                    continue
+                else:
+                    error_msg = f"达到最大重试次数。清空聊天确认对话框未消失。请求 ID: {self.req_id}"
+                    self.logger.error(error_msg)
+                    await save_error_snapshot(f"clear_chat_dialog_disappear_timeout_{self.req_id}")
+                    raise Exception(error_msg)
+            except ClientDisconnectedError:
+                self.logger.info(f"[{self.req_id}] 客户端在等待清空确认对话框消失时断开连接。")
+                raise
+            except Exception as other_err:
+                self.logger.warning(f"[{self.req_id}] 等待清空确认对话框消失时发生其他错误: {other_err}")
+                if attempt_disappear < max_retries_disappear - 1:
+                    await asyncio.sleep(1.0)
+                    continue
+                else:
+                    raise
+
+            await self._check_disconnect(check_client_disconnected, f"清空聊天 - 消失检查尝试 {attempt_disappear + 1} 后")
+
     async def _verify_chat_cleared(self, check_client_disconnected: Callable):
         """验证聊天已清空"""
         last_response_container = self.page.locator(RESPONSE_CONTAINER_SELECTOR).last

config/selectors.py

@@ -11,7 +11,7 @@
 # --- 按钮选择器 ---
 SUBMIT_BUTTON_SELECTOR = 'button[aria-label="Run"].run-button'
 CLEAR_CHAT_BUTTON_SELECTOR = 'button[data-test-clear="outside"][aria-label="New chat"]'
-CLEAR_CHAT_CONFIRM_BUTTON_SELECTOR = 'button.ms-button-primary:has-text("Continue")'
+CLEAR_CHAT_CONFIRM_BUTTON_SELECTOR = 'button.ms-button-primary:has-text("Discard and continue")'
 UPLOAD_BUTTON_SELECTOR = 'button[aria-label^="Insert assets"]'
 
 # --- 响应相关选择器 ---