5iv. dZddlmZddlZddlZddlZddlZddlZddlZddl Z ddl m Z m Z ddl m Z ddlmZddlZddlmZej&eZdZd Zd d d d Zdddddddddd ZgdZdZddddZGddeZy)u Google ImageFX / Flow Image Provider – generates images via the aisandbox-pa.googleapis.com API using manually extracted browser cookies. Auth flow (reverse-engineered): 1. Exchange browser cookies for a Bearer access_token via https://labs.google/fx/api/auth/session 2. Call https://aisandbox-pa.googleapis.com/v1:runImageFx with Bearer auth HOW TO GET YOUR COOKIES: 1. Open https://labs.google/fx/tools/image-fx in Chrome 2. Open DevTools (F12) → Application → Cookies → https://labs.google 3. Copy `__Secure-next-auth.session-token` and `__Host-next-auth.csrf-token` 4. Paste them into your .env as GOOGLE_FLOW_SESSION_TOKEN and GOOGLE_FLOW_CSRF_TOKEN respectively. ) annotationsN)datetimetimezone)Path)Any) ImageProviderz'https://labs.google/fx/api/auth/sessionz1https://aisandbox-pa.googleapis.com/v1:runImageFxIMAGE_ASPECT_RATIO_PORTRAITIMAGE_ASPECT_RATIO_LANDSCAPEIMAGE_ASPECT_RATIO_SQUARE)portrait landscapesquare IMAGEN_3_5IMAGEN_3_5_FASTIMAGEN_3 IMAGEN_3_FAST GEM_PIX_2IMAGEN_4 NANO_BANANANANO_BANANA_PRO) nano_banana_pro imagen_3_5imagen_3_5_fastimagen_3 imagen_3_fast gem_pix_2imagen_4 nano_banananano_banana_pro_flow))rzGemini Pix 2 (newest))rzImagen 4)rzNano Banana Pro (Flow))rzImagen 3.5 (Nano Banana Pro))rzImagen 3.5 Fast)rzImagen 3zhttps://labs.googlez%https://labs.google/fx/tools/image-fxzapplication/json)OriginRefererz Content-TypeceZdZdZdZdZddZddZddZddZ ddZ d dd Z d dd Z dd Z d dd ZdZdZdZy )GoogleFlowImageProviderzKGenerates images via Google ImageFX (labs.google/fx) using browser cookies. google_flowc4tjdd|_tjdd|_tj tjddj d|_tjddj }tj ||j|_ |js td |js td d|_ d|_ tjd d |_y)NGOOGLE_FLOW_SESSION_TOKENGOOGLE_FLOW_CSRF_TOKENGOOGLE_FLOW_ASPECT_RATIOr r GOOGLE_FLOW_MODELruGOOGLE_FLOW_SESSION_TOKEN is required. Extract it from Chrome DevTools → Application → Cookies → https://labs.google → __Secure-next-auth.session-tokenuGOOGLE_FLOW_CSRF_TOKEN is required. Extract it from Chrome DevTools → Application → Cookies → https://labs.google → __Host-next-auth.csrf-token<T)timeoutfollow_redirects)osgetenv session_token csrf_token ASPECT_RATIOSgetlower aspect_ratio IMAGE_MODELSuppermodelEnvironmentError _access_token _token_expiryhttpxClientclient)self model_keys G/root/.openclaw/workspace/visionaryfx/providers/google_flow_provider.py__init__z GoogleFlowImageProvider.__init__YsYY'BBG))$))+2299FB  M(('(:G+w6**, !/D'CD!((+EeW)MN!((. / ##DIIi$84$?'#R ;<7 M NNH! L L MsB C)) D2D  Dc|jrH|jrr3 SESSION_URLDEFAULT_HEADERSrGrTraise_for_statusjson RuntimeErrorlistkeys fromisoformatreplace ValueError isoformat)r?respdata expires_strs rA_refresh_access_tokenz-GoogleFlowImageProvider._refresh_access_tokens   "" X\\*T-?-??%% % EF{{ !$--/  ##D) yy{!XXn5P-9Phhy) !!=d499;>O=PQEE   *%-%;%;''X6&"  0.2.@.@D   ( ( *i !!! *%)" *s*F55GGc^|j}it|jd|dS)z7Return headers with Bearer auth for the generation API.zBearer )rl Authorization)rrurG)r?rSs rA _auth_headersz%GoogleFlowImageProvider._auth_headerss=**,  ))+&ug.  rCNc|tjdd}||g|ddttjdzddd|ji|j d S) z&Build the JSON body for v1:runImageFx.ri?B)candidatesCountpromptsseedrJiIMAGE_FX) sessionIdtool modelNameType) userInput clientContext modelInput aspectRatio)randomrandintinttimer8r5)r?promptr num_imagess rA_build_generate_bodyz,GoogleFlowImageProvider._build_generate_bodysu <>>!V,D$."8 !TYY[4%7!8 9:"   ,,  rCcft|}|jjdd|j|||}tj d|j |j||jjt||j}|j||j|j}tjdt!|j#|j%dxsg}|s$t'dtj(|d d |d j%d xsg} | s(t'd t!|d j#| d } | j%d} | j%d} | rt+j,| } n%t'dt!| j#t/|d5}|j1| d d d tj d|t3| | |_|S#1swY2xYw)zqGenerate an image from *prompt* and save to *output_path*. Returns the Path of the saved image. T)parentsexist_okrru?Generating image via Google ImageFX model=%s aspect=%s → %s)rwrKzGenerate response keys: %s imagePanelsz+No imagePanels in response. Full response: NirgeneratedImagesz/No generatedImages in first panel. Panel keys: encodedImagemediaGenerationIdz*No encodedImage in generated image. Keys: wbzImage saved: %s (%d bytes))rrZmkdirrrNrbr8r5r>post GENERATE_URLrrTrvrwrOryrzr3rxdumpsbase64 b64decodeopenwritelen_last_media_id)r?r output_pathrrbodyrresultpanelsimages first_imageencodedmedia_idimg_datafs rA_generate_syncz&GoogleFlowImageProvider._generate_syncs ;'    =((dz(R M JJ      {{ &&(   ##D)  14 3FGM*0b=djj>PQURU>V=WX 017RA$vay~~GWBXAYZ Qi //.1??#67 ''0H    s &/-/cH|j||||}||jfS)zGenerate an image and return both the path and the mediaGenerationId. The mediaGenerationId can be used as startImage.mediaId for I2V video generation. r)rr)r?rrrrpaths rAgenerate_with_media_idz.GoogleFlowImageProvider.generate_with_media_idCs5"" Kdz# T((((rCc8|jjy)zClose the HTTP client.N)r>closerFs rArzGoogleFlowImageProvider.closeSs rCc|SNrFs rA __enter__z!GoogleFlowImageProvider.__enter__Ws rCc$|jyr)r)r?argss rA__exit__z GoogleFlowImageProvider.__exit__Zs  rC)returnstr)rQzhttpx.ResponserNone)rSrrr)rzdict[str, str])NrI)rrr int | Nonerrrdict) rrr str | Pathrrrrrr)rrrrrrrr) rrrrrrrrrztuple[Path, str])__name__ __module__ __qualname____doc__namerBrGrTrPrrrrrrrrrrrCrAr#r#TsU DF@  M0*"X GH  !+ @C  6  BB B B  B  BH            )) ) )  )  ) rCr#)r __future__rrrrwloggingr.rrrrpathlibrtypingrr<providers.portsr getLoggerrrNrtrr2r6IMAGE_MODEL_OPTIONS DEFAULT_MODELrur#rrCrArs$#  ' )   8 $ 8 B ./) $($ -   $6&GmGrC