Ŕi 6dZddlZddlmZd dedededefdZy) zO Image generation service using Replicate (black-forest-labs/flux-2-klein-4b). N)loggerprompt aspect_ratio resolutionreturnc K tjd|d|d|dtjd|gd|ddd d  d {}|r5t |d j }tjd||ddStj ddddS7U#t$r0}tjd|t |ddcYd }~Sd }~wwxYww)aPGenerate an image using Replicate's flux-2-klein-4b model. Args: prompt: Text description of the image to generate. aspect_ratio: Aspect ratio (e.g. "1:1", "16:9", "4:3"). resolution: Output resolution ("1K", "2K", "4K"). Returns: dict with "url" key on success, or "error" key on failure. z![ImageService] Generating image: z (z, )z!black-forest-labs/flux-2-klein-4bFjpg_1)rimagesgo_fastr output_formatoutput_qualityoutput_megapixels)inputNrz [ImageService] Image generated: T)urlsuccessz/[ImageService] No image returned from Replicatez)No image was generated. Please try again.)errorrz'[ImageService] Error generating image: ) rinfo replicate async_runstrrwarning Exceptionr)rrroutputres L/Users/thippareddysaicharanreddy/Desktop/agent_all/services/image_service.pygenerate_imager s&3 7zL>QST^S__`ab!** /  ,!&"$%(    fQimm$C KK:3%@ A40 0HIDQVWW'  * 3 >qcBCQE223sSCABB:BCBCB C&%C C CCC)z1:11K)__doc__rlogururrdictrrr&s41313C13S13\`13r%