Update app.py

app.py

@@ -1187,40 +1187,52 @@ async def websocket_endpoint(ws: WebSocket):
     active_connections.append(ws)
 
     # ── عزل كامل: إنشاء Xvfb + متصفح خاصين بهذا الاتصال فقط ──
-    # كل اتصال WebSocket يحصل على بيئة كمبيوتر افتراضي مستقلة تماماً عن أي اتصال آخر،
-    # بحيث لا يرى أي مستخدم أي أثر من نشاط مستخدم آخر مهما كانوا متصلين في نفس اللحظة.
-    session_display = await create_session_for(ws)
+    # كل اتصال WebSocket يحصل على بيئة كمبيوتر افتراضي مستقلة تماماً عن أي اتصال آخر.
+    # إذا فشل إنشاء جلسة معزولة لأي سبب (صلاحيات، موارد، Xvfb غير متاح...)، نرجع فوراً
+    # للوضع المشترك القديم (DISPLAY الأساسي) بدل أن ينهار الاتصال بصمت بالكامل.
+    session_display = _BASE_DISPLAY
+    try:
+        session_display = await asyncio.wait_for(create_session_for(ws), timeout=8.0)
+    except asyncio.TimeoutError:
+        print("[session] ⚠️ session creation timed out (>8s), falling back to shared display")
+        session_display = _BASE_DISPLAY
+    except Exception as e:
+        print(f"[session] ⚠️ failed to create isolated session, falling back to shared display: {e}")
+        session_display = _BASE_DISPLAY
     _current_display.set(session_display)  # يُربط بـ task هذا الاتصال — لا يؤثر على اتصالات أخرى
 
-    w, h = await asyncio.to_thread(_get_screen_size)
-    await ws.send_text(json.dumps({
-        "type": "connected",
-        "screen_width": w,
-        "screen_height": h,
-        "browser": BROWSER,
-        "display": session_display,
-        "msg": f"Z Computer Mode v7 — Smart Control | Browser: {BROWSER} | Screen: {w}x{h} | Isolated session: {session_display}",
-        "commands": [
-            "screenshot", "terminal {cmd}", "mouse_move {x,y}", "mouse_click {x,y,button}",
-            "keyboard_type {text}", "keyboard_hotkey {keys}", "scroll {x,y,clicks}",
-            "open_app {cmd}", "open_browser {url}", "simple_command {cmd}",
-            "Simple cmds: 'open firefox', 'mouse go x y', 'click x y', 'type TEXT', 'key enter'"
-        ]
-    }, ensure_ascii=False))
-    # أرسل screenshot أولية مع grid (لشاشة هذه الجلسة المعزولة فقط)
-    result = await asyncio.to_thread(capture_screen_with_grid, scale=0.65, quality=72)
-    if result["data"]:
+    try:
+        w, h = await asyncio.to_thread(_get_screen_size)
         await ws.send_text(json.dumps({
-            "type": "screenshot",
-            "data": result["data"],
-            "ts": int(time.time() * 1000),
-            "label": "الشاشة الأولية",
-            "screen_width": result["width"],
-            "screen_height": result["height"],
-            "mouse_x": result["mouse_x"],
-            "mouse_y": result["mouse_y"],
-            "has_grid": True,
+            "type": "connected",
+            "screen_width": w,
+            "screen_height": h,
+            "browser": BROWSER,
+            "display": session_display,
+            "msg": f"Z Computer Mode v7 — Smart Control | Browser: {BROWSER} | Screen: {w}x{h} | Isolated session: {session_display}",
+            "commands": [
+                "screenshot", "terminal {cmd}", "mouse_move {x,y}", "mouse_click {x,y,button}",
+                "keyboard_type {text}", "keyboard_hotkey {keys}", "scroll {x,y,clicks}",
+                "open_app {cmd}", "open_browser {url}", "simple_command {cmd}",
+                "Simple cmds: 'open firefox', 'mouse go x y', 'click x y', 'type TEXT', 'key enter'"
+            ]
         }, ensure_ascii=False))
+        # أرسل screenshot أولية مع grid (لشاشة هذه الجلسة المعزولة فقط)
+        result = await asyncio.to_thread(capture_screen_with_grid, scale=0.65, quality=72)
+        if result["data"]:
+            await ws.send_text(json.dumps({
+                "type": "screenshot",
+                "data": result["data"],
+                "ts": int(time.time() * 1000),
+                "label": "الشاشة الأولية",
+                "screen_width": result["width"],
+                "screen_height": result["height"],
+                "mouse_x": result["mouse_x"],
+                "mouse_y": result["mouse_y"],
+                "has_grid": True,
+            }, ensure_ascii=False))
+    except Exception as e:
+        print(f"[ws] ⚠️ error sending initial messages: {e}")
     try:
         while True:
             raw = await ws.receive_text()