Update main.py

main.py

@@ -77,7 +77,6 @@ class AsyncBrowserThread(threading.Thread):
             "Object.defineProperty(navigator, 'webdriver', {get: () => undefined})"
         )
 
-        # قبول الشروط مرة واحدة عند الإقلاع
         setup_page = await self.persistent_context.new_page()
         try:
             print("[SERVER] Opening duck.ai for setup...")
@@ -102,9 +101,10 @@ class AsyncBrowserThread(threading.Thread):
     async def _chat(self, model_label: str, prompt: str) -> str:
         page = await self.persistent_context.new_page()
         try:
-            page.set_default_timeout(30000)
+            # !! timeout واحد كبير لكل الصفحة — لا يُغيَّر أبداً
+            page.set_default_timeout(180000)
 
-            # ── 1. فتح صفحة جديدة ──────────────────────────────────
+            # ── 1. فتح الصفحة ──────────────────────────────────────
             await page.goto(
                 "https://duckduckgo.com/aichat", wait_until="domcontentloaded"
             )
@@ -121,12 +121,10 @@ class AsyncBrowserThread(threading.Thread):
                 pass
 
             # ── 3. انتظار textarea ─────────────────────────────────
-            await page.wait_for_selector(
-                'textarea[name="user-prompt"]', timeout=30000
-            )
+            await page.wait_for_selector('textarea[name="user-prompt"]')
             print("[DUCK] Input ready ✓")
 
-            # ── 4. تغيير النموذج بطريقة آمنة ──────────────────────
+            # ── 4. تغيير النموذج ───────────────────────────────────
             try:
                 model_btn    = page.locator('button[data-testid="model-select-button"]')
                 current_text = await model_btn.inner_text()
@@ -145,27 +143,18 @@ class AsyncBrowserThread(threading.Thread):
                         print(f"[DUCK] Model → {model_label} ✓")
                     else:
                         await page.keyboard.press("Escape")
-                        print(f"[DUCK] Model not found, using default")
+                        print("[DUCK] Model not found, using default")
 
-                    # انتظر إعادة render بعد تغيير النموذج
                     await asyncio.sleep(3)
                     await page.wait_for_selector(
-                        'textarea[name="user-prompt"]',
-                        state="visible",
-                        timeout=15000
+                        'textarea[name="user-prompt"]', state="visible"
                     )
                     print("[DUCK] Textarea re-confirmed ✓")
 
             except Exception as e:
                 print(f"[DUCK] Model select (non-fatal): {e}")
-                await page.wait_for_selector(
-                    'textarea[name="user-prompt"]',
-                    state="visible",
-                    timeout=10000
-                )
-
-            # ── 5. كتابة الرسالة بـ JavaScript ────────────────────
-            # يتجاوز React's synthetic events ويضمن تفعيل زر Send
+
+            # ── 5. حقن النص عبر JavaScript ────────────────────────
             await page.evaluate(
                 """
                 (text) => {
@@ -175,40 +164,40 @@ class AsyncBrowserThread(threading.Thread):
                         window.HTMLTextAreaElement.prototype, 'value'
                     ).set;
                     nativeSetter.call(ta, text);
-                    ta.dispatchEvent(new InputEvent('input', { bubbles: true }));
+                    ta.dispatchEvent(new InputEvent('input',  { bubbles: true }));
                     ta.dispatchEvent(new Event('change', { bubbles: true }));
                     ta.focus();
                 }
                 """,
                 prompt
             )
-            await asyncio.sleep(1.5)
+            await asyncio.sleep(2)
 
             # ── 6. إرسال الرسالة ───────────────────────────────────
-            send_btn = page.locator('button[type="submit"][aria-label="Send"]')
-
-            # انتظر حتى 5 ثوان لزر Send
-            send_enabled = False
-            for _ in range(10):
+            # محاولة 1: زر Submit
+            sent = False
+            try:
+                send_btn = page.locator('button[type="submit"][aria-label="Send"]')
                 is_disabled = await send_btn.get_attribute("disabled")
                 if is_disabled is None:
-                    send_enabled = True
-                    break
-                await asyncio.sleep(0.5)
+                    await send_btn.click()
+                    print(f"[DUCK] Sent via button ✓")
+                    sent = True
+            except Exception:
+                pass
 
-            if send_enabled:
-                await send_btn.click()
-                print(f"[DUCK] Sent via button ✓ ({len(prompt)} chars)")
-            else:
-                # fallback: Enter على الـ textarea
+            # محاولة 2: Enter مباشرة
+            if not sent:
                 ta = page.locator('textarea[name="user-prompt"]')
                 await ta.click()
