DinY@dZddlmZddlZddlZddlZddlZddlZddlm Z ddl m Z m Z m Z ddlZejeZdZdZdZGd d ZGd d Zed k(r=ddlZej0ej2ddZej6eyy)u4 Grok Playwright Provider - Image/Video generation via UI automation. Uses Playwright in headless mode with stealth techniques to bypass Cloudflare. This is the ONLY viable approach for Grok automation without paying for the API. Auth: Cookie-based (sso, sso-rw, x-userid JWTs from grok.com). Flow: Navigate to Imagine → Type prompt → Wait for images → Click "Make video" → Download. Env vars: GROK_SSO_TOKEN - 'sso' cookie value (JWT) GROK_SSO_RW_TOKEN - 'sso-rw' cookie value (JWT) GROK_USER_ID - 'x-userid' cookie value (UUID) ) annotationsN)Path)CallableOptionalListzhttps://grok.comi,xceZdZdZd d dZdZ d ddZ d ddZdddZddZ d Z d Z d Z y)GrokPlaywrightProviderzLGenerates images and videos via Grok's Imagine UI using Playwright headless.c|xstjdd|_|xstjdd|_|xstjdd|_||_t |j|j|jgstjdd|_ d|_ d|_ d|_ tjdd|_d |_y) NGROK_SSO_TOKENGROK_SSO_RW_TOKEN GROK_USER_IDzGrok credentials not fully set (sso_token/sso_rw_token/user_id). Browser operations will fail until valid credentials are provided.rT)timeoutfollow_redirectsHhttp://gabslocked:Pryl02123_country-us_streaming-1@geo.iproyal.com:12321)osgetenv sso_token sso_rw_tokenuser_idheadlessallloggerwarning _playwright_browser_context_pagehttpxClientclient proxy_urlselfrrrrs K/root/.openclaw/workspace/visionaryfx/providers/grok_playwright_provider.py__init__zGrokPlaywrightProvider.__init__&s"Ebii0@"&E(NBII6I2,N?"))NB"?   DNND$5$5t||DE NNU      ll!  dc K|jyddlm}|jd{|_|jj j |jgdd{|_|jjdddd d d d d ddd{|_ |jjd|jdddddd|jdddddd|jdddddgd{|jjd{|_|jj!dd{t"j%d|jy7B777w7W71w)z1Launch Playwright browser if not already running.Nr)async_playwright)z---disable-blink-features=AutomationControlledz--disable-dev-shm-usagez --no-sandboxz--disable-setuid-sandboxz--disable-infobarsz--window-size=1920,1080)rargsii8)widthheighteMozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/144.0.0.0 Safari/537.36zen-USzAmerica/Sao_Paulozhttp://geo.iproyal.com:12321 gabslockedz Pryl02123_country-us_streaming-1)serverusernamepassword)viewport user_agentlocale timezone_idproxyssoz .grok.com/T)namevaluedomainpathsecurehttpOnlyzsso-rwzx-userida: // Remove webdriver property Object.defineProperty(navigator, 'webdriver', { get: () => undefined }); // Mock plugins Object.defineProperty(navigator, 'plugins', { get: () => [1, 2, 3, 4, 5] }); // Mock languages Object.defineProperty(navigator, 'languages', { get: () => ['en-US', 'en'] }); // Mock permissions const originalQuery = window.navigator.permissions.query; window.navigator.permissions.query = (parameters) => ( parameters.name === 'notifications' ? Promise.resolve({ state: Notification.permission }) : originalQuery(parameters) ); z&Playwright browser ready (headless=%s))rplaywright.async_apir*startrchromiumlaunchrr new_contextr add_cookiesrrrnew_pageadd_init_scriptrinfo)r%r*s r&_ensure_browserz&GrokPlaywrightProvider._ensure_browser>s :: ! 9!1!3!9!9!;;#..77>>]]?    #mm77#t4~+;dFG 8   mm''%  !**%  #%  #)    6 ==1133 jj((*    .  E0?E>.E3/3E>"E6#AE>E8!E>"E:#'E> E< &E>3E>6E>8E>:E><E>NcvK|jd{|j}t|}|jddtj d|dd|j tddd{|jd d{tjd d{|r|d |jd j}|jdd d{tjdd{|jjdsd|n|}|j j#|dd{tjdd{|r|d|j j%dd{tj dt'j&}g} t'j&|z t(krtjd d{t+t'j&|z } |j-dd{} | rPt/| dk\rBt1t2j5| d|} tj dt/| | no|r/| d kDr*t7ddt+| t(z dzz} || | dzdk(rtj d| t'j&|z t(kr| st9dt(d |r|d!g} t;| D]R\}}|d"|d#zd$z }|j=||d{| j?|tj d%|T|r|d&| S7777777O757 77q7Zw)'ax Generate images using Grok Imagine UI. Args: prompt: Text prompt for image generation output_dir: Directory to save generated images num_images: Number of images to download (max available) on_progress: Optional callback for progress updates Returns: List of paths to saved images NTparentsexist_okzGenerating images: %s<r9`rdomcontentloadedzdiv[contenteditable="true"]')forcer333333?generatezGenerate an image of )delay? Enterz'Prompt submitted, waiting for images...a  () => { return Array.from(document.querySelectorAll('img')) .map(img => img.src) .filter(src => src.includes('/generated/') && src.includes('.jpg')); } zFound %d images after %dsZPrzWaiting for images... (%ds)z!Image generation timed out after s_ grok_image_z.pngz Saved: %sd) rIrrmkdirrrHgoto GROK_BASEwait_for_load_stateasynciosleeplocatorfirstclicklower startswithkeyboardtypepresstimeIMAGE_GEN_TIMEOUTintevaluatelenlistdictfromkeysmin RuntimeError enumerate_download_fileappend)r%prompt output_dir num_images on_progresspage input_area full_promptrA image_urlselapsed found_imagesprogress saved_pathsiurl output_paths r&generate_imagez%GrokPlaywrightProvider.generate_images\&""$$$zz*% 5 +VCR[9ii9+Qi777&&'9:::mmA  N\\"?@FF T5999mmC   ?Elln>W>WXb>c-fX6io mm  B 777mmC    Omm!!'*** =>  iikE!$55--" " "$))+-.G"&0"LL 1Q 6!$-- "=>{ K  7Z'Rw{r2G6G,G",L(M#MNH%|q  97C1iikE!$554!BCTBUUVWX X  O  + 2FAs$QqSE'>>K%%c;7 7 7   { + KK [ 1  2   a % 8: :  8 + #> 8sN9NA$N9;N<N9NN91N2?N91N"2N9N%A N9N(N96N+7,N9#N.$A&N9 N1 8N9N4CN9 AN9N7:N9N9N9N9"N9%N9(N9+N9.N91N94N97N9crK|jd{|j}t|}||jd}t|}|jj dd|j std|tjd|j|jtdd d{|jd d{tjd d{|r|d |j!d j"}|j%t'|j)d{tjd|r|dd}t+j*}t+j*|z dkrtjdd{|j,} t/j0d| } | r(| j3d}tjd|nt+j*|z dkr|st5d|j,tjdd{|r|d|rtjd|dd |j!dj"} | j7d{tjdd{| j9|d{tjdd{tjd|r|dtjd |j!d j"} | j7d! d{tjd d{tjd"tjd# |j!d$j"}|j7d% d{tjdd{tjd&|r|d)tjd* |jCd+d{tjd,tjd d{|r|d.tjd/tD|dd0t+j*}d}t+j*|z tDkr"tjdd{tGt+j*|z }|jCd1d{}|r[|D]}||vs|}tjd2n|s|d3}tj=d4tjd5||dd6no|r/|dkDr*tId7d.tG|tDz d8zz}|||dzd3k(rtjd9|t+j*|z tDkr"|s"t5d:tDd;|d<|j,|r|d=t/jJd>d?|}|jM||d{tjd@||r|dA|S7|7777K77G7777#t:$rV} tj=d|  |j>jA|d{7n#t:$rYnwxYwYd} ~ d} ~ wwxYw777+7#t:$r!} tj=d'| Yd} ~ d} ~ wwxYw#t:$r!} tj=d(| Yd} ~ Ld} ~ wwxYw7 #t:$r!} tj=d-| Yd} ~ .d} ~ wwxYw777q7Iw)BaX Generate video from image using Grok Imagine I2V. Args: image_path: Path to source image video_prompt: Optional motion prompt output_path: Where to save the video on_progress: Optional callback for progress updates Returns: Path to saved video file Nz.mp4TrKzImage not found: zGenerating video from: %sz/imaginerOrPrQr]rSzinput[type="file"]z4Image uploaded, waiting for redirect to post page...r[rXrcz/imagine/post/([a-f0-9-]+)zPost page loaded with token: %sz4Failed to get post token after upload. Current URL: rRzTyping video prompt: %srNz|input[placeholder*="customize video"], textarea[placeholder*="customize video"], [role="textbox"][aria-label="Make a video"]rVrZz!Video prompt entered successfullyzCould not type video prompt: %sz,Opening Video Options menu to select 720p...z"button[aria-label="Video Options"]rTzVideo Options menu openedzSelecting 720p resolution...zbutton[aria-label="720p"]izSelected 720p resolutionz!720p button not found in menu: %sz%Could not open Video Options menu: %szClicking 'Make video' button...a () => { // Find button with aria-label="Make video" const btn = document.querySelector('button[aria-label="Make video"]'); if (btn) { btn.click(); return; } // Fallback: find button containing "Make video" text const btns = document.querySelectorAll('button'); for (const b of btns) { if (b.textContent.includes('Make video') || b.textContent.includes('Redo')) { b.click(); break; } } } zClicked 'Make video' buttonz%Could not click Make video button: %sz8Waiting for video generation (timeout %ds, token: %s)...a () => { const srcs = []; document.querySelectorAll('video').forEach(v => { if (v.src && v.src.includes('/generated/') && v.src.includes('.mp4')) { srcs.push(v.src); } // Also check source elements v.querySelectorAll('source').forEach(s => { if (s.src && s.src.includes('/generated/') && s.src.includes('.mp4')) { srcs.push(s.src); } }); }); return srcs; } z Found video matching post token!rz-No video matched token, using first availablezVideo ready after %ds: %sr_raFzStill generating video... (%ds)z!Video generation timed out after zs. Post token: z , Final URL: bz \?cache=\d+$r zVideo saved: %srd)'rIrr with_suffixparentreexistsFileNotFoundErrorrrHr:rfrgrhrirjrkrlset_input_filesstrresolversrresearchgroupr|rmfill ExceptionrrprqrvVIDEO_GEN_TIMEOUTrur{subr~)r% image_path video_promptrrr file_input post_token start_wait current_urlmatch prompt_inputevideo_options_btnresolution_btnrA video_urlr video_srcssrcrs r&generate_videoz%GrokPlaywrightProvider.generate_videos&""$$$zz*%  $008K;'    =  "#&7 |$DE E /Aii9+X.i>>>&&'9:::mmA  N\\"67== ((Z-?-?-A)BCCC JK  O YY[ iikJ&+--" " "((KII;[IE"[[^  =zJiikJ&+FtxxjQ  mmA  O  KK1<3D E #||B %  #((***mmC((("'' 555mmC((( ?@  O  BC G $ -Q R X X #))%)8 8 8--" " " KK3 4 KK6 7 G!%.I!J!P!P$**4*888mmC((( 67  O  56 G--!  $ KK5 6mmA  O  NPacmnpopcqr  iikE!$55--" " "$))+-.G $}}. J$%C!S($'  $FG ! *1 INN#RS 7)CR.Qw|r2G6G,G",L(M#MNH%|q  =wGYiikE!$55\34E3FG)l-zC   OFF?B : !!)[999 %{3   a % ?: D # +(5( @!D--,,\:::   " 9 "9( GBAFF G G NNBA F F G & G NNBA F F G  #l :sJ^7ZB1^7Z ^7!Z"^7>Z?A^7ZA+^7?Z!A*^7+2^7Z$)^7.Z36Z'7Z3Z*Z3,Z--Z3 Z0 Z3#^70]3\4]\.]0\!0\1\! \\!'^7]>];]>5^7 ^+A9^7^.8^7^1^7B>^7A^7+^4,'^7^7^7^7^7!^7$^7'Z3*Z3-Z30Z33 \<\ [81[42[87\ 8 \\ \\ ^7 \^7]]\!\!! ] *]]]  ] ]8]3-^73]88^7;]>> ^(^#^7#^((^7.^71^74^7cFK|j||||d{S7w)uAAlias for generate_video — matches GrokVideoProvider interface.N)r)r%rrrrs r&rWzGrokPlaywrightProvider.generates$((\;P[\\\\s !!cKdjd|jd|jd|jg}|tddd}dd l}dd l}|j5}d d d d ddd|ddtdddd|j|g}|j|dd} | jdk7rn| jr| jjnd} | jr| jjnd} td| jd| d| |j} | s td|j!| d d d y #1swYy xYww)z3Download file from URL using authenticated session.z; zsso=zsso-rw=z x-userid=r9r.)CookieRefererz User-AgentrNcurlz-sz-Lz-xrz-HzCookie: z Referer: zqUser-Agent: Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/144.0.0.0 Safari/537.36z-oTr)capture_outputrzNo error messagez No stdoutzCurl download failed (code z): z | stdout: zDownloaded file is empty)joinrrrrg subprocesstempfileNamedTemporaryFiler:run returncodestderrdecodestdoutrread write_bytes) r%rr cookie_headerheadersrrtmp_filecurl_cmdresult stderr_msg stdout_msg tmp_contents r&r~z%GrokPlaywrightProvider._download_files 4>>" #d''( ) ~ &#  $#AB   ( ( * 1hd`0 )A.JhmmH ^^HT3^OF  A%7=}}V]]113J\ 7=}}V]]113+ "=f>O>O=PPST^S__jkujv wxx#--/K :;;  # #K 0+ 1 1 1sA!E#C%E EEEcK|jr)|jjd{d|_|jr)|jjd{d|_|jr)|jjd{d|_|jr)|jj d{d|_|j jy777_7,w)zClose browser and cleanup.N)rcloserrrstopr"r%s r&rzGrokPlaywrightProvider.close s ::**""$ $ $DJ ==--%%' ' ' DM ==--%%' ' ' DM   ""'') ) )#D   % ( ( *sE*C:C24C:!C4"4C:C64C: C8 'C:4C:6C:8C:cK|SwNrs r& __aenter__z!GrokPlaywrightProvider.__aenter__s  sc@K|jd{y7wrrr%r+s r& __aexit__z GrokPlaywrightProvider.__aexit__ sjjls )r r r Frrrrrrrbool)N) rrr str | PathrrurOptional[Callable[[int], None]]return List[Path])r NN) rrrrrrrrrr)rrrr) __name__ __module__ __qualname____doc__r'rIrrrWr~rrrrr(r&r r #sVd0UMv7; ccc c 5 c  cP"&7; ccc c 5 c  cJ](1T r(r cVeZdZdZd d dZdZd dZ d ddZdZdZ d Z y)GrokPlaywrightProviderSyncz/Synchronous wrapper for GrokPlaywrightProvider.c:t|||||_d|_y)N)rrrr)r _provider_loopr$s r&r'z#GrokPlaywrightProviderSync.__init__(s%/)R^ho{CD r(c|j|jjr8tj|_tj|j|jSr)r is_closedrinew_event_loopset_event_looprs r& _get_loopz$GrokPlaywrightProviderSync._get_loop,sG :: !5!5!7 //1DJ  " "4:: .zzr(c x|jj|jj||fi|Sr)rrun_until_completerr)r%rrkwargss r&rz)GrokPlaywrightProviderSync.generate_image2s7~~22 )DNN ) )&* G G  r(Nc z|jj|jj|||fi|Sr)rrrr)r%rrrrs r&rz)GrokPlaywrightProviderSync.generate_video7s<~~22 )DNN ) )*lK ZSY Z  r(c|jri|jjsN|jj|jj |jj yyyr)rrrrrrs r&rz GrokPlaywrightProviderSync.closeBsN ::djj224 JJ ) )$..*>*>*@ A JJ   5:r(c|Srrrs r& __enter__z$GrokPlaywrightProviderSync.__enter__Gs r(c$|jyrrrs r&__exit__z#GrokPlaywrightProviderSync.__exit__Js  r()r r r Tr)rrrrrr)r N)rrrrrrrr) rrrrr'rrrrrrrr(r&rr%sV9  "&           r(r__main__z)%(asctime)s - %(levelname)s - %(message)s)levelformatc0Ktdtdtdtd4d{}td |jdddd d{}td t|d |D]}td | dddd{tdtdtdy77f#t$r,}td|ddl}|j Yd}~fd}~wwxYw7b#1d{7swYrxYww)Nz<============================================================z(Grok Playwright Provider Test (Headless)T)rz 1. Testing image generation...z9a majestic dragon flying over a medieval castle at sunsetz ./test_outputr]c td|dS)Nz Progress: %)print)ps r&ztest.._s%-s!0D*Er()rrrru ✓ Generated z images:z - u ✗ Error: rz= ============================================================z Test complete)rr rrwr traceback print_exc)providerimagesimgrrs r&testrRs h 89 h)48 & &H 4 5 &'66V. E 7  )#f+h?@!+CGC5/*+ & &$ o o h) & &qc*+ ##%% & & & & &s2DCD DCC1CD DC?'DC C<"C72D7C<<D?DDD DD)r __future__rriloggingrrrspathlibrtypingrrrr getLoggerrrrgrrtr rsys basicConfigINFOrrrr(r&rs # ++   8 $  ~~D&&R zGgll3^_6GKK?r(