Update main.py

main.py

@@ -1,6 +1,7 @@
 import os
 import uuid
 import time
+import random
 import asyncio
 import threading
 import json
@@ -11,8 +12,9 @@ from fastapi.responses import JSONResponse
 # ====================================================================
 # Configuration
 # ====================================================================
-API_SECRET_KEY  = os.getenv("API_SECRET_KEY", "change-me-secret")
-POOL_SIZE       = int(os.getenv("POOL_SIZE", "3"))   # عدد المتصفحات المتوازية
+API_SECRET_KEY = os.getenv("API_SECRET_KEY", "change-me-secret")
+POOL_SIZE      = int(os.getenv("POOL_SIZE", "3"))
+MAX_REQUESTS   = int(os.getenv("MAX_REQUESTS", "50"))   # تجديد context كل N طلب
 
 DUCK_MODELS = {
     "gpt-5-mini":       "GPT-5 mini",
@@ -28,29 +30,42 @@ DUCK_MODELS = {
 ALL_MODELS    = list(DUCK_MODELS.keys())
 DEFAULT_MODEL = "gpt-5-mini"
 
+USER_AGENTS = [
+    "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/124.0.0.0 Safari/537.36",
+    "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/123.0.0.0 Safari/537.36",
+    "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/124.0.0.0 Safari/537.36",
+    "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/123.0.0.0 Safari/537.36",
+    "Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/124.0.0.0 Safari/537.36",
+    "Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/122.0.0.0 Safari/537.36",
+]
+
 
 # ====================================================================
-# Single Browser Worker
+# Browser Worker
 # ====================================================================
 
 class BrowserWorker:
-    """
-    متصفح مستقل كامل — context + قبول الشروط.
-    كل worker يعمل بشكل مستقل تماماً عن البقية.
-    """
 
     def __init__(self, worker_id: int, loop: asyncio.AbstractEventLoop):
-        self.id      = worker_id
-        self.loop    = loop
-        self.context = None
-        self.busy    = False
-        self._lock   = asyncio.Lock()
+        self.id             = worker_id
+        self.loop           = loop
+        self.context        = None
+        self._browser       = None
+        self.busy           = False
+        self._lock          = asyncio.Lock()
+        self._request_count = 0
+        self._total_count   = 0
 
     def tag(self, msg: str) -> str:
         return f"[W{self.id}] {msg}"
 
+    # ──────────────────────────────────────────────
+    # Init
+    # ──────────────────────────────────────────────
+
     async def init(self, playwright):
-        browser = await playwright.chromium.launch(
+        self._playwright = playwright
+        self._browser = await playwright.chromium.launch(
             headless=True,
             channel="chrome",
             args=[
@@ -60,22 +75,25 @@ class BrowserWorker:
                 "--single-process", "--no-zygote",
             ],
         )
-        self.context = await browser.new_context(
-            user_agent=(
-                "Mozilla/5.0 (Windows NT 10.0; Win64; x64) "
-                "AppleWebKit/537.36 (KHTML, like Gecko) "
-                "Chrome/124.0.0.0 Safari/537.36"
-            ),
+        await self._create_context()
+        await self._accept_terms()
+        print(self.tag("Ready ✓"))
+
+    async def _create_context(self, ua: str = None):
+        user_agent = ua or random.choice(USER_AGENTS)
+        self.context = await self._browser.new_context(
+            user_agent=user_agent,
             viewport={"width": 1920, "height": 1080},
         )
         await self.context.add_init_script(
             "Object.defineProperty(navigator,'webdriver',{get:()=>undefined})"
         )
+        print(self.tag(f"Context created (UA: ...{user_agent[-30:]}) ✓"))
 
-        # قبول الشروط
+    async def _accept_terms(self):
         page = await self.context.new_page()
         try:
-            print(self.tag("Opening duck.ai for setup..."))
+            print(self.tag("Accepting terms..."))
             await page.goto("https://duckduckgo.com/aichat", wait_until="domcontentloaded")
             await asyncio.sleep(5)
             agree = page.locator('button:has-text("Agree and Continue")')
@@ -84,12 +102,30 @@ class BrowserWorker:
             print(self.tag("Terms accepted ✓"))
             await asyncio.sleep(3)
             await page.wait_for_selector('textarea[name="user-prompt"]', timeout=20000)
-            print(self.tag("Ready ✓"))
         except Exception as e:
-            print(self.tag(f"Setup note: {e}"))
+            print(self.tag(f"Terms note: {e}"))
         finally:
             await page.close()
 
+    # ───────────────────���──────────────────────────
+    # Context Rotation
+    # ──────────────────────────────────────────────
+
+    async def _rotate_context(self):
+        print(self.tag(f"Rotating context after {self._request_count} requests..."))
+        try:
+            await self.context.close()
+        except Exception:
+            pass
+        await self._create_context(ua=random.choice(USER_AGENTS))
+        await self._accept_terms()
+        self._request_count = 0
+        print(self.tag("Context rotated ✓"))
+
+    # ──────────────────────────────────────────────
+    # Chat
+    # ──────────────────────────────────────────────
+
     async def chat(self, model_label: str, prompt: str) -> str:
         async with self._lock:
             self.busy = True
@@ -99,15 +135,24 @@ class BrowserWorker:
                 self.busy = False
 
     async def _do_chat(self, model_label: str, prompt: str) -> str:
+        # ── تأخير عشوائي لتبدو الطلبات طبيعية ──────────────
+        await asyncio.sleep(random.uniform(0.5, 2.0))
+
+        # ── تجديد context دورياً ────────────────────────────
+        self._request_count += 1
+        self._total_count   += 1
+        if self._request_count >= MAX_REQUESTS:
+            await self._rotate_context()
+
         page = await self.context.new_page()
         try:
             page.set_default_timeout(180000)
 
-            # ── 1. فتح الصفحة ──────────────────────────────────
+            # ── 1. فتح الصفحة ──────────────────────────────
             await page.goto("https://duckduckgo.com/aichat", wait_until="domcontentloaded")
             await asyncio.sleep(3)
 
-            # ── 2. Popup احتياطي ────────────────────────────────
+            # ── 2. Popup احتياطي ────────────────────────────
             try:
                 agree = page.locator('button:has-text("Agree and Continue")')
                 if await agree.count() > 0 and await agree.first.is_visible():
@@ -116,17 +161,17 @@ class BrowserWorker:
             except Exception:
                 pass
 
-            # ── 3. انتظار textarea ──────────────────────────────
+            # ── 3. انتظار textarea ──────────────────────────
             await page.wait_for_selector('textarea[name="user-prompt"]')
-            print(self.tag(f"Input ready ✓"))
+            print(self.tag(f"Input ready (req #{self._total_count}) ✓"))
 
-            # ── 4. تغيير النموذج ────────────────────────────────
+            # ── 4. تغيير النموذج ────────────────────────────
             await self._select_model(page, model_label)
 
-            # ── 5. إرسال ────────────────────────────────────────
+            # ── 5. إرسال ────────────────────────────────────
             await self._send(page, prompt)
 
-            # ── 6. استخراج الرد ─────────────────────────────────
+            # ── 6. استخراج الرد ─────────────────────────────
             return await self._extract(page)
 
         except Exception as e:
@@ -135,6 +180,10 @@ class BrowserWorker:
         finally:
             await page.close()
 
+    # ──────────────────────────────────────────────
+    # Helpers
+    # ──────────────────────────────────────────────
+
     async def _select_model(self, page, model_label: str):
         try:
             btn  = page.locator('button[data-testid="model-select-button"]')
@@ -209,18 +258,16 @@ class BrowserWorker:
             await ta.click()
             await asyncio.sleep(0.3)
             await page.keyboard.press("Enter")
-            print(self.tag(f"Sent via Enter ✓"))
+            print(self.tag("Sent via Enter ✓"))
 
     async def _extract(self, page) -> str:
         await asyncio.sleep(3)
 
-        # انتظر زر Copy — يظهر فقط بعد اكتمال الرد
         try:
             copy = page.locator('button[data-copyairesponse="true"]')
             await copy.last.wait_for(state="visible", timeout=90000)
             print(self.tag("Response complete ✓"))
         except Exception:
-            # fallback: انتظر اختفاء Stop button
             for _ in range(75):
                 await asyncio.sleep(2)
                 if await page.locator(
@@ -237,6 +284,8 @@ class BrowserWorker:
                 if (active) {
                     const sp = active.querySelector('.space-y-4');
                     if (sp && sp.innerText.trim().length > 0) return sp.innerText.trim();
+                    const prose = active.querySelector('[class*="whitespace-normal"],[class*="prose"]');
+                    if (prose && prose.innerText.trim().length > 0) return prose.innerText.trim();
                 }
                 const all = document.querySelectorAll('[data-activeresponse]');
                 if (all.length > 0) {
@@ -274,10 +323,6 @@ class BrowserWorker:
 # ====================================================================
 
 class BrowserPool:
-    """
-    Pool من المتصفحات المتوازية.
-    POOL_SIZE = عدد الطلبات المتزامنة الممكنة.
-    """
 
     def __init__(self):
         self.loop        = asyncio.new_event_loop()
@@ -285,10 +330,11 @@ class BrowserPool:
         self.ready_event = threading.Event()
         self._thread     = threading.Thread(target=self._run, daemon=True)
         self._queue: asyncio.Queue | None = None
+        self._total_requests = 0
 
     def start(self):
         self._thread.start()
-        print(f"[POOL] Starting {POOL_SIZE} browsers...")
+        print(f"[POOL] Starting {POOL_SIZE} browsers (rotation every {MAX_REQUESTS} req)...")
 
     def _run(self):
         asyncio.set_event_loop(self.loop)
@@ -301,24 +347,20 @@ class BrowserPool:
         from playwright.async_api import async_playwright
         self._queue = asyncio.Queue()
         pw = await async_playwright().start()
-
-        # إنشاء كل المتصفحات بالتوازي
         workers = [BrowserWorker(i + 1, self.loop) for i in range(POOL_SIZE)]
         await asyncio.gather(*[w.init(pw) for w in workers])
-
         self.workers = workers
         for w in workers:
             await self._queue.put(w)
 
     async def _process(self, model_label: str, prompt: str) -> str:
-        # انتظر worker حر من الـ queue
+        self._total_requests += 1
         worker: BrowserWorker = await self._queue.get()
-        print(f"[POOL] Assigned W{worker.id} ✓")
+        print(f"[POOL] Assigned W{worker.id} (total req: {self._total_requests}) ✓")
         try:
-            result = await worker.chat(model_label, prompt)
-            return result
+            return await worker.chat(model_label, prompt)
         finally:
-            await self._queue.put(worker)  # أعد الـ worker للـ pool
+            await self._queue.put(worker)
             print(f"[POOL] W{worker.id} returned to pool ✓")
 
     def process(self, model_label: str, prompt: str) -> str:
@@ -539,14 +581,21 @@ async def list_models(request: Request):
 @app.get("/health")
 @app.get("/")
 async def health():
-    busy = sum(1 for w in pool.workers if w.busy)
+    busy  = sum(1 for w in pool.workers if w.busy)
+    stats = [
+        {"id": w.id, "busy": w.busy, "total_requests": w._total_count,
+         "requests_until_rotation": MAX_REQUESTS - w._request_count}
+        for w in pool.workers
+    ]
     return {
-        "status":       "running",
-        "message":      "Duck.ai API Pool Server is active!",
-        "models":       ALL_MODELS,
-        "pool_size":    POOL_SIZE,
-        "workers_busy": busy,
-        "workers_free": POOL_SIZE - busy,
+        "status":         "running",
+        "message":        "Duck.ai API Pool Server is active!",
+        "models":         ALL_MODELS,
+        "pool_size":      POOL_SIZE,
+        "workers_busy":   busy,
+        "workers_free":   POOL_SIZE - busy,
+        "total_requests": pool._total_requests,
+        "workers":        stats,
     }