Update app.py

app.py

@@ -1,6 +1,8 @@
 import gradio as gr
 import time
 import random
+import os
+import shutil
 from selenium import webdriver
 from selenium.webdriver.firefox.service import Service
 from selenium.webdriver.firefox.options import Options
@@ -21,11 +23,23 @@ def generate_traffic(url, views, stay_duration_min, stay_duration_max):
 
     # 配置 Firefox 无头模式
     options = Options()
-    options.add_argument("--headless")  # 必须使用无头模式
+    options.add_argument("--headless")
     options.add_argument("--disable-gpu")
     options.add_argument("--no-sandbox")
     options.add_argument("--disable-dev-shm-usage")
 
+    # 🔧 关键修复：为 Hugging Face Spaces 指定 Firefox 二进制路径
+    # Space 中 Firefox 通常安装在 /usr/bin/firefox
+    firefox_binary_path = "/usr/bin/firefox"
+    if not os.path.exists(firefox_binary_path):
+        # 作为备选方案，尝试通过 which 命令查找
+        firefox_binary_path = shutil.which("firefox")
+    
+    if not firefox_binary_path:
+        return "错误：在 Space 中未找到 Firefox 浏览器。请确保已在 apt.txt 中添加 'firefox'。"
+    
+    options.binary_location = firefox_binary_path
+
     # 使用 webdriver-manager 自动安装和管理 GeckoDriver
     try:
         service = Service(GeckoDriverManager().install())
@@ -52,7 +66,7 @@ with gr.Blocks(title="网站流量生成器") as demo:
     
     with gr.Row():
         with gr.Column():
-            url_input = gr.Textbox(label="目标网址 (URL)", placeholder="https://example.com")
+            url_input = gr.Textbox(label="目标网址", placeholder="https://example.com")
             views_input = gr.Number(label="访问次数", value=10, minimum=1, maximum=100, precision=0)
             stay_min_input = gr.Number(label="最小停留时间 (秒)", value=5, minimum=1, precision=0)
             stay_max_input = gr.Number(label="最大停留时间 (秒)", value=15, minimum=1, precision=0)