Update app.py

app.py

@@ -705,6 +705,38 @@ async def handle_action(ws: WebSocket, msg: dict, sess: dict):
         except Exception:
             pass
 
+    # ── إصلاح جذري: قفل خاص بكل session لمنع تراكم/تضارب عمليات capture المتزامنة ──
+    # المشكلة الأصلية: action == "screenshot" كان يُنفَّذ بـ await مباشر ضمن حلقة
+    # while True الرئيسية في websocket_endpoint، فإذا تأخرت capture_with_grid (قد تصل
+    # لدقيقة ونصف في أسوأ سيناريو فشل/تأخر scrot+import+ffmpeg)، تبقى الحلقة كلها
+    # محجوبة ولا تستقبل أي رسالة عميل جديدة (نقرة، طلب screenshot آخر، إلخ) حتى تنتهي.
+    # الحل: كل طلب screenshot يُشغَّل فوراً عبر create_task (لا يحجب الحلقة أبداً)،
+    # ونستخدم قفلاً لمنع تراكم أكثر من عملية capture فعلية واحدة بنفس اللحظة على نفس
+    # الـ display (لتفادي تضارب ملفات /tmp المؤقتة واستهلاك CPU مضاعف بلا فائدة) —
+    # فيُصبح بإمكان الذكاء الاصطناعي طلب لقطة شاشة في أي وقت ومن أي مكان دون أن يعلق
+    # خلف طابور أي إجراء سابق بطيء.
+    if "_shot_lock" not in sess:
+        sess["_shot_lock"] = asyncio.Lock()
+    _shot_lock = sess["_shot_lock"]
+
+    async def _safe_capture(scale, quality, force_mx=None, force_my=None):
+        """
+        يحصل على القفل ثم يلتقط الشاشة، مع سقف زمني إجمالي صارم (25 ثانية).
+        هذا يضمن أنه حتى لو طال انتظار القفل (طلبات متراكمة) أو علقت capture_with_grid
+        لسبب غير متوقع، لا يبقى أي طلب لقطة شاشة معلّقاً للأبد — يفشل بهدوء بدل ذلك
+        ويُعاد المحاولة في الطلب التالي (كل الاستدعاءات تتعامل بالفعل مع data="" كفشل صامت).
+        """
+        async def _do_capture():
+            async with _shot_lock:
+                return await asyncio.to_thread(
+                    capture_with_grid, display, scale, quality, force_mx, force_my
+                )
+        try:
+            return await asyncio.wait_for(_do_capture(), timeout=25)
+        except asyncio.TimeoutError:
+            print(f"[shot:{display}] ⚠️ capture timed out after 25s — skipping this frame")
+            return {"data": "", "width": 1920, "height": 1080, "mouse_x": 0, "mouse_y": 0}
+
     async def shot_bg(label: str = "", delay: float = 0.5,
                       force_mx: int | None = None, force_my: int | None = None):
         """auto_shot في الخلفية — مع rate limit + delta suppression."""
@@ -713,9 +745,7 @@ async def handle_action(ws: WebSocket, msg: dict, sess: dict):
         if now - sess.get("last_bg_shot_ts", 0) < 0.3:
             await asyncio.sleep(0.3 - (now - sess["last_bg_shot_ts"]))
 
-        result = await asyncio.to_thread(
-            capture_with_grid, display, 0.65, 72, force_mx, force_my
-        )
+        result = await _safe_capture(0.65, 72, force_mx, force_my)
         sess["last_bg_shot_ts"] = time.time()
         if not result["data"]:
             return
@@ -739,7 +769,7 @@ async def handle_action(ws: WebSocket, msg: dict, sess: dict):
 
     async def shot_explicit(label: str = ""):
         """screenshot صريح — يُرسل دائماً بدون delta suppression."""
-        result = await asyncio.to_thread(capture_with_grid, display, 0.65, 75)
+        result = await _safe_capture(0.65, 75)
         if not result["data"]:
             return
         sess["last_bg_hash"] = _frame_hash(result["data"])
@@ -767,8 +797,12 @@ async def handle_action(ws: WebSocket, msg: dict, sess: dict):
         return
 
     # ── screenshot ────────────────────────────────────────────
+    # إصلاح: create_task بدل await مباشر — لا يحجب حلقة استقبال الرسائل أبداً،
+    # حتى لو تأخرت capture_with_grid لأي سبب (CPU ضعيف، X11 بطيء، إلخ).
+    # هذا يضمن أن أي إجراء لاحق (نقرة، طباعة، طلب screenshot آخر) يصل ويُعالَج
+    # فوراً دون انتظار اكتمال هذه اللقطة أولاً.
     if action == "screenshot":
-        await shot_explicit("explicit screenshot")
+        asyncio.create_task(shot_explicit("explicit screenshot"))
 
     # ── terminal ─────────────────────────────────────────────
     elif action == "terminal":