Fix: still startup status

Dockerfile

@@ -8,7 +8,7 @@ COPY . .
 COPY example.config.json config.json
 
 RUN apk update
-RUN apk add --no-cache chromium chromium-chromedriver
+RUN apk add --no-cache chromium chromium-chromedriver xvfb-run
 
 RUN pip install --no-cache-dir -r requirements.txt
 RUN chmod +x /app/startup.sh

main.py

@@ -241,9 +241,24 @@ def build_chrome_options():
         options.binary_location = chrome_bin
     return options
 
-def register_one_account():
+def get_chromedriver_path():
+    candidate = os.getenv("CHROMEDRIVER_PATH")
+    if candidate and os.path.exists(candidate):
+        return candidate
+    for candidate in ("/usr/bin/chromedriver", "/usr/lib/chromium/chromedriver"):
+        if os.path.exists(candidate):
+            return candidate
+    return None
+
+def create_driver():
     options = build_chrome_options()
-    driver = uc.Chrome(options=options, use_subprocess=True)
+    driver_path = get_chromedriver_path()
+    if driver_path:
+        return uc.Chrome(options=options, use_subprocess=True, driver_executable_path=driver_path)
+    return uc.Chrome(options=options, use_subprocess=True)
+
+def register_one_account():
+    driver = create_driver()
     email = None
     try:
         email, ok, cfg = register(driver)
@@ -262,8 +277,7 @@ def main():
     apply_config(cfg)
     print(f"\n{'='*50}\nGemini Business 批量注册 - 共 {TOTAL_ACCOUNTS} 个\n{'='*50}\n")
 
-    options = build_chrome_options()
-    driver = uc.Chrome(options=options, use_subprocess=True)
+    driver = create_driver()
     success, fail, accounts = 0, 0, []
 
     for i in range(TOTAL_ACCOUNTS):
@@ -272,8 +286,7 @@ def main():
         try:
             driver.current_url  # 检查driver是否有效
         except:
-            options = build_chrome_options()
-            driver = uc.Chrome(options=options, use_subprocess=True)
+            driver = create_driver()
 
         email = None
         try:
@@ -287,8 +300,7 @@ def main():
             save_error_screenshot(driver, email)
             try: driver.quit()
             except: pass
-            options = build_chrome_options()
-            driver = uc.Chrome(options=options, use_subprocess=True)
+            driver = create_driver()
 
         print(f"\n进度: {i+1}/{TOTAL_ACCOUNTS} | 成功: {success} | 失败: {fail}")
 

server.py

@@ -1,5 +1,6 @@
 from datetime import datetime
 from concurrent.futures import ThreadPoolExecutor
+import traceback
 import threading
 import uuid
 
@@ -44,6 +45,8 @@ def run_task(task_id):
         else:
             update_task(task_id, status="failed", error=f"register failed: {email}")
     except Exception as exc:
+        print(f"[ERR] task {task_id} failed: {exc}")
+        print(traceback.format_exc())
         update_task(task_id, status="failed", error=str(exc))
 
 
@@ -107,4 +110,3 @@ def task_info(task_id: str, x_api_key: str | None = Header(default=None)):
     if task["status"] == "failed":
         payload["error"] = task["error"]
     return payload
-

startup.sh

@@ -15,4 +15,9 @@ if [ -z "${CHROME_BIN:-}" ]; then
   fi
 fi
 
+if command -v xvfb-run >/dev/null 2>&1; then
+  echo "Using xvfb-run for virtual display."
+  exec xvfb-run -a uvicorn server:app --host 0.0.0.0 --port "${PORT:-7860}" --log-level debug --access-log
+fi
+
 exec uvicorn server:app --host 0.0.0.0 --port "${PORT:-7860}" --log-level debug --access-log