5i;dZddlmZddlZddlZddlZddlZddlZddlm Z ddl m Z eje ZdZdZGdd Z d d Zy) ui Flow Token Service — uses headless Playwright to automatically extract project_id and fresh reCAPTCHA tokens from labs.google/fx. This service: 1. Loads labs.google/fx with the user's cookies 2. Intercepts network requests to capture the project_id 3. Extracts reCAPTCHA tokens on-demand via grecaptcha.enterprise.execute() 4. Keeps the browser alive so tokens can be refreshed without reloading Usage: service = FlowTokenService(session_token, csrf_token) await service.start() project_id = service.project_id recaptcha_token = await service.get_fresh_recaptcha_token() await service.stop() ) annotationsN)Path)Optional(6LdsFiUsAAAAAIjVDZcuLhaHiDn5nnHVXVRQGeMVz%https://labs.google/fx/tools/image-fxcHeZdZdZ d dZd d dZd dZd dZd dZdZ y) FlowTokenServicez +$)-/3    c *Kddlm}dddd} |jd{_jjj dddgd{_j jd d d d  d{_jjdjdddddddjddddddgd{jjd{_ dfd}jjd|tj!dt"jj%t"d|d{tj!djj&t)j*dd{s< jj-dd{rtj!dj2s j5d{_s.tj!d j9d{_|d!<j2|d"<d_tj!d#xsd$t?j2|S7v7C7777E77#t.$r }tj1d|Yd}~d}~wwxYw7#t.$r }tj7d|Yd}~d}~wwxYw7#t.$r }tj7d |Yd}~d}~wwxYw#t.$rH}tA||d%<tjCd&|jEd{7Yd}~|Sd}~wwxYww)'z Launch headless browser, navigate to labs.google/fx, and extract credentials. Returns dict with: - project_id: str or None - recaptcha_token: str or None - error: str or None r)async_playwrightN)r recaptcha_tokenerrorTz --no-sandboxz---disable-blink-features=AutomationControlled)headlessargszeMozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/131.0.0.0 Safari/537.36ii )widthheight) user_agentviewportz __Secure-next-auth.session-tokenz labs.google/Lax)namevaluedomainpathsecurehttpOnlysameSitez__Host-next-auth.csrf-tokencK|j}tjd|}|r'|jd tj d d|vr |j }|rtj|}|jdxs|jdi}|jdxs|jd}|r| tj d ||jd xs|jd i}|jd }|r'| _ tj d t|yyyy#t$rYywxYww)Nz2aisandbox-pa\.googleapis\.com/v1/projects/([^/]+)/z$Captured project_id from request: %szaisandbox-pa.googleapis.com clientContextclient_context projectIdr z!Captured project_id from body: %srecaptchaContextrecaptcha_contexttokenz.Captured reCAPTCHA token from request (len=%d)) urlresearchgrouploggerinfo post_datajsonloadsgetr len Exception) requestr1matchbodydatactxpidrecapr0captured_project_idrs r on_requestz*FlowTokenService.start..on_requestps;kk I*/++a.'KK FH[\1C7&00#'::d#3D"&((?";"]txxHXZ\?]C"%''+"6"O#'',:OC"69 3 & ,OQT U%(GG,>$?$c377K^`bCcE$)IIg$6E$8= 5 & ,\^abg^h i % 8$%s+AE C$D<8E < EE EE r=zNavigating to %s... networkidle) wait_untiltimeoutzPage loaded: %saL () => { // Deep search: recursively look for projectId in __NEXT_DATA__ const nextData = document.getElementById('__NEXT_DATA__'); if (nextData) { try { const text = nextData.textContent; // Regex search for projectId pattern const match = text.match(/"projectId"\s*:\s*"([a-z0-9]+)"/); if (match) return match[1]; } catch(e) {} } // Search all script tags for projectId const scripts = document.querySelectorAll('script'); for (const s of scripts) { const t = s.textContent || ''; const match = t.match(/projectId['"\s:=]+['"]([a-z0-9]{10,})['"]/); if (match) return match[1]; } // Check window/global if (window.__PROJECT_ID__) return window.__PROJECT_ID__; return null; } zFound project_id in page JS: %sz(Could not extract project_id from JS: %sz%Could not extract reCAPTCHA token: %szHproject_id not found passively. Triggering a generation to capture it...z(Could not trigger project_id capture: %sr rz4Auto-setup complete: project_id=%s, has_recaptcha=%sz NOT FOUNDrzAuto-setup failed: %s)#playwright.async_apirstart_pwchromiumlaunchr new_contextr add_cookiesr r new_pageronr5r6LABS_URLgotor1asynciosleepevaluater<debugr _extract_recaptcha_tokenwarning_trigger_and_capture_project_idr rboolstrrstop)r timeout_msrresultrEerDs` @rrKzFlowTokenService.start5sb : $MW -/5577DH"&(("3"3":":"C#;#DM#'--";";I$(37 #<#DM--++>!//+" $ %:!__+" $ %-  * $}}5577DJ#'  D JJMM)Z 0 KK-x 8**//(}j/Y Y Y KK)4::>> :--" " "'P04 0C0CE1+'0+ $EGZ[ ((O262O2O2Q,QD) ' fgR040T0T0V*V'2DO#6F< (,(=(=F$ % DM KKF#2{T**+  q8 *8V Z # +4!PLL!KQOOP -R ONN#JANNO+W RNN#MqQQR !!fF7O LL0! 4))+    sn NL?J 5L? J#!,L? J&A L?J)!L?=J,>A,L?*J/+AL?1J22L?9J7J5J74 L?K%K# K%L?6L L LAL?N L?#L?&L?)L?,L?/L?2L?5J77 K KL?K  L?#K%% L.L L? LL?L L<L72L?7L<<L?? N7N ?NN N NNc@K|jsy |jjddd{|jjd td td d{}|rtjd t||S7Y#t$rtj dt j dd{7 |jjddd{7n%#t$rtj d YYywxYwYwxYw7w)z>Execute reCAPTCHA Enterprise on the page to get a fresh token.NzW() => typeof grecaptcha !== 'undefined' && typeof grecaptcha.enterprise !== 'undefined'i'rHz>grecaptcha.enterprise not found on page, trying alternative...z'() => typeof grecaptcha !== 'undefined'izNo grecaptcha found at allz async () => { try { if (typeof grecaptcha !== 'undefined' && grecaptcha.enterprise) { const token = await grecaptcha.enterprise.execute('z', {action: 'generate'}); return token; } if (typeof grecaptcha !== 'undefined' && grecaptcha.execute) { const token = await grecaptcha.execute('z', {action: 'generate'}); return token; } return null; } catch(e) { return null; } } z(Extracted fresh reCAPTCHA token (len=%d)) rwait_for_functionr<r5rZrUrVrWRECAPTCHA_SITE_KEYr6r;)rr0s rrYz)FlowTokenService._extract_recaptcha_tokens-zz **..i/  "jj))/LM_K_`ABT@TU +   "  KKBCJ O I   NN[ \--" " " jj22= 3 ;<    sD B B B -D#D$'D B 6DCD C1*C-+C10D1DDDDDDDDc:K|jsy gd}d}|D]} |jj|j}|jdd{rO|j dd{|j ddd{d}t jd |n|st jd ytjd d{gd }d}|D]v} |jj|j}|jdd{r4|j dd{d}t jd |nx|sBt jd|jjjdd{t jdtdD]Y}tjd d{|js-t jd|j|jcSt jdy777#t$rY5wxYw7u7-7#t$rY}wxYw77#t$r } t jd| Yd} ~ yd} ~ wwxYww)zSType a simple prompt and click generate to capture project_id from the API request.N)z textarea[aria-label*="prompt" i]z!textarea[placeholder*="prompt" i]zdiv[contenteditable="true"]textareazinput[type="text"]Fircza red apple on a tablei TzFilled prompt input via: %sz Could not find/fill prompt inputr*)z button[aria-label*="generate" i]zbutton[aria-label*="create" i]zbutton:has-text("Generate")zbutton:has-text("Create")zbutton[type="submit"]zbutton[aria-label*="send" i]zClicked generate button via: %sz+No generate button found, pressing Enter...Enterz*Waiting for API request with project_id...zCaptured project_id: %sz3project_id not captured after triggering generationzError triggering generation: %s)rlocatorfirst is_visibleclickfillr5r6r<rZrUrVkeyboardpressranger ) rprompt_selectorsfilledselelbutton_selectorsclickedbtn_ras rr[z0FlowTokenService._trigger_and_capture_project_id sczzG   F'  ++C066B]]4]888 hhth444 gg&>gMMM!% $A3G 9 AB--" " "  G' **,,S177C ^^D^999!iii555"& $EsK :  IJjj))//888 KKD E2Y +mmA&&&??KK 94??K??*  + NNP Qi94M! #":5! 9 '  NNrsm$#     8 $? 2S-S-l  r