Update fun.py

fun.py

@@ -12,7 +12,7 @@ import random
 from typing import Dict
 from loguru import logger
 from starlette.responses import StreamingResponse
-
+import socket
 app = FastAPI()
 
 # 浏览器实例管理
@@ -47,7 +47,16 @@ def get_default_launch_args():
         "--no-default-browser-check",
         "--remote-allow-origins=*"
     ]
-
+def wait_port(host, port, timeout=5.0):
+    """等待端口可用"""
+    start = time.time()
+    while time.time() - start < timeout:
+        try:
+            with socket.create_connection((host, port), timeout=0.5):
+                return True
+        except Exception:
+            time.sleep(0.1)
+    return False
 # BitBrowser风格API
 @app.post("/browser/open")
 async def open_browser(request: Request):
@@ -78,6 +87,9 @@ async def open_browser(request: Request):
                 deduped_args.append(arg)
         args = [CHROME_PATH] + deduped_args
     proc = subprocess.Popen(args, stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
+    if not wait_port("127.0.0.1", port, timeout=8):
+        proc.terminate()
+        return {"code": 2, "msg": f"chrome端口{port}未就绪"}
     async with httpx.AsyncClient() as client:
         resp = await client.get(f"http://127.0.0.1:{port}/json/version")
         if resp.status_code == 200: