Update app.py

app.py

@@ -349,6 +349,7 @@ async def get_or_create_user_session(user_id: str, ws: WebSocket) -> dict:
             "last_bg_hash": "",
             "active_ws": ws,
             "created": time.time(),
+            "consecutive_capture_failures": 0,  # عدّاد فشل الالتقاط المتتالي — يُصفَّر عند أي نجاح أو reset
         }
         _user_displays[user_id] = sess
         print(f"[session] ✅ New user '{user_id}' → display {display}")
@@ -411,6 +412,7 @@ async def reset_user_computer(user_id: str) -> dict:
             _user_displays[user_id]["xvfb_proc"] = new_proc
             _user_displays[user_id]["last_bg_shot_ts"] = 0.0
             _user_displays[user_id]["last_bg_hash"] = ""
+            _user_displays[user_id]["consecutive_capture_failures"] = 0
 
     print(f"[reset] ✅ Computer reset done for '{user_id}' on {display}")
     return _user_displays.get(user_id, {})
@@ -492,7 +494,7 @@ def _capture_via_import_pipe(display: str):
     try:
         r = subprocess.run(
             ["import", "-window", "root", "-silent", "png:-"],
-            env=env, timeout=6, capture_output=True
+            env=env, capture_output=True
         )
         if r.returncode != 0 or not r.stdout or len(r.stdout) < 500:
             return None, 0, 0
@@ -548,7 +550,7 @@ def _capture_via_xwd(display: str):
     try:
         r = subprocess.run(
             ["xwd", "-root", "-display", display, "-silent"],
-            env=env, timeout=6, capture_output=True
+            env=env, capture_output=True
         )
         if r.returncode != 0 or not r.stdout or len(r.stdout) < 500:
             return None, 0, 0
@@ -575,7 +577,16 @@ def _capture_raw(display: str) -> tuple:
     ImageMagick وغير python-xlib) — هذا يضمن أن فشل مصدر واحد (مثل
     ImageMagick معطوب أو تعارض في مكتبة xlib) لا يعني فشل الالتقاط
     بالكامل، لأن xwd تسلك مساراً مستقلاً تماماً وقد تنجح حين تفشل الأخريان.
-    بحد أقصى محاولتين سريعتين لكل طريقة.
+    لا يوجد أي حد زمني (timeout) على أي محاولة — كل محاولة تُترك تكتمل
+    فعلياً مهما طال الوقت؛ الحماية من التعليق الفعلي (X server ميت) تأتي
+    من طبقة أعلى (شمّاعة الفشل المتتالي + reset تلقائي) وليس من قطع
+    العملية في المنتصف.
+
+    ── القيمة الرابعة المُعادة (is_placeholder) ──
+    True فقط عندما فشلت الطرق الثلاث فعلياً ورجعنا لصورة "Screenshot
+    failed" المرسومة يدوياً. الطبقة الأعلى (capture_with_grid) تستخدم
+    هذا العلم لتفعيل عدّاد الفشل المتتالي وتحفيز reset تلقائي كامل —
+    بدل الاعتماد على فحص محتوى النص داخل الصورة.
     """
     for method_name, method in (
         ("import-pipe", _capture_via_import_pipe),
@@ -588,7 +599,7 @@ def _capture_raw(display: str) -> tuple:
                 if img and not _is_black_screen(img):
                     if attempt > 0:
                         print(f"[cap:{display}] ✅ {method_name} succeeded on retry")
-                    return img, w, h
+                    return img, w, h, False
             except Exception as e:
                 print(f"[cap:{display}] {method_name}: {e}")
             if attempt == 0:
@@ -602,7 +613,7 @@ def _capture_raw(display: str) -> tuple:
     draw.rectangle([(0, 0), (sw, 55)], fill=(30, 40, 80))
     draw.text((10, 10), f"⚠️ Screenshot failed — DISPLAY={display}", fill=(255, 120, 80))
     draw.text((10, 32), "Methods tried: import-pipe, xlib-direct, xwd", fill=(130, 130, 150))
-    return img, sw or 1280, sh or 800
+    return img, sw or 1280, sh or 800, True
 
 
 def _get_screen_size(display: str) -> tuple:
@@ -610,7 +621,7 @@ def _get_screen_size(display: str) -> tuple:
         r = subprocess.run(
             ["xdotool", "getdisplaygeometry"],
             env={**os.environ, "DISPLAY": display},
-            capture_output=True, text=True, timeout=5
+            capture_output=True, text=True
         )
         parts = r.stdout.strip().split()
         return int(parts[0]), int(parts[1])
@@ -623,7 +634,7 @@ def _get_mouse_pos(display: str) -> tuple:
         r = subprocess.run(
             ["xdotool", "getmouselocation"],
             env={**os.environ, "DISPLAY": display},
-            capture_output=True, text=True, timeout=5
+            capture_output=True, text=True
         )
         mx = int(re.search(r"x:(\d+)", r.stdout).group(1))
         my = int(re.search(r"y:(\d+)", r.stdout).group(1))
@@ -720,10 +731,11 @@ def capture_with_grid(display: str, scale: float = 0.85, quality: int = 72,
     """
     from PIL import Image, ImageDraw
 
