Update app.py

app.py

@@ -1,9 +1,9 @@
 from fastapi import FastAPI
 from playwright.async_api import async_playwright
+import base64
 
 app = FastAPI()
 
-# Your 2026 Token
 BEARER = "eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJpZCI6IjY5M2NmYTBkMGU4YzlmMzdiZTA4Y2IyMCIsImlhdCI6MTczMjM5NTc5OSwiZXhwIjoxNzczMDAwNTk5fQ.Mc-LS_NMe0lpBQrQE4DEGs83iVBHIp8LvzBdeaUbmHo"
 
 @app.get("/")
@@ -19,7 +19,6 @@ async def do_react(url: str, emojis: str):
                 user_agent="Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36"
             )
             
-            # 🟢 PROPER COOKIE INJECTION (Forces the browser to be logged in)
             await context.add_cookies([{
                 "name": "ain",
                 "value": BEARER,
@@ -28,37 +27,32 @@ async def do_react(url: str, emojis: str):
             }])
             
             page = await context.new_page()
-            
-            # Go to the website
             await page.goto("https://asitha.top/channel-react", wait_until="networkidle")
             
-            # Wait for the Pay button to appear (Proves we successfully bypassed login)
             try:
+                # Wait for the button. If it doesn't appear in 10 seconds, it will fail and jump to the 'except' block to take a screenshot.
                 await page.wait_for_selector("button:has-text('Pay 1 COIN')", timeout=10000)
-            except:
+            except Exception as e:
+                # 📸 TAKE SCREENSHOT IF IT FAILS
+                screenshot_bytes = await page.screenshot(full_page=True)
+                screenshot_b64 = base64.b64encode(screenshot_bytes).decode('utf-8')
                 await browser.close()
-                return {"success": False, "error": "Login failed. The bot is stuck on the Google Login screen."}
+                return {"success": False, "error": "Stuck on login screen.", "image": screenshot_b64}
 
-            # Grab all input boxes (Bypasses the 'type=url' bug)
             inputs = await page.locator("input").all()
-            
             if len(inputs) >= 2:
                 await inputs[0].fill(url)
-                await page.wait_for_timeout(500)
                 await inputs[1].fill(emojis)
-                await page.wait_for_timeout(500)
-
-                # Click the yellow button!
                 await page.locator("button:has-text('Pay 1 COIN')").click()
-                
-                # Wait 4 seconds for the server to process the reactions
                 await page.wait_for_timeout(4000)
-                
                 await browser.close()
                 return {"success": True, "message": "Reaction injected successfully!"}
             else:
+                # 📸 TAKE SCREENSHOT IF BOXES ARE MISSING
+                screenshot_bytes = await page.screenshot(full_page=True)
+                screenshot_b64 = base64.b64encode(screenshot_bytes).decode('utf-8')
                 await browser.close()
-                return {"success": False, "error": f"Found {len(inputs)} inputs instead of 2."}
+                return {"success": False, "error": "Could not find text boxes.", "image": screenshot_b64}
                 
     except Exception as e:
         return {"success": False, "error": str(e)}