Update app.py

app.py

@@ -4,14 +4,13 @@ from selenium import webdriver
 from selenium.webdriver.common.by import By
 from selenium.webdriver.common.keys import Keys
 from selenium.webdriver.chrome.service import Service as ChromeService
-# 不再需要 webdriver-manager 了
 from selenium.webdriver.support.ui import WebDriverWait
 from selenium.webdriver.support import expected_conditions as EC
 
 def chat_with_sai_automation(prompt_text: str, progress=gr.Progress(track_tqdm=True)) -> str:
     """
-    使用 Selenium 4.6+ 内置的 SeleniumManager，自动下载并管理匹配的
-    Chrome for Testing 和 ChromeDriver。
+    最终版：使用系统中通过 apt-get 安装的 chromium 和 chromedriver。
+    这是在 Hugging Face Spaces 上最稳定、最可靠的方案。
     """
     full_response = ""
     try:
@@ -22,47 +21,46 @@ def chat_with_sai_automation(prompt_text: str, progress=gr.Progress(track_tqdm=T
         options.add_argument("--no-sandbox")
         options.add_argument("--disable-dev-shm-usage")
         options.add_argument("--disable-gpu")
-        # 我们不再需要指定 binary_location，让 SeleniumManager 自动处理
+        # 当通过 apt 安装时，chromium 通常在这里
+        options.binary_location = "/usr/bin/chromium"
 
-        progress(0.1, desc="SeleniumManager 正在自动下载匹配的浏览器和驱动...")
-        # Selenium 4.6.0 及以上版本，不再需要 Service 对象或 webdriver-manager
-        # 直接调用 webdriver.Chrome(options=options) 即可
-        # SeleniumManager 会在后台自动完成所有下载和匹配工作
-        driver = webdriver.Chrome(options=options)
+        progress(0.2, desc="正在指定系统驱动...")
+        # 直接指定由 apt 安装的 chromedriver 的路径
+        service = ChromeService(executable_path='/usr/bin/chromedriver')
+        
+        progress(0.3, desc="正在启动浏览器...")
+        driver = webdriver.Chrome(service=service, options=options)
         
         # --- 2. 访问网页并等待输入框加载 ---
-        progress(0.3, desc=f"正在访问 sai.coludai.cn ...")
+        progress(0.4, desc="正在访问 sai.coludai.cn ...")
         driver.get("https://sai.coludai.cn/")
         
         wait = WebDriverWait(driver, 20)
         textarea_selector = 'textarea[placeholder="随时与未来对话，探索无限可能...."]'
-        progress(0.4, desc="等待页面输入框加载...")
+        progress(0.5, desc="等待页面输入框加载...")
         textarea = wait.until(EC.presence_of_element_located((By.CSS_SELECTOR, textarea_selector)))
         
         # --- 3. 输入问题并发送 ---
-        progress(0.5, desc=f"输入问题并发送...")
+        progress(0.6, desc="输入问题并发送...")
         textarea.send_keys(prompt_text)
         textarea.send_keys(Keys.RETURN)
         
         # --- 4. 等待并捕获 AI 回复 ---
-        progress(0.6, desc="等待 AI 回复...")
+        progress(0.7, desc="等待 AI 回复...")
         
         last_assistant_selector = "(.//div[@class='message-item' and @type='assistant'])[last()]"
         wait.until(EC.presence_of_element_located((By.XPATH, last_assistant_selector)))
         last_response_element = driver.find_element(By.XPATH, last_assistant_selector)
 
         previous_text = ""
-        max_wait_time = 120  # 增加超时时间
+        max_wait_time = 120
         start_time = time.time()
         
         while time.time() - start_time < max_wait_time:
             try:
                 markdown_body = last_response_element.find_element(By.CSS_SELECTOR, '.markdown-body')
                 current_text = markdown_body.text
-                
-                # 更新进度条
-                progress(0.6 + 0.4 * (len(current_text) / 500), desc=f"正在接收... (长度: {len(current_text)})")
-
+                progress(0.7 + 0.3 * (min(len(current_text), 500) / 500), desc=f"正在接收... (长度: {len(current_text)})")
                 if current_text == previous_text and current_text != "":
                     full_response = current_text
                     break
@@ -86,7 +84,7 @@ def chat_with_sai_automation(prompt_text: str, progress=gr.Progress(track_tqdm=T
 # --- 使用 Gradio 创建 Web 界面 ---
 with gr.Blocks() as iface:
     gr.Markdown("# SAI-ChatBot 自动化工具")
-    gr.Markdown("在后台使用 Selenium 自动与 SAI-ChatBot 交互。请耐心等待，首次运行可能需要一些时间来下载浏览器和驱动。")
+    gr.Markdown("在后台使用 Selenium 自动与 SAI-ChatBot 交互。请耐心等待，首次运行可能需要一些时间来构建环境。")
     with gr.Row():
         inp = gr.Textbox(lines=5, label="你的问题", placeholder="在这里输入你想问 SAI-ChatBot 的问题...")
         out = gr.Textbox(lines=15, label="AI 的回复")