Update server.js

server.js

@@ -8,26 +8,25 @@ const PORT = 7860;
 const CACHE_TTL = 3600;
 const cache = new Map();
 
-// ===== تابع برای Decode کردن URL از /api/click =====
-function decodeDestUrl(clickUrl) {
-    if (!clickUrl || !clickUrl.startsWith('/api/click')) {
-        return clickUrl;
+// ===== تابع تبدیل URL برتینا به لینک واقعی =====
+function decodeBertinaUrl(bertinaUrl) {
+    if (!bertinaUrl || !bertinaUrl.startsWith('/api/click')) {
+        return bertinaUrl;
     }
+
     try {
-        const urlParams = new URL(`https://search.bertina.ir${clickUrl}`);
-        const dest = urlParams.searchParams.get('dest');
+        const url = new URL(bertinaUrl, 'https://search.bertina.ir');
+        const dest = url.searchParams.get('dest');
         if (dest) {
-            // Decode Base64
-            const decoded = Buffer.from(dest, 'base64').toString('utf-8');
-            return decoded;
+            return Buffer.from(dest, 'base64').toString('utf-8');
         }
-    } catch (e) {
-        console.error('Error decoding dest URL:', e.message);
+        return bertinaUrl;
+    } catch {
+        return bertinaUrl;
     }
-    return clickUrl;
 }
 
-// ===== استخراج نتایج معمولی با Cheerio =====
+// ===== استخراج نتایج ارگانیک =====
 async function searchBertinaOrganic(query) {
     const url = `https://search.bertina.ir/search?q=${encodeURIComponent(query)}`;
     console.log(`🔍 Fetching organic results from Bertina...`);
@@ -51,21 +50,18 @@ async function searchBertinaOrganic(query) {
 
             const link = $(el).find('h3 a');
             const title = link.text().trim();
-            let href = link.attr('href') || '';
+            const href = link.attr('href') || '';
             const domain = $(el).find('cite.not-italic').text().trim();
             const snippet = $(el).find('p.mt-1.text-sm').text().trim();
 
-            // ===== تصحیح URL =====
-            if (href) {
-                href = decodeDestUrl(href);
-            }
-
             if (title && href) {
+                // تبدیل URL به لینک واقعی
+                const realUrl = decodeBertinaUrl(href);
                 results.push({
                     title: title.slice(0, 200),
-                    url: href,
+                    url: realUrl,
                     snippet: snippet.slice(0, 300),
-                    site: domain || new URL(href).hostname.replace('www.', '')
+                    site: domain || new URL(realUrl).hostname.replace('www.', '')
                 });
             }
         });
@@ -78,7 +74,7 @@ async function searchBertinaOrganic(query) {
     }
 }
 
-// ===== استخراج پاسخ هوش مصنوعی با Playwright =====
+// ===== استخراج پاسخ هوش مصنوعی =====
 async function searchBertinaAI(query) {
     console.log(`🤖 Fetching AI answer with Playwright...`);
     let browser;
@@ -94,9 +90,9 @@ async function searchBertinaAI(query) {
             timeout: 30000
         });
 
-        // منتظر می‌مانیم تا پاسخ هوش مصنوعی بارگذاری شود
+        // منتظر می‌مانیم تا بخش پاسخ هوش مصنوعی بارگذاری شود
         try {
-            await page.waitForSelector('aside[aria-label="پاسخ هوش مصنوعی"] .text-sm', {
+            await page.waitForSelector('aside[aria-label="پاسخ هوش مصنوعی"]', {
                 timeout: 10000
             });
         } catch (e) {
@@ -104,16 +100,30 @@ async function searchBertinaAI(query) {
             return null;
         }
 
-        // استخراج متن پاسخ از <div> داخل aside
+        // استخراج متن پاسخ با سلکتور دقیق‌تر
+        // متن پاسخ داخل یک div با کلاس text-sm leading-relaxed قرار دارد
         const aiAnswer = await page.$eval(
-            'aside[aria-label="پاسخ هوش مصنوعی"] .text-sm',
+            'aside[aria-label="پاسخ هوش مصنوعی"] .text-sm.leading-relaxed',
             el => el.textContent.trim()
         ).catch(() => null);
 
-        if (aiAnswer && aiAnswer.length > 10 && !aiAnswer.includes('در حال تفکر')) {
-            console.log(`✅ AI answer extracted: ${aiAnswer.slice(0, 50)}...`);
-            return aiAnswer;
+        // اگر با سلکتور بالا پیدا نشد، از سلکتور جایگزین استفاده کن
+        let aiAnswer2 = null;
+        if (!aiAnswer || aiAnswer.length < 10) {
+            aiAnswer2 = await page.$eval(
+                'aside[aria-label="پاسخ هوش مصنوعی"] .space-y-2',
+                el => el.textContent.trim()
+            ).catch(() => null);
         }
+
+        const finalAnswer = aiAnswer || aiAnswer2;
+
+        if (finalAnswer && finalAnswer.length > 10 && !finalAnswer.includes('در حال تفکر')) {
+            console.log(`✅ AI answer extracted: ${finalAnswer.slice(0, 50)}...`);
+            return finalAnswer;
+        }
+
+        console.log('⏭️ AI answer content is empty or still loading');
         return null;
 
     } catch (error) {
@@ -166,7 +176,7 @@ async function search(query) {
     const output = {
         query,
         results,
-        ai_answer: aiAnswer || null,
+        ai_answer: aiAnswer,
         source: results.length > 0 ? 'bertina' : 'duckduckgo'
     };
 
@@ -199,7 +209,7 @@ app.get('/', (req, res) => {
     res.json({
         status: 'running',
         usage: '/search?q=YOUR_QUERY',
-        note: 'Bertina search + AI answer with Playwright'
+        note: 'Bertina search + AI answer + real URLs'
     });
 });