Create app.py

app.py

@@ -0,0 +1,94 @@
+import gradio as gr
+import time
+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
+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:
+    """
+    使用 Selenium 自动化浏览器，与 SAI-ChatBot 进行交互。
+    此函数专门为在 Hugging Face Spaces 的环境中运行而优化。
+    """
+    # --- 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:
+        # --- 2. 访问网页并等待输入框加载 ---
+        driver.get("https://sai.coludai.cn/")
+        
+        wait = WebDriverWait(driver, 20)
+        textarea_selector = 'textarea[placeholder="随时与未来对话，探索无限可能...."]'
+        textarea = wait.until(EC.presence_of_element_located((By.CSS_SELECTOR, textarea_selector)))
+        
+        # --- 3. 输入问题并发送 ---
+        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']"))
+        
+        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
+        ))
+        
+        last_response_element = driver.find_element(By.XPATH, last_assistant_selector)
+
+        previous_text = ""
+        # 设置一个最长等待时间，防止无限循环
+        max_wait_time = 60  # 秒
+        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
+                
+                if current_text == previous_text and current_text != "":
+                    full_response = current_text
+                    break
+                else:
+                    previous_text = current_text
+                    time.sleep(1.5)
+            except Exception:
+                time.sleep(0.5)
+        
+        if not full_response:
+            full_response = previous_text if previous_text else "超时或未能捕获到完整回复。"
+
+    except Exception as e:
+        full_response = f"自动化过程中发生错误: {e}"
+    finally:
+        # --- 5. 任务结束，关闭浏览器 ---
+        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 交互。请耐心等待，首次运行可能需要一些时间来构建环境。"
+)
+
+if __name__ == "__main__":
+    iface.launch()