-    img, ow, oh = _capture_raw(display)
+    img, ow, oh, is_placeholder = _capture_raw(display)
     if img is None:
         return {"data": "", "data_grid": "", "data_grid2": "", "data_grid3": "",
-                "width": 1280, "height": 800, "mouse_x": 0, "mouse_y": 0}
+                "width": 1280, "height": 800, "mouse_x": 0, "mouse_y": 0,
+                "is_placeholder": True}
 
     mx, my = (force_mx, force_my) if force_mx is not None else _get_mouse_pos(display)
 
@@ -759,6 +771,7 @@ def capture_with_grid(display: str, scale: float = 0.85, quality: int = 72,
         "data_grid2": data_grid2,
         "data_grid3": data_grid2,  # توافقية رجعية: نفس قيمة data_grid2 لأي كود قديم يقرأ data_grid3
         "width": ow, "height": oh, "mouse_x": mx, "mouse_y": my,
+        "is_placeholder": is_placeholder,  # True إذا فشلت الطرق الثلاث وهذه صورة "Screenshot failed"
     }
 
 
@@ -1060,7 +1073,17 @@ 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 result["data"]:
+            # ── مساهمة في عدّاد الفشل المشترك مع shot_explicit ──
+            # shot_bg تبقى صامتة عند الفشل (لا ترسل شيئاً — هذا سلوكها الأصلي
+            # المقصود، فهي مجرد محاولة إضافية سريعة وليست خط الدفاع الموثوق)،
+            # لكنها الآن تُحدّث نفس عدّاد الفشل المتتالي الذي يستخدمه
+            # shot_explicit. هذا يعني: لو الـ display معطوب فعلاً، العدّاد
+            # يبدأ بالارتفاع من أول لقطة تلقائية فاشلة، فتصل shot_explicit
+            # (التي دائماً تُطلَب صراحة بعدها من العميل) للعتبة وتُفعّل الـ
+            # reset التلقائي أسرع، بدل الانتظار حتى تتراكم فشلتان داخل
+            # shot_explicit نفسها فقط.
+            if result.get("data") and not result.get("is_placeholder"):
+                sess["consecutive_capture_failures"] = 0
                 fh = _frame_hash(result["data"])
                 if fh != sess.get("last_bg_hash", ""):
                     sess["last_bg_hash"] = fh
@@ -1078,6 +1101,8 @@ async def handle_action(ws: WebSocket, msg: dict, sess: dict):
                         "mouse_y": result["mouse_y"],
                         "has_grid": True,
                     })
+            else:
+                sess["consecutive_capture_failures"] = sess.get("consecutive_capture_failures", 0) + 1
 
             if extra_shot_delay:
                 await asyncio.sleep(extra_shot_delay)
@@ -1104,42 +1129,127 @@ async def handle_action(ws: WebSocket, msg: dict, sess: dict):
             # لقطة خلفية فاشلة لا يجب أن توقف الجلسة أو تظهر بصمت في اللوق فقط.
             print(f"[shot_bg:{display}] ⚠️ {e}")
 
