$ iddlmZddlmZddlmZddlZddlZddlZddl Z ddl Z ddl m Z GddZ ded efd Zd ed ed e fd ZdS))genai)types)ImageN) load_dotenvcbeZdZdeedeededefdZdefdZdefdZ d Z d Z d S) BBoxResult bounding_boxcenter tokens_used costs_usdc>||_||_||_||_dSNr r r r )selfr r r r s ,/home/byschii/byschiidev/penelope/llm/llm.py__init__zBBoxResult.__init__s$( &"returncNt|jSr)sumr valuesrs r token_totalzBBoxResult.token_totals4#**,,---rcNt|jSr)rr rrs r cost_totalzBBoxResult.cost_totals4>((**+++rc8|j|j|j|jdS)Nrrrs r__dict__zBBoxResult.__dict__s' -k+    rc Hd|jd|jd|jd|jd S)NzBBoxResult(bounding_box=z , center=z, tokens_used=z , costs_usd=)rrs r__repr__zBBoxResult.__repr__"saQ$*;QQdkQQaeaqQQ@D@NQQQ QrN) __name__ __module__ __qualname__listintdictrrfloatrrr rrrr s#T#Y#S #PT#ae#### .S....,E,,,,   QQQQQrr thresholdrctj}||krtjddStjddS)z Returns GEMINI_API_KEY_PAGA if a random float is under the threshold, otherwise returns GEMINI_API_KEY_GRATIS. The bigger the threshold, the more likely to get the GRATIS key. GEMINI_API_KEY_PAGAGEMINI_API_KEY_GRATIS)randomosgetenv)r)rs r&get_gemini_api_key_by_random_thresholdr2&s>  A9}}y.333 9,b 1 11robject_description image_base64c4ttjtd}t j|}t jtj |}d|d}tj dtj dd }|j d ||g| }|jj}|jj} |jj} d } d } |j\} }t)d| d|dt)d|jt-j|j}t1|dddz | zt1|dddz |zt1|dddz | zt1|dddz |zg}t)d||d|dzdz|d|dzdzd}t3|||| | d|| zdz | z| dz | zd}|S)a Calls Gemini to detect bounding boxes in an image. Args: prompt: The prompt to send to Gemini image_base64: Base64-encoded image string Returns: List of bounding boxes in format [x1, y1, x2, y2] with absolute pixel coordinates g?)api_keyzU Detect the one object in the image that matches the following description: z Return slight larger bounding box of the object in the following JSON format: {"box_2d": [ymin, xmin, ymax, xmax]} The box_2d should be [ymin, xmin, ymax, xmax] normalized to 0-1000. Only one object should be detected. zapplication/jsoni)thinking_budgetMEDIA_RESOLUTION_MEDIUM)response_mime_typethinking_configmedia_resolutionzmodels/gemini-flash-latest)modelcontentsconfigg333333?g@u🖼️ Image size: xz pixelsu💬 Gemini response: box_2diru📦 Converted bounding box: )r?y)promptthoughtansweri@B) input_cost output_costr)rrClientr2base64 b64decoderopenioBytesIOrGenerateContentConfigThinkingConfigmodelsgenerate_contentusage_metadataprompt_token_countthoughts_token_countcandidates_token_countsizeprinttextjsonloadsr%r)r3r4client image_dataimagerEr>response prompt_len thought_len answer_leninput_cost_per_1M_tokensoutput_cost_per_1M_tokenswidthheightr converted_bounding_boxesr results rdetect_bounding_boxesrj2sMMM \6s;;F !,//J Jrz*-- . .EF (-.   3 F}--*.H (;J)>K(?J" #JME6 8 8 8 8 8 8999 28= 2 2333:hm,,L L "1 % ,u 455 L "1 % ,v 566 L "1 % ,u 455 L "1 % ,v 566    D*B D DEEE'q),DQ,GGA M&q),DQ,GGA MF - "   & 3y@C[[% 14MM     F Mr)googler google.genairPILrr[r/r.rKrNdotenvrrr'strr2rjr(rrrps QQQQQQQQ2 2e 2 2 2 2 2VcVVVVVVVVr