Update main.py

main.py

@@ -80,24 +80,58 @@ async def auto_click_task():
     if not page:
         return
     
+    # 匹配多种大小写形式
+    text_patterns = ["Got it", "GOT IT", "Got It", "got it", "GOTIT", "Gotit"]
+    
     try:
-        selectors = [
-            'button:has-text("Got it")',
-            'a:has-text("Got it")',
-            '[role="button"]:has-text("Got it")',
-        ]
+        # 1. 先在主页面查找
+        for text in text_patterns:
+            selectors = [
+                f'button:has-text("{text}")',
+                f'a:has-text("{text}")',
+                f'[role="button"]:has-text("{text}")',
+                f'div:has-text("{text}"):not(:has(div:has-text("{text}")))',  # 最内层div
+                f'span:has-text("{text}")',
+                f'mat-button:has-text("{text}")',  # Angular Material
+                f'[data-mdc-dialog-action]:has-text("{text}")',  # Material Dialog
+            ]
+            
+            for selector in selectors:
+                try:
+                    element = page.locator(selector).first
+                    if await element.is_visible(timeout=100):
+                        await element.click(timeout=2000)
+                        print(f'[AutoClick] Clicked "{text}" button in main page')
+                        return
+                except:
+                    continue
         
-        for selector in selectors:
-            try:
-                element = page.locator(selector).first
-                if await element.is_visible(timeout=200):
-                    await element.click(timeout=1000)
-                    print(f'[AutoClick] Clicked "Got it" button')
-                    return
-            except:
+        # 2. 在所有 iframe 中查找
+        frames = page.frames
+        for frame in frames:
+            if frame == page.main_frame:
                 continue
+            
+            for text in text_patterns:
+                selectors = [
+                    f'button:has-text("{text}")',
+                    f'a:has-text("{text}")',
+                    f'[role="button"]:has-text("{text}")',
+                    f'div:has-text("{text}"):not(:has(div:has-text("{text}")))',
+                    f'span:has-text("{text}")',
+                ]
                 
-    except:
+                for selector in selectors:
+                    try:
+                        element = frame.locator(selector).first
+                        if await element.is_visible(timeout=100):
+                            await element.click(timeout=2000)
+                            print(f'[AutoClick] Clicked "{text}" button in iframe')
+                            return
+                    except:
+                        continue
+                        
+    except Exception as e:
         pass
 
 async def init_browser():
@@ -244,6 +278,113 @@ async def navigate_to(url: str):
     except Exception as e:
         return {"status": "warning", "message": f"Navigation completed with issue: {e}"}
 
+@app.get("/click")
+async def manual_click(text: str):
+    """手动点击包含指定文本的按钮"""
+    global page
+    if not page:
+        return {"status": "error", "message": "Browser not initialized"}
+    
+    try:
+        selectors = [
+            f'button:has-text("{text}")',
+            f'a:has-text("{text}")',
+            f'[role="button"]:has-text("{text}")',
+            f'div:has-text("{text}")',
+            f'span:has-text("{text}")',
+        ]
+        
+        # 主页面
+        for selector in selectors:
+            try:
+                element = page.locator(selector).first
+                if await element.is_visible(timeout=500):
+                    await element.click(timeout=3000)
+                    return {"status": "success", "message": f'Clicked "{text}" in main page'}
+            except:
+                continue
+        
+        # iframe
+        for frame in page.frames:
+            if frame == page.main_frame:
+                continue
+            for selector in selectors:
+                try:
+                    element = frame.locator(selector).first
+                    if await element.is_visible(timeout=500):
+                        await element.click(timeout=3000)
+                        return {"status": "success", "message": f'Clicked "{text}" in iframe'}
+                except:
+                    continue
+        
+        return {"status": "error", "message": f'Button with text "{text}" not found'}
+    except Exception as e:
+        return {"status": "error", "message": str(e)}
+
+@app.get("/debug/buttons")
+async def debug_buttons():
+    """调试：列出页面上所有按钮"""
+    global page
+    if not page:
+        return {"status": "error", "message": "Browser not initialized"}
+    
+    try:
+        buttons = []
+        
+        # 主页面按钮
+        button_elements = await page.locator('button, [role="button"], a[class*="button"], a[class*="btn"]').all()
+        for i, btn in enumerate(button_elements[:20]):  # 最多20个
+            try:
+                text = await btn.inner_text(timeout=500)
+                visible = await btn.is_visible(timeout=500)
+                buttons.append({
+                    "index": i,
+                    "text": text.strip()[:50],
+                    "visible": visible,
+                    "frame": "main"
+                })
+            except:
+                continue
+        
+        # iframe 中的按钮
+        for frame_idx, frame in enumerate(page.frames):
+            if frame == page.main_frame:
+                continue
+            try:
+                frame_buttons = await frame.locator('button, [role="button"]').all()
+                for i, btn in enumerate(frame_buttons[:10]):
+                    try:
+                        text = await btn.inner_text(timeout=500)
+                        visible = await btn.is_visible(timeout=500)
+                        buttons.append({
+                            "index": i,
+                            "text": text.strip()[:50],
+                            "visible": visible,
+                            "frame": f"iframe-{frame_idx}"
+                        })
+                    except:
+                        continue
+            except:
+                continue
+        
+        return {"status": "success", "count": len(buttons), "buttons": buttons}
+    except Exception as e:
+        return {"status": "error", "message": str(e)}
+
+@app.get("/debug/html")
+async def debug_html(selector: str = "body"):
+    """调试：获取元素HTML"""
+    global page
+    if not page:
+        return {"status": "error", "message": "Browser not initialized"}
+    
+    try:
+        html = await page.locator(selector).first.inner_html(timeout=5000)
+        # 截取部分
+        return {"status": "success", "html": html[:5000]}
+    except Exception as e:
+        return {"status": "error", "message": str(e)}
+
 @app.get("/health")
 async def health_check():
     """健康检查"""