+    # ── عتبة الفشل المتتالي قبل تفعيل reset تلقائي كامل للـ display ──
+    # فشلة واحدة عابرة (ضغط CPU لحظي، تزاحم مع shot_bg) لا تستدعي reset —
+    # لكن فشلتان متتاليتان فعليتان (كلتاهما رجعتا is_placeholder=True) تعني
+    # أن هناك مشكلة حقيقية في الـ display نفسه (Xvfb ميت/عالق)، وليس عطلاً
+    # عابراً، فنعيد ضبط الكمبيوتر تلقائياً بدل الاستسلام وإرسال "error".
+    _AUTO_RESET_FAILURE_THRESHOLD = 2
+    # حد أقصى لعدد جولات (محاولة التقاط → عند الفشل المتكرر → reset → إعادة
+    # محاولة) داخل نفس طلب screenshot الواحد، لضمان أن الحلقة تنتهي دائماً
+    # ولو استمر الفشل حتى بعد عدة عمليات reset (بيئة معطوبة جذرياً)، بدل أن
+    # تدور للأبد بصمت. هذا ليس "timeout" زمنياً — هو سقف على عدد المحاولات.
+    _MAX_CAPTURE_ROUNDS = 6
+
     async def shot_explicit(label: str = ""):
         """screenshot صريح — يُرسل دائماً بدون delta suppression، وله أولوية مطلقة
         على أي عملية shot_bg تلقائية جارية (لا ينتظر خلفها طويلاً).
-        ── إصلاح جذري: try/except شامل + إرسال رسالة خطأ صريحة للعميل ──
-        المشكلة القديمة: هذه الدالة تُشغَّل عبر asyncio.create_task بدون أي
-        معالجة استثناء محيطة بها. أي خطأ غير متوقع كان يُسقط الـ task بالكامل
-        بصمت تام — لا رسالة تصل للعميل ولا حتى سطر في اللوق، فيبقى العميل
-        ينتظر frame لن يصل أبداً. الآن أي فشل يُسجَّل ويُرسَل كرسالة "error"
-        صريحة، والعميل (index.html) يتعامل معها ويعيد المحاولة فوراً."""
+
+        ── إصلاح جذري v14: لا "error" نهائي بعد الآن — بل reset تلقائي كامل ──
+        سابقاً: فشل التقاط واحد (نتيجة فارغة) كان يُرسل "error" فوراً للعميل،
+        فيبقى العميل يعيد طلب screenshot بلا نهاية بينما السيرفر لا يفعل شيئاً
+        مختلفاً في كل مرة (نفس الـ display المعطوب، بلا أي محاولة إصلاح فعلية).
+
+        الآن: فشل واحد → إعادة محاولة فورية (قد يكون عابراً). فشلان متتاليان
+        فعليان (كلاهما is_placeholder=True) → reset تلقائي كامل لهذا المستخدم
+        (نفس ما يفعله المستخدم يدوياً بزر "إعادة الضبط") ثم إعادة المحاولة
+        تلقائياً على الـ display الجديد النظيف — كل هذا داخل نفس الطلب، بلا
+        أي تدخل من العميل. رسالة "error" الصريحة لم تعد تُرسَل أبداً من هذا
+        المسار؛ العميل يستقبل إما screenshot ناجحاً حقيقياً، أو (في أسوأ حالة
+        نادرة جداً: بيئة معطوبة جذرياً حتى بعد عدة محاولات reset) رسالة
+        "computer_reset" تخبره أن الجلسة بدأت من جديد ليطلب screenshot تالياً
+        بنفسه — أبداً حالة صمت لا نهائية.
+        """
+        for _round in range(_MAX_CAPTURE_ROUNDS):
+            try:
+                result = await _priority_capture(0.65, 75)
+                got_real_frame = bool(result and result.get("data") and not result.get("is_placeholder"))
+
+                if got_real_frame:
+                    sess["consecutive_capture_failures"] = 0
+                    sess["last_bg_hash"] = _frame_hash(result["data"])
+                    await send({
+                        "type": "screenshot",
+                        "data": result["data"],
+                        "data_grid": result.get("data_grid", ""),
+                        "data_grid2": result.get("data_grid2", ""),
+                        "data_grid3": result.get("data_grid3", ""),
+                        "ts": int(time.time() * 1000),
+                        "auto": False, "label": label,
+                        "screen_width": result["width"],
+                        "screen_height": result["height"],
+                        "mouse_x": result["mouse_x"],
+                        "mouse_y": result["mouse_y"],
+                        "has_grid": True,
+                    })
+                    return
+
+                # ── فشل هذه الجولة (نتيجة فارغة تماماً، أو placeholder) ──
+                sess["consecutive_capture_failures"] = sess.get("consecutive_capture_failures", 0) + 1
+                fail_count = sess["consecutive_capture_failures"]
+                print(f"[shot_explicit:{display}] ⚠️ capture failed (round {_round+1}/{_MAX_CAPTURE_ROUNDS}, "
+                      f"consecutive={fail_count})")
+
+                if fail_count >= _AUTO_RESET_FAILURE_THRESHOLD:
+                    # ── مشكلة حقيقية وليست عابرة — reset تلقائي كامل ──
+                    print(f"[shot_explicit:{display}] 🔄 auto-reset triggered after {fail_count} consecutive failures")
+                    await send({
+                        "type": "computer_reset",
+                        "msg": "🔄 إعادة ضبط تلقائية بعد فشل التقاط متكرر — الشاشة تبدأ من جديد نظيفة",
+                        "auto": True,
+                    })
+                    await reset_user_computer(sess["user_id"])
+                    # بعد الـ reset الكامل، أمهل Xvfb الجديد لحظة قصيرة ليستقر
+                    # قبل أول محاولة التقاط عليه (نفس المهلة المستخدمة في مسار
+                    # reset_computer اليدوي أعلاه لضمان اتساق السلوك).
+                    await asyncio.sleep(2.0)
+                    # الجولة التالية من هذه الحلقة نفسها ستعيد المحاولة على
+                    # الـ display الجديد النظيف تلقائياً — بلا حاجة لأي طلب
+                    # إضافي من العميل ولا لإرسال "error" له في الأثناء.
+                    continue
+
+                # فشل عابر فقط (أول فشلة) — أعد المحاولة فوراً بلا reset
+                await asyncio.sleep(0.4)
+                continue
+
+            except Exception as e:
+                # استثناء غير متوقع تماماً (وليس مجرد نتيجة فارغة) — يُعامَل
+                # بنفس م��طق الفشل أعلاه بدل الاستسلام الفوري، لأن أسباباً
+                # كهذه (اتصال X عالق، انهيار عملية) هي بالضبط ما يُصلحه reset.
+                sess["consecutive_capture_failures"] = sess.get("consecutive_capture_failures", 0) + 1
+                fail_count = sess["consecutive_capture_failures"]
+                print(f"[shot_explicit:{display}] ❌ EXCEPTION (round {_round+1}/{_MAX_CAPTURE_ROUNDS}, "
+                      f"consecutive={fail_count}): {e}")
+                if fail_count >= _AUTO_RESET_FAILURE_THRESHOLD:
+                    try:
+                        await send({
+                            "type": "computer_reset",
+                            "msg": "🔄 إعادة ضبط تلقائية بعد خطأ متكرر في الالتقاط — الشاشة تبدأ من جديد نظيفة",
+                            "auto": True,
+                        })
+                    except Exception:
+                        pass
+                    try:
+                        await reset_user_computer(sess["user_id"])
+                        await asyncio.sleep(2.0)
+                    except Exception as reset_ex:
+                        print(f"[shot_explicit:{display}] ❌ reset itself failed: {reset_ex}")
+                continue
+
+        # ── وصلنا لحد _MAX_CAPTURE_ROUNDS جولة دون أي نجاح حقيقي، حتى بعد
+        # محاولات reset متعددة — بيئة معطوبة جذرياً (نادر جداً). بدل الصمت
+        # التام، نُعلم العميل صراحة أن الجلسة أُعيد ضبطها ليطلب screenshot
+        # تالياً بنفسه بدل انتظار frame لن يصل أبداً في هذا الطلب تحديداً.
+        print(f"[shot_explicit:{display}] ❌ giving up after {_MAX_CAPTURE_ROUNDS} rounds — "
+              f"environment appears broken even after reset attempts")
         try:
-            result = await _priority_capture(0.65, 75)
-            if not result or not result["data"]:
-                print(f"[shot_explicit:{display}] ⚠️ empty capture result")
-                await send({"type": "error", "action": "screenshot", "msg": "capture returned empty"})
-                return
-            sess["last_bg_hash"] = _frame_hash(result["data"])
             await send({
-                "type": "screenshot",
-                "data": result["data"],
-                "data_grid": result.get("data_grid", ""),
-                "data_grid2": result.get("data_grid2", ""),
-                "data_grid3": result.get("data_grid3", ""),
-                "ts": int(time.time() * 1000),
-                "auto": False, "label": label,
-                "screen_width": result["width"],
-                "screen_height": result["height"],
-                "mouse_x": result["mouse_x"],
-                "mouse_y": result["mouse_y"],
-                "has_grid": True,
+                "type": "computer_reset",
+                "msg": "⚠️ استمرت مشكلة الالتقاط رغم إعادة الضبط — الرجاء طلب لقطة شاشة جديدة",
+                "auto": True,
             })
-        except Exception as e:
-            print(f"[shot_explicit:{display}] ❌ EXCEPTION: {e}")
-            try:
-                await send({"type": "error", "action": "screenshot", "msg": str(e)})
-            except Exception:
-                pass
+        except Exception:
+            pass
 
     # ── reset_computer: إعادة ضبط الكمبيوتر كأنه جديد ─────────
     if action == "reset_computer":