-                await asyncio.sleep(0.3)
+                await asyncio.sleep(0.5)
                 await page.keyboard.press("Enter")
-                print(f"[DUCK] Sent via Enter ✓ ({len(prompt)} chars)")
+                print(f"[DUCK] Sent via Enter ✓")
+
+            print(f"[DUCK] Prompt length: {len(prompt)} chars")
 
             # ── 7. انتظار بدء الرد ────────────────────────────────
-            await asyncio.sleep(2)
+            await asyncio.sleep(3)
             response_started = False
             try:
                 stop_btn = page.locator('button[aria-label="Stop generating"]')
@@ -216,11 +205,10 @@ class AsyncBrowserThread(threading.Thread):
                 print("[DUCK] Response started ✓")
                 response_started = True
             except Exception:
-                print("[DUCK] Stop btn not detected, will watch for content...")
+                print("[DUCK] Stop btn not detected, watching for content...")
 
             # ── 8. انتظار اكتمال الرد ─────────────────────────────
-            page.set_default_timeout(120000)
-            max_wait = 120
+            max_wait = 150
             elapsed  = 0
 
             if response_started:
@@ -231,17 +219,16 @@ class AsyncBrowserThread(threading.Thread):
                         'button[aria-label="Stop generating"]:not([disabled])'
                     ).count()
                     if still_running == 0:
-                        print(f"[DUCK] Response complete (~{elapsed}s) ✓")
+                        print(f"[DUCK] Complete (~{elapsed}s) ✓")
                         break
             else:
-                # انتظر ظهور article (محتوى الرد)
+                # انتظر ظهور article
                 while elapsed < max_wait:
                     await asyncio.sleep(2)
                     elapsed += 2
-                    arts = await page.locator('article').count()
-                    if arts > 0:
-                        await asyncio.sleep(5)
-                        print(f"[DUCK] Content appeared (~{elapsed}s) ✓")
+                    if await page.locator('article').count() > 0:
+                        await asyncio.sleep(6)
+                        print(f"[DUCK] Article appeared (~{elapsed}s) ✓")
                         break
 
             await asyncio.sleep(2)
@@ -249,31 +236,26 @@ class AsyncBrowserThread(threading.Thread):
             # ── 9. استخراج الرد ───────────────────────────────────
             response_text = await page.evaluate("""
                 () => {
-                    // طريقة 1: article
                     const articles = document.querySelectorAll('article');
                     if (articles.length > 0) {
                         return articles[articles.length - 1].innerText.trim();
                     }
-                    // طريقة 2: class يحتوي message
                     const msgDivs = document.querySelectorAll(
                         '[class*="message"]:not([class*="user"])' +
                         ':not([class*="User"]):not([class*="input"])'
                     );
                     if (msgDivs.length > 0) {
                         const valid = [...msgDivs].filter(el =>
-                            el.innerText &&
-                            el.innerText.trim().length > 20 &&
+                            el.innerText && el.innerText.trim().length > 20 &&
                             !el.querySelector('textarea')
                         );
                         if (valid.length > 0) {
                             return valid[valid.length - 1].innerText.trim();
                         }
                     }
-                    // طريقة 3: أطول div نظيف
                     const divs = [...document.querySelectorAll('div')].filter(el =>
                         el.children.length < 10 &&
-                        el.innerText &&
-                        el.innerText.trim().length > 50 &&
+                        el.innerText && el.innerText.trim().length > 50 &&
                         !el.querySelector('textarea') &&
                         !el.querySelector('button[type="submit"]')
                     );
@@ -286,7 +268,6 @@ class AsyncBrowserThread(threading.Thread):
                 }
             """)
 
-            # fallback نهائي
             if not response_text or len(response_text.strip()) < 10:
                 await asyncio.sleep(5)
                 response_text = await page.evaluate("""
@@ -312,7 +293,7 @@ class AsyncBrowserThread(threading.Thread):
         future = asyncio.run_coroutine_threadsafe(
             self._chat(model_label, prompt), self.loop
         )
-        return future.result(timeout=200)
+        return future.result(timeout=210)
 
 
 browser_engine = AsyncBrowserThread()
@@ -344,7 +325,9 @@ def _build_prompt(messages: list) -> str:
         if not content.strip():
             continue
         if role == "system":
-            parts.append(f"=== SYSTEM INSTRUCTIONS ===\n{content}\n=== END INSTRUCTIONS ===")
+            parts.append(
+                f"=== SYSTEM INSTRUCTIONS ===\n{content}\n=== END INSTRUCTIONS ==="
+            )
         elif role == "assistant":
             parts.append(f"[Assistant]: {content}")
         else: