Update app.py

app.py

@@ -772,8 +772,18 @@ async def handle_action(ws: WebSocket, msg: dict, sess: dict):
                 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."""
+                      force_mx: int | None = None, force_my: int | None = None,
+                      extra_shot_delay: float | None = None):
+        """
+        auto_shot في الخلفية — مع rate limit + delta suppression.
+        إذا تم تمرير extra_shot_delay، تُرسل لقطة ثانية إضافية بعد ذلك التأخير
+        الإضافي (محسوباً من وقت انتهاء اللقطة الأولى) — هذا يغطي حالة النقر على
+        رابط ينقل لصفحة جديدة قد لا تكتمل تحميلها خلال التأخير الأولي القصير
+        (مثلاً صفحة بطيئة على سيرفر محدود الموارد)، دون الحاجة لمعرفة مسبقة بنوع
+        العنصر المنقور عليه. تتجاهل delta suppression لهذه اللقطة الثانية تحديداً
+        لأن المحتوى متوقع أن يكون مختلفاً (صفحة تحمّلت أكثر) حتى لو تشابه الـ hash
+        جزئياً مع خلفية مشابهة.
+        """
         await asyncio.sleep(delay)
         now = time.time()
         if now - sess.get("last_bg_shot_ts", 0) < 0.3:
@@ -781,25 +791,40 @@ async def handle_action(ws: WebSocket, msg: dict, sess: dict):
 
         result = await _safe_capture(0.65, 72, force_mx, force_my)
         sess["last_bg_shot_ts"] = time.time()
-        if not result["data"]:
-            return
-
-        fh = _frame_hash(result["data"])
-        if fh == sess.get("last_bg_hash", ""):
-            return
-        sess["last_bg_hash"] = fh
-
-        await send({
-            "type": "screenshot",
-            "data": result["data"],
-            "ts": int(time.time() * 1000),
-            "auto": True, "label": label,
-            "screen_width": result["width"],
-            "screen_height": result["height"],
-            "mouse_x": result["mouse_x"],
-            "mouse_y": result["mouse_y"],
-            "has_grid": True,
-        })
+        if result["data"]:
+            fh = _frame_hash(result["data"])
+            if fh != sess.get("last_bg_hash", ""):
+                sess["last_bg_hash"] = fh
+                await send({
+                    "type": "screenshot",
+                    "data": result["data"],
+                    "ts": int(time.time() * 1000),
+                    "auto": True, "label": label,
+                    "screen_width": result["width"],
+                    "screen_height": result["height"],
+                    "mouse_x": result["mouse_x"],
+                    "mouse_y": result["mouse_y"],
+                    "has_grid": True,
+                })
+
+        if extra_shot_delay:
+            await asyncio.sleep(extra_shot_delay)
+            result2 = await _safe_capture(0.65, 72, force_mx, force_my)
+            sess["last_bg_shot_ts"] = time.time()
+            if result2["data"]:
+                fh2 = _frame_hash(result2["data"])
+                sess["last_bg_hash"] = fh2
+                await send({
+                    "type": "screenshot",
+                    "data": result2["data"],
+                    "ts": int(time.time() * 1000),
+                    "auto": True, "label": f"{label} (delayed)",
+                    "screen_width": result2["width"],
+                    "screen_height": result2["height"],
+                    "mouse_x": result2["mouse_x"],
+                    "mouse_y": result2["mouse_y"],
+                    "has_grid": True,
+                })
 
     async def shot_explicit(label: str = ""):
         """screenshot صريح — يُرسل دائماً بدون delta suppression، وله أولوية مطلقة
@@ -875,7 +900,7 @@ async def handle_action(ws: WebSocket, msg: dict, sess: dict):
             await xdo(["click", btn], display)
         btn_name = {"1": "left", "2": "middle", "3": "right"}.get(btn, "left")
         await send({"type": "ack", "action": "mouse_click", "x": x, "y": y, "button": btn_name})
-        asyncio.create_task(shot_bg(f"click {btn_name} ({x},{y})", delay=0.5, force_mx=x, force_my=y))
+        asyncio.create_task(shot_bg(f"click {btn_name} ({x},{y})", delay=0.5, force_mx=x, force_my=y, extra_shot_delay=3.0))
 
     # ── mouse_drag ───────────────────────────────────────────
     elif action == "mouse_drag":