Update app.py

app.py

@@ -710,21 +710,27 @@ async def handle_action(ws: WebSocket, msg: dict, sess: dict):
     # while True الرئيسية في websocket_endpoint، فإذا تأخرت capture_with_grid (قد تصل
     # لدقيقة ونصف في أسوأ سيناريو فشل/تأخر scrot+import+ffmpeg)، تبقى الحلقة كلها
     # محجوبة ولا تستقبل أي رسالة عميل جديدة (نقرة، طلب screenshot آخر، إلخ) حتى تنتهي.
-    # الحل: كل طلب screenshot يُشغَّل فوراً عبر create_task (لا يحجب الحلقة أبداً)،
-    # ونستخدم قفلاً لمنع تراكم أكثر من عملية capture فعلية واحدة بنفس اللحظة على نفس
-    # الـ display (لتفادي تضارب ملفات /tmp المؤقتة واستهلاك CPU مضاعف بلا فائدة) —
-    # فيُصبح بإمكان الذكاء الاصطناعي طلب لقطة شاشة في أي وقت ومن أي مكان دون أن يعلق
-    # خلف طابور أي إجراء سابق بطيء.
+    # الحل: كل طلب screenshot يُشغَّل فوراً عبر create_task (لا يحجب الحلقة أبداً).
+    #
+    # ── إصلاح ثانٍ (مهم جداً): طلب screenshot الصريح من العميل له أولوية مطلقة ──
+    # المشكلة المكتشفة لاحقاً: لو استُخدم قفل واحد مشترك بين shot_bg (تلقائية، بعد
+    # كل فعل كنقرة/فتح تبويب) و shot_explicit (صريحة، يطلبها العميل مباشرة بعد كل
+    # خطوة لتحديث الصورة المعروضة)، فإن shot_explicit قد تنتظر خلف shot_bg طويلاً
+    # إذا كانت الأخيرة قد بدأت فعلاً وعلقت داخل capture_with_grid (محاولات فاشلة
+    # متتالية). هذا يجعل العميل يرى "لم تصل لقطة شاشة" بشكل متكرر حتى بعد نجاح
+    # الخطوة الفعلية (مثل open_tab)، لأن طلبه العاجل كان يصطف خلف عملية تلقائية بطيئة.
+    # الحل: shot_explicit (الصريحة فقط) تحاول الحصول على القفل لفترة قصيرة جداً
+    # (1 ثانية)، وإن لم تنجح (لأن shot_bg تستخدمه)، تُنفَّذ التصوير مباشرة بدون قفل
+    # بدل الانتظار — لأن استجابة العميل الفورية أهم من تفادي تزاحم CPU عرضي بسيط،
+    # وقراءة الشاشة (X11) عملية قراءة فقط لا تُسبب أي تلف بيانات عند التزاحم.
     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="" كفشل صامت).
+        تُستخدم من shot_bg (التلقائية) — تنتظر القفل بسقف زمني 25 ثانية لتفادي
+        تراكم عدة عمليات capture تلقائية فوق بعضها.
         """
         async def _do_capture():
             async with _shot_lock:
@@ -737,6 +743,34 @@ async def handle_action(ws: WebSocket, msg: dict, sess: dict):
             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 _priority_capture(scale, quality, force_mx=None, force_my=None):
+        """
+        تُستخدم من shot_explicit (الطلب الصريح من العميل) فقط — أولوية قصوى.
+        تحاول الحصول على القفل لمدة قصيرة (1 ثانية) فقط؛ إن لم تنجح (القفل
+        محجوز من shot_bg تلقائية بطيئة)، تُنفَّذ التصوير فوراً بدون قفل بدل
+        الانتظار خلف عملية أخرى — العميل يجب أن يحصل على رد سريع دائماً.
+        """
+        try:
+            await asyncio.wait_for(_shot_lock.acquire(), timeout=1.0)
+            try:
+                return await asyncio.wait_for(
+                    asyncio.to_thread(capture_with_grid, display, scale, quality, force_mx, force_my),
+                    timeout=20
+                )
+            finally:
+                _shot_lock.release()
+        except asyncio.TimeoutError:
+            # لم نحصل على القفل بسرعة كافية، أو انتهت مهلة التصوير نفسه —
+            # ننفّذ التصوير مباشرة بدون قفل بدل ترك العميل بلا رد إطلاقاً
+            try:
+                return await asyncio.wait_for(
+                    asyncio.to_thread(capture_with_grid, display, scale, quality, force_mx, force_my),
+                    timeout=20
+                )
+            except asyncio.TimeoutError:
+                print(f"[shot:{display}] ⚠️ priority capture timed out — 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."""
@@ -768,8 +802,9 @@ async def handle_action(ws: WebSocket, msg: dict, sess: dict):
         })
 
     async def shot_explicit(label: str = ""):
-        """screenshot صريح — يُرسل دائماً بدون delta suppression."""
-        result = await _safe_capture(0.65, 75)
+        """screenshot صريح — يُرسل دائماً بدون delta suppression، وله أولوية مطلقة
+        على أي عملية shot_bg تلقائية جارية (لا ينتظر خلفها طويلاً)."""
+        result = await _priority_capture(0.65, 75)
         if not result["data"]:
             return
         sess["last_bg_hash"] = _frame_hash(result["data"])
@@ -985,7 +1020,7 @@ async def handle_action(ws: WebSocket, msg: dict, sess: dict):
         await asyncio.sleep(0.2)
         await xdo(["key", "--clearmodifiers", "Return"], display)
         await send({"type": "ack", "action": "open_tab", "url": url})
-        asyncio.create_task(shot_bg("after open_tab", delay=1.5))
+        asyncio.create_task(shot_bg("after open_tab", delay=2.5))
 
     # ── close_tab ────────────────────────────────────────────
     elif action == "close_tab":
@@ -1014,7 +1049,7 @@ async def handle_action(ws: WebSocket, msg: dict, sess: dict):
         await asyncio.sleep(0.15)
         await xdo(["key", "--clearmodifiers", "Return"], display)
         await send({"type": "ack", "action": "browser_search"})
-        asyncio.create_task(shot_bg("after search", delay=1.5))
+        asyncio.create_task(shot_bg("after search", delay=2.5))
 
     # ── screen_info ──────────────────────────────────────────
     elif action == "screen_info":