Update app.py

app.py

@@ -1,85 +1,23 @@
-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
-from webdriver_manager.firefox import GeckoDriverManager
+from playwright.sync_api import sync_playwright
 
-# --- 核心流量生成逻辑 ---
-def generate_traffic(url, views, stay_duration_min, stay_duration_max):
-    """
-    使用 Selenium 无头浏览器模拟访问指定 URL
-    :param url: 目标网址
-    :param views: 访问次数
-    :param stay_duration_min: 每次访问停留最小时间（秒）
-    :param stay_duration_max: 每次访问停留最大时间（秒）
-    :return: 生成日志
-    """
+def generate_traffic_playwright(url, views, stay_duration_min, stay_duration_max):
     if not url.startswith('http'):
-        return "错误：请输入有效的网址（以 http:// 或 https:// 开头）。"
-
-    # 配置 Firefox 无头模式
-    options = Options()
-    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'。"
+        return "错误：请输入有效的网址。"
     
-    options.binary_location = firefox_binary_path
-
-    # 使用 webdriver-manager 自动安装和管理 GeckoDriver
-    try:
-        service = Service(GeckoDriverManager().install())
-        driver = webdriver.Firefox(service=service, options=options)
-    except Exception as e:
-        return f"初始化浏览器失败: {e}"
-
     logs = []
-    for i in range(views):
-        try:
-            driver.get(url)
-            stay_time = random.randint(stay_duration_min, stay_duration_max)
-            logs.append(f"✅ 第 {i+1}/{views} 次访问成功，停留 {stay_time} 秒。")
-            time.sleep(stay_time)
-        except Exception as e:
-            logs.append(f"❌ 第 {i+1} 次访问失败: {e}")
-    
-    driver.quit()
-    return "\n".join(logs)
-
-# --- Gradio 界面 ---
-with gr.Blocks(title="网站流量生成器") as demo:
-    gr.Markdown("## 🤖 网站流量生成器\n基于 Selenium 和 Hugging Face Spaces")
-    
-    with gr.Row():
-        with gr.Column():
-            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)
-            run_btn = gr.Button("🚀 开始生成流量", variant="primary")
+    with sync_playwright() as p:
+        # Playwright 会自动下载和管理浏览器
+        browser = p.firefox.launch(headless=True)
+        page = browser.new_page()
         
-        with gr.Column():
-            output_log = gr.Textbox(label="运行日志", lines=10, max_lines=20, interactive=False)
-    
-    run_btn.click(
-        fn=generate_traffic,
-        inputs=[url_input, views_input, stay_min_input, stay_max_input],
-        outputs=output_log
-    )
-
-# 启动应用
-demo.launch()
+        for i in range(views):
+            try:
+                page.goto(url)
+                stay_time = random.randint(stay_duration_min, stay_duration_max)
+                logs.append(f"✅ 第 {i+1}/{views} 次访问成功，停留 {stay_time} 秒。")
+                page.wait_for_timeout(stay_time * 1000)
+            except Exception as e:
+                logs.append(f"❌ 第 {i+1} 次访问失败: {e}")
+        
+        browser.close()
+    return "\n".join(logs)