Update app.py

app.py

@@ -4,56 +4,55 @@ 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
-from webdriver_manager.chrome import ChromeDriverManager
+# 不再需要 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) -> str:
+def chat_with_sai_automation(prompt_text: str, progress=gr.Progress(track_tqdm=True)) -> str:
     """
-    使用 Selenium 自动化浏览器，与 SAI-ChatBot 进行交互。
-    此函数专门为在 Hugging Face Spaces 的环境中运行而优化。
+    使用 Selenium 4.6+ 内置的 SeleniumManager，自动下载并管理匹配的
+    Chrome for Testing 和 ChromeDriver。
     """
-    # --- 1. 设置和初始化浏览器 (Hugging Face 环境) ---
-    options = webdriver.ChromeOptions()
-    options.add_argument("--headless")  # 在服务器上，必须使用无头模式
-    options.add_argument("--no-sandbox")
-    options.add_argument("--disable-dev-shm-usage")
-    options.add_argument("--disable-gpu")
-    
-    # 自动安装和管理 ChromeDriver
-    try:
-        service = ChromeService(ChromeDriverManager().install())
-        driver = webdriver.Chrome(service=service, options=options)
-    except Exception as e:
-        return f"初始化浏览器失败: {e}\n请检查 packages.txt 和 requirements.txt 是否配置正确。"
-    
     full_response = ""
     try:
+        # --- 1. 设置和初始化浏览器 ---
+        progress(0, desc="正在设置浏览器选项...")
+        options = webdriver.ChromeOptions()
+        options.add_argument("--headless")
+        options.add_argument("--no-sandbox")
+        options.add_argument("--disable-dev-shm-usage")
+        options.add_argument("--disable-gpu")
+        # 我们不再需要指定 binary_location，让 SeleniumManager 自动处理
+
+        progress(0.1, desc="SeleniumManager 正在自动下载匹配的浏览器和驱动...")
+        # Selenium 4.6.0 及以上版本，不再需要 Service 对象或 webdriver-manager
+        # 直接调用 webdriver.Chrome(options=options) 即可
+        # SeleniumManager 会在后台自动完成所有下载和匹配工作
+        driver = webdriver.Chrome(options=options)
+        
         # --- 2. 访问网页并等待输入框加载 ---
+        progress(0.3, desc=f"正在访问 sai.coludai.cn ...")
         driver.get("https://sai.coludai.cn/")
         
         wait = WebDriverWait(driver, 20)
         textarea_selector = 'textarea[placeholder="随时与未来对话，探索无限可能...."]'
+        progress(0.4, desc="等待页面输入框加载...")
         textarea = wait.until(EC.presence_of_element_located((By.CSS_SELECTOR, textarea_selector)))
         
         # --- 3. 输入问题并发送 ---
+        progress(0.5, desc=f"输入问题并发送...")
         textarea.send_keys(prompt_text)
         textarea.send_keys(Keys.RETURN)
         
         # --- 4. 等待并捕获 AI 回复 ---
-        initial_assistant_count = len(driver.find_elements(By.CSS_SELECTOR, ".message-item[type='assistant']"))
+        progress(0.6, desc="等待 AI 回复...")
         
         last_assistant_selector = "(.//div[@class='message-item' and @type='assistant'])[last()]"
-        wait.until(EC.numberOfElementsToBeMoreThan(
-            (By.CSS_SELECTOR, ".message-item[type='assistant']"), 
-            initial_assistant_count
-        ))
-        
+        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 = 60  # 秒
+        max_wait_time = 120  # 增加超时时间
         start_time = time.time()
         
         while time.time() - start_time < max_wait_time:
@@ -61,6 +60,9 @@ def chat_with_sai_automation(prompt_text: str) -> str:
                 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)})")
+
                 if current_text == previous_text and current_text != "":
                     full_response = current_text
                     break
@@ -74,21 +76,22 @@ def chat_with_sai_automation(prompt_text: str) -> str:
             full_response = previous_text if previous_text else "超时或未能捕获到完整回复。"
 
     except Exception as e:
-        full_response = f"自动化过程中发生错误: {e}"
+        full_response = f"自动化过程中发生错误: {e}\n\n请检查 Hugging Face Space 的日志获取详细信息。"
     finally:
-        # --- 5. 任务结束，关闭浏览器 ---
-        driver.quit()
-        
+        if 'driver' in locals() and driver:
+            driver.quit()
+            
     return full_response
 
 # --- 使用 Gradio 创建 Web 界面 ---
-iface = gr.Interface(
-    fn=chat_with_sai_automation,
-    inputs=gr.Textbox(lines=5, label="你的问题", placeholder="在这里输入你想问 SAI-ChatBot 的问题..."),
-    outputs=gr.Textbox(lines=15, label="AI 的回复"),
-    title="SAI-ChatBot 自动化工具",
-    description="在后台使用 Selenium 驱动一个真实的 Chrome 浏览器来与 SAI-ChatBot 交互。请耐心等待，首次运行可能需要一些时间来构建环境。"
-)
+with gr.Blocks() as iface:
+    gr.Markdown("# 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 的回复")
+    btn = gr.Button("提交")
+    btn.click(fn=chat_with_sai_automation, inputs=inp, outputs=out)
 
 if __name__ == "__main__":
     iface.launch()