Update main.py

main.py

@@ -64,7 +64,6 @@ class AsyncBrowserThread(threading.Thread):
                 "--no-zygote",
             ],
         )
-
         self.persistent_context = await self.browser.new_context(
             user_agent=(
                 "Mozilla/5.0 (Windows NT 10.0; Win64; x64) "
@@ -76,7 +75,7 @@ class AsyncBrowserThread(threading.Thread):
         await self.persistent_context.add_init_script(
             "Object.defineProperty(navigator, 'webdriver', {get: () => undefined})"
         )
-
+        # قبول الشروط مرة واحدة
         setup_page = await self.persistent_context.new_page()
         try:
             print("[SERVER] Opening duck.ai for setup...")
@@ -98,10 +97,197 @@ class AsyncBrowserThread(threading.Thread):
         finally:
             await setup_page.close()
 
+    async def _select_model(self, page, model_label: str):
+        """
+        تغيير النموذج عبر:
+        1. Ctrl+Alt+M لفتح قائمة النماذج
+        2. اختيار النموذج
+        3. الضغط على "Start chat"
+        """
+        try:
+            model_btn = page.locator('button[data-testid="model-select-button"]')
+            current_text = (await model_btn.inner_text()).strip()
+            print(f"[DUCK] Current model: {current_text}")
+
+            if model_label.lower() in current_text.lower():
+                print(f"[DUCK] Model already correct ✓")
+                return True
+
+            # فتح قائمة النماذج
+            await model_btn.click()
+            await asyncio.sleep(2)
+
+            # البحث عن النموذج في القائمة
+            # من HTML: قائمة فيها اسم النموذج كنص
+            option = page.locator(
+                f'li:has-text("{model_label}"), '
+                f'[role="option"]:has-text("{model_label}"), '
+                f'button:has-text("{model_label}")'
+            )
+
+            if await option.count() > 0:
+                await option.first.click()
+                print(f"[DUCK] Model selected: {model_label} ✓")
+                await asyncio.sleep(1)
+
+                # البحث عن زر "Start chat" بعد اختيار النموذج
+                start_btn = page.locator(
+                    'button:has-text("Start chat"), '
+                    'button:has-text("Start new chat")'
+                )
+                if await start_btn.count() > 0:
+                    await start_btn.first.click()
+                    print("[DUCK] Start chat clicked ✓")
+                    await asyncio.sleep(2)
+
+                # انتظر textarea بعد التغيير
+                await page.wait_for_selector(
+                    'textarea[name="user-prompt"]',
+                    state="visible",
+                    timeout=15000
+                )
+                print("[DUCK] Textarea ready after model change ✓")
+                return True
+            else:
+                await page.keyboard.press("Escape")
+                print(f"[DUCK] Model '{model_label}' not in list, using default")
+                return False
+
+        except Exception as e:
+            print(f"[DUCK] Model select error (non-fatal): {e}")
+            return False
+
+    async def _inject_and_send(self, page, prompt: str) -> bool:
+        """
+        حقن النص في textarea وإرساله
+        """
+        # حقن النص عبر JavaScript مع React events
+        await page.evaluate(
+            """
+            (text) => {
+                const ta = document.querySelector('textarea[name="user-prompt"]');
+                if (!ta) return;
+                const nativeSetter = Object.getOwnPropertyDescriptor(
+                    window.HTMLTextAreaElement.prototype, 'value'
+                ).set;
+                nativeSetter.call(ta, text);
+                ta.dispatchEvent(new InputEvent('input',  { bubbles: true }));
+                ta.dispatchEvent(new Event('change', { bubbles: true }));
+                ta.focus();
+            }
+            """,
+            prompt
+        )
+        await asyncio.sleep(2)
+
+        # محاولة 1: زر Submit
+        try:
+            send_btn = page.locator('button[type="submit"][aria-label="Send"]')
+            is_disabled = await send_btn.get_attribute("disabled")
+            if is_disabled is None:
+                await send_btn.click()
+                print(f"[DUCK] Sent via button ✓ ({len(prompt)} chars)")
+                return True
+        except Exception:
+            pass
+
+        # محاولة 2: Enter على الـ textarea
+        try:
+            ta = page.locator('textarea[name="user-prompt"]')
+            await ta.click()
+            await asyncio.sleep(0.3)
+            await page.keyboard.press("Enter")
+            print(f"[DUCK] Sent via Enter ✓ ({len(prompt)} chars)")
+            return True
+        except Exception as e:
+            print(f"[DUCK] Send failed: {e}")
+            return False
+
+    async def _wait_for_response(self, page) -> str:
+        """
+        انتظار اكتمال الرد واستخراجه
+        """
+        # انتظر ظهور الرد
+        await asyncio.sleep(3)
+
+        # انتظر زر Copy بـ data-copyairesponse="true"
+        # هذا يظهر فقط بعد اكتمال الرد الأخير
+        try:
+            copy_btn = page.locator('button[data-copyairesponse="true"]')
+            await copy_btn.last.wait_for(state="visible", timeout=90000)
+            print("[DUCK] Response complete (copy button appeared) ✓")
+            await asyncio.sleep(1)
+        except Exception:
+            print("[DUCK] Copy button not detected, using fallback wait...")
+            # fallback: انتظر اختفاء Stop button
+            max_wait = 120
+            elapsed  = 0
+            while elapsed < max_wait:
+                await asyncio.sleep(2)
+                elapsed += 2
+                stop_active = await page.locator(
+                    'button[aria-label="Stop generating"]:not([disabled])'
+                ).count()
+                if stop_active == 0:
+                    print(f"[DUCK] Response complete (stop gone ~{elapsed}s) ✓")
+                    break
+
+        await asyncio.sleep(1)
+
+        # استخراج النص من البنية الحقيقية
+        # من HTML: div[data-activeresponse="true"] > ... > div.space-y-4
+        response_text = await page.evaluate("""
+            () => {
+                // طريقة 1: div[data-activeresponse="true"] — البنية الحقيقية لـ duck.ai
+                const activeResp = document.querySelector('[data-activeresponse="true"]');
+                if (activeResp) {
+                    // النص في div.space-y-4 داخل الرد
+                    const textDiv = activeResp.querySelector('.space-y-4');
+                    if (textDiv && textDiv.innerText.trim().length > 0) {
+                        return textDiv.innerText.trim();
+                    }
+                    // أو النص المباشر
+                    const prose = activeResp.querySelector(
+                        '[class*="whitespace-normal"], [class*="prose"]'
+                    );
+                    if (prose && prose.innerText.trim().length > 0) {
+                        return prose.innerText.trim();
+                    }
+                }
+
+                // طريقة 2: آخر div[data-activeresponse]
+                const allResps = document.querySelectorAll('[data-activeresponse]');
+                if (allResps.length > 0) {
+                    const last = allResps[allResps.length - 1];
+                    const textDiv = last.querySelector('.space-y-4');
+                    if (textDiv) return textDiv.innerText.trim();
+                    return last.innerText.trim();
+                }
+
+                // طريقة 3: article
+                const articles = document.querySelectorAll('article');
+                if (articles.length > 0) {
+                    return articles[articles.length - 1].innerText.trim();
+                }
+
+                // طريقة 4: div.space-y-4 الأخير
+                const spaceDivs = document.querySelectorAll('.space-y-4');
+                if (spaceDivs.length > 0) {
+                    for (let i = spaceDivs.length - 1; i >= 0; i--) {
+                        const text = spaceDivs[i].innerText.trim();
+                        if (text.length > 10) return text;
+                    }
+                }
+
+                return '';
+            }
+        """)
+
+        return response_text.strip()
+
     async def _chat(self, model_label: str, prompt: str) -> str:
         page = await self.persistent_context.new_page()
         try:
-            # !! timeout واحد كبير لكل الصفحة — لا يُغيَّر أبداً
             page.set_default_timeout(180000)
 
             # ── 1. فتح الصفحة ──────────────────────────────────────
@@ -125,155 +311,25 @@ class AsyncBrowserThread(threading.Thread):
             print("[DUCK] Input ready ✓")
 
             # ── 4. تغيير النموذج ───────────────────────────────────
-            try:
-                model_btn    = page.locator('button[data-testid="model-select-button"]')
-                current_text = await model_btn.inner_text()
-                print(f"[DUCK] Current model: {current_text.strip()}")
+            await self._select_model(page, model_label)
 
-                if model_label.lower() not in current_text.lower():
-                    await model_btn.click()
-                    await asyncio.sleep(2)
-
-                    option = page.locator(
-                        f"li:has-text('{model_label}'), "
-                        f"[role='option']:has-text('{model_label}')"
-                    )
-                    if await option.count() > 0:
-                        await option.first.click()
-                        print(f"[DUCK] Model → {model_label} ✓")
-                    else:
-                        await page.keyboard.press("Escape")
-                        print("[DUCK] Model not found, using default")
-
-                    await asyncio.sleep(3)
-                    await page.wait_for_selector(
-                        'textarea[name="user-prompt"]', state="visible"
-                    )
-                    print("[DUCK] Textarea re-confirmed ✓")
-
-            except Exception as e:
-                print(f"[DUCK] Model select (non-fatal): {e}")
-
-            # ── 5. حقن النص عبر JavaScript ────────────────────────
-            await page.evaluate(
-                """
-                (text) => {
-                    const ta = document.querySelector('textarea[name="user-prompt"]');
-                    if (!ta) return;
-                    const nativeSetter = Object.getOwnPropertyDescriptor(
-                        window.HTMLTextAreaElement.prototype, 'value'
-                    ).set;
-                    nativeSetter.call(ta, text);
-                    ta.dispatchEvent(new InputEvent('input',  { bubbles: true }));
-                    ta.dispatchEvent(new Event('change', { bubbles: true }));
-                    ta.focus();
-                }
-                """,
-                prompt
-            )
-            await asyncio.sleep(2)
-
-            # ── 6. إرسال الرسالة ───────────────────────────────────
-            # محاولة 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:
-                    await send_btn.click()
-                    print(f"[DUCK] Sent via button ✓")
-                    sent = True
-            except Exception:
-                pass
-
-            # محاولة 2: Enter مباشرة
+            # ── 5. إرسال الرسالة ───────────────────────────────────
+            sent = await self._inject_and_send(page, prompt)
             if not sent:
-                ta = page.locator('textarea[name="user-prompt"]')
-                await ta.click()
-                await asyncio.sleep(0.5)
-                await page.keyboard.press("Enter")
-                print(f"[DUCK] Sent via Enter ✓")
-
-            print(f"[DUCK] Prompt length: {len(prompt)} chars")
+                raise RuntimeError("Failed to send message")
 
-            # ── 7. انتظار بدء الرد ────────────────────────────────
-            await asyncio.sleep(3)
-            response_started = False
-            try:
-                stop_btn = page.locator('button[aria-label="Stop generating"]')
-                await stop_btn.wait_for(state="visible", timeout=30000)
-                print("[DUCK] Response started ✓")
-                response_started = True
-            except Exception:
-                print("[DUCK] Stop btn not detected, watching for content...")
-
-            # ── 8. انتظار اكتمال الرد ─────────────────────────────
-            max_wait = 150
-            elapsed  = 0
-
-            if response_started:
-                while elapsed < max_wait:
-                    await asyncio.sleep(2)
-                    elapsed += 2
-                    still_running = await page.locator(
-                        'button[aria-label="Stop generating"]:not([disabled])'
-                    ).count()
-                    if still_running == 0:
-                        print(f"[DUCK] Complete (~{elapsed}s) ✓")
-                        break
-            else:
-                # انتظر ظهور article
-                while elapsed < max_wait:
-                    await asyncio.sleep(2)
-                    elapsed += 2
-                    if await page.locator('article').count() > 0:
-                        await asyncio.sleep(6)
-                        print(f"[DUCK] Article appeared (~{elapsed}s) ✓")
-                        break
-
-            await asyncio.sleep(2)
-
-            # ── 9. استخراج الرد ───────────────────────────────────
-            response_text = await page.evaluate("""
-                () => {
-                    const articles = document.querySelectorAll('article');
-                    if (articles.length > 0) {
-                        return articles[articles.length - 1].innerText.trim();
-                    }
-                    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.querySelector('textarea')
-                        );
-                        if (valid.length > 0) {
-                            return valid[valid.length - 1].innerText.trim();
-                        }
-                    }
-                    const divs = [...document.querySelectorAll('div')].filter(el =>
-                        el.children.length < 10 &&
-                        el.innerText && el.innerText.trim().length > 50 &&
-                        !el.querySelector('textarea') &&
-                        !el.querySelector('button[type="submit"]')
-                    );
-                    if (divs.length > 0) {
-                        return divs.sort(
-                            (a, b) => b.innerText.length - a.innerText.length
-                        )[0].innerText.trim();
-                    }
-                    return '';
-                }
-            """)
+            # ── 6. انتظار واستخراج الرد ────────────────────────────
+            response_text = await self._wait_for_response(page)
 
-            if not response_text or len(response_text.strip()) < 10:
+            # fallback إذا كان فارغاً
+            if not response_text or len(response_text.strip()) < 5:
                 await asyncio.sleep(5)
                 response_text = await page.evaluate("""
                     () => {
                         const arts = document.querySelectorAll('article');
                         if (arts.length > 0) return arts[arts.length-1].innerText.trim();
+                        const sp = document.querySelectorAll('.space-y-4');
+                        if (sp.length > 0) return sp[sp.length-1].innerText.trim();
                         return document.body.innerText.slice(0, 5000);
                     }
                 """)