Update worker.js

worker.js

@@ -36,7 +36,7 @@ async function chatWithBrowser(prompt, model) {
     const page = await browser.newPage();
     try {
         console.log(`📝 聊天请求: ${prompt.substring(0, 50)}...`);
-        await page.goto('https://toolbaz.com', { waitUntil: 'networkidle2', timeout: 30000 });
+        await page.goto('https://toolbaz.com/writer/ai-writer', { waitUntil: 'networkidle2', timeout: 30000 });
 
         // 等待输入框
         await page.waitForSelector('#input', { timeout: 10000 });
@@ -65,17 +65,59 @@ async function chatWithBrowser(prompt, model) {
     }
 }
 
-async function generateImageWithBrowser(prompt, model) {
+async function generateImageWithBrowser(prompt, model, size) {
     const browser = await initBrowser();
     const page = await browser.newPage();
     try {
         console.log(`🎨 图片生成: ${prompt.substring(0, 50)}...`);
-        await page.goto('https://toolbaz.com', { waitUntil: 'networkidle2', timeout: 30000 });
-        await page.waitForSelector('textarea, input[type="text"]', { timeout: 10000 });
-        await page.type('textarea, input[type="text"]', prompt);
-        await page.click('button[type="submit"], button:has-text("Generate")');
-        await page.waitForSelector('img[src*="generated"]', { timeout: 120000 });
-        const imageUrl = await page.evaluate(() => document.querySelector('img[src*="generated"]')?.src || null);
+        await page.goto('https://toolbaz.com/image/ai-image-generator', { waitUntil: 'networkidle2', timeout: 30000 });
+
+        // 等待输入框
+        await page.waitForSelector('#prompt', { timeout: 10000 });
+
+        // 输入提示词
+        await page.click('#prompt', { clickCount: 3 });
+        await page.type('#prompt', prompt);
+
+        // 选择模型（如果需要）
+        if (model) {
+            try {
+                await page.select('#model', model);
+            } catch (e) {
+                console.log('⚠️ 模型选择失败，使用默认');
+            }
+        }
+
+        // 选择尺寸（如果需要）
+        if (size) {
+            try {
+                await page.select('#size', size);
+            } catch (e) {
+                console.log('⚠️ 尺寸选择失败，使用默认');
+            }
+        }
+
+        // 点击生成按钮
+        await page.click('#generat');
+
+        // 等待图片生成（等待结果区域出现图片）
+        await page.waitForSelector('img[src*="generated"], img[src*="image"]', { timeout: 120000 });
+
+        // 提取图片 URL
+        const imageUrl = await page.evaluate(() => {
+            const imgs = document.querySelectorAll('img[src*="generated"], img[src*="image"], img[src*="toolbaz"]');
+            for (let img of imgs) {
+                if (img.src && img.src.startsWith('http') && !img.src.includes('logo')) {
+                    return img.src;
+                }
+            }
+            return null;
+        });
+
+        if (!imageUrl) {
+            throw new Error('未找到生成的图片');
+        }
+
         console.log(`✅ 图片: ${imageUrl}`);
         return imageUrl;
     } catch (error) {