љi!SSKrSSKJr SSKJrJrJr SSKJr SSK J r SSK r SSK J r SSKJr SSKJr \"\5r"S S 5r"S S \5r"S S\5rg)N) getLogger)TupleDictUnion)Path)uuid4)genai)types)Toolc\rSrSrSS\S\S\S\S\S\S \S \S \4S jjrSSSS .S\S\S\ \ \4S\ \\ 44Sjjr S\S\ 4Sjr S\S\ 4SjrSrg) NanoBananaNgemini_api_keymodelsystem_instruction id_prefixtimeout get_image save_imagesave_dirdebugc X lX0l[R"U[R "US9S9UlU=(d SUlX`lXpl Xl U n g)N)r)api_key http_options nanobanana) rrr Clientr HttpOptionsclientr _get_image _save_imager) selfrrrrrrrrrs [/home/james-whalen/.local/lib/python3.13/site-packages/aiavatar/sts/llm/tools/nanobanana.py__init__NanoBanana.__init__sS "4ll"**7;  "1\#% )image_idpromptreference_imager&r'r(returnc# U(dURS[53n/nU(a,UR[RR US95 U(a[ U[5(a.UR[RRUSS95 OB[ U[5(a-UR[RRUSS95 [R"SUS9/n[R"S/S 9nUR(aURUl URRR R#UR$UUS 9IShvN nUR&(aUR&S R((aUR&S R(R*HnUR,(dMUR,R.(dM3UR,R.R1S 5(aXR,R.4s $U[2R4"UR,R.54s $ [7S 5eGN7f)N_)textz image/png)data mime_type)file_urir.user)rolepartsIMAGE)response_modalities)rcontentsconfigrsPNGzImage is not generated.)rrappendr Part from_text isinstancebytes from_bytesstrfrom_uriContentGenerateContentConfigrraiomodelsgenerate_contentr candidatescontentr2 inline_datar- startswithbase64 b64decode Exception) r!r&r'r(r2r5generate_content_configrespps r"generate_imageNanoBanana.generate_image's..)57)4H  LL--6-: ; /511 UZZ22S^2_`OS11 UZZ00/U`0abMMvU;<"'"="=SZR["\  " "9=9P9P # 6[[__++<<***=   ??tq199__Q'//55===Q]]%7%7%7}}))44Z@@'););;;')9)9!--:L:L)MMM 6122 s E+I8-I5.A(I8I87A?I8id image_bytescj# UR(aURX5IShvN g[R"[UR5US3- S5IShvN nUR U5IShvN SSS5IShvN gNhN0NN !,IShvN (df  g=f7f)N.pngwb)r aiofilesopenrrwrite)r!rPrQfs r"rNanoBanana.save_imageJs   ""23 3 3}}T$--%8bT;%FMMQRggk***NMM 4M*NMMMsi&B3B9B3"B#B3&B;B<B B3 B B3B3BB3B0B" B0,B3cf# UR(aURU5IShvN $[R"[UR5US3- S5IShvN nUR 5IShvN sSSS5IShvN $NfN/NN !,IShvN (df  g=f7f)NrSrb)rrUrVrrread)r!rPrXs r" read_imageNanoBanana.read_imageQsw ??,, ,}}T$--%8bT;%FMMQRVVX~NMM-M%NMMMsi&B1B8B1!B"B1%B9B:B= B1 B B1B1BB1B.B  B.*B1)rr rrrrr)gemini-2.5-flash-imageNN`NNnanobanana_imagesF)__name__ __module__ __qualname____firstlineno__r=intcallableboolr#rr;rrNrr]__static_attributes__r%r"r r s."&"#+     07;$uy!3!3C!3afglnqgqar!3DEHJOEOP!3F+3+U+&3&5&r%r c^\rSrSrSSSSSSSSSSSSSS. S \S \S \S \S \S\\\4S\S\S\S\S\S\S\ S\ 4U4Sjjjr SS\S\S\ \\44Sjjr S\S\4Sjr SrU=r$) NanoBananaToolYr_Nimager`raF rrrrr(rrrnamespec instruction is_dynamicrrrrrrr(rrrrprqrrrsrc >[UUUUUUUU US9 UlX`lXl[TU]U =(d SU =(d SU =(d SSSSSS.SS S.S .S /S .S .S.UR U U S9 g)N) rrrrrrrrrrNfunctionzGenerate image by given prompt.objectstringzPrompt to generate image.)type descriptionzURI for reference image.)r'reference_image_urir'rx propertiesrequiredrpry parametersrxru)rprqfuncrrrs)r rr(rsuperr#rNr!rrrrrr(rrrrprqrrrsr __class__s r"r#NanoBananaTool.__init__Zs$%)1!   /  ))" 4$4#D (/7Hc&d 9 LL4RD9 :78 8 9sVB1A'A#$A'A%A'"B#A'%A'' B1B B BBBrPcT# URRU5IShvN $N7fN)rr])r!rPs r"rNanoBananaTool.get_images __//3333s (&()rrr(r)rbrcrdrer=rfrr;rgrhr#rrNrri __classcell__rs@r"rlrlYs  ."& -1"#+ !4 4  4  4  4 4 ucz*4 4 4 4 4 4 4 4 !4 4 l 93 9S 9TXY\^aYaTb 94#4%44r%rlc^\rSrSrSSSSSSSSSSSSSS. S \S \S \S \S \S\\\4S\S\S\S\S\S\S\ S\ 4U4Sjjjr S\S\S\S\ \\44Sjr Sr U=r$)NanoBananaSelfieToolr_Nselfier`raFrorrrrrr(rrrrprqrrrsrc>[TU]UUU=(d SUUUUUU U =(d SU =(d" SU =(d SSSSS0SS0SS0S./SQS .S .S .U U US 9 URUlg) NaYou are the selfie generator. Please update the selfie image of the given character in the given situation according to the following rules. ## Rules - Keep the selfie composition. - Do not output smartphones or cameras. We want to output an image taken by the character themselves. generate_selfieruz*Generate selfie image by given situations.rvrxrw)face_expressionclothlocation_backgroundr{r~r)rrrrrr(rrrrprqrrrsr)rr#rrrs r"r#NanoBananaSelfieTool.__init__s$ )1 6 +!**" 5$5#O (06/A&,h%74:H3E' %X#  $!K & N(( r%rrrr)c # SUSUSUS3nUR(a[RSU35 URU5IShvN nSUS0$N ![a&n[R SU35 S S 0sSnA$SnAff=f7f) Nz:Modify this selfie image: - face expression: z - cloth: z$ - location background: z zPrompt for selfie: selfie_idr&zError at generate_selfie: rzError at generate_selfie.)rrinforNrJr)r!rrrr'resultrs r"r$NanoBananaSelfieTool.generate_selfies : /01W$$7#89 F zz 1&:;..v66F !34 47 : LL5bT: ;89 9 :sGBA A A ABA B %BB BB  B)r)rbrcrdrer=rfrr;rgrhr#rrrirrs@r"rrs ."&!-1"#+ !9)9) 9)  9)  9)9)ucz*9)9)9)9)9)9)9)9) !9)9)v:S::[^:cghkmphpcq::r%r)rHloggingrtypingrrrpathlibruuidrrUgoogler google.genair aiavatar.sts.llmr rbrr rlrrjr%r"rsW %%! 8 H&H&VD4TD4NJ:>J:r%