5iodZddlmZddlZddlZddlZddlmZddlm Z ddl m Z ddl Z ddl mZejeZdZgd ZGd d eZy) uo Pollinations Image Provider – generates images from text prompts via Pollinations.ai API. Supports multiple models: flux, zimage, gptimage, seedream, nanobanana, etc. Env vars: POLLINATIONS_API_KEY - API key from enter.pollinations.ai (optional for limited use) POLLINATIONS_IMAGE_MODEL - Model to use (default: flux) Docs: https://gen.pollinations.ai/docs ) annotationsN)Path)Any)quote) ImageProviderzhttps://gen.pollinations.ai) fluxzimagekontext nanobananaznanobanana-proseedreamz seedream-progptimagegptimage-largekleinz klein-largeceZdZdZdZdd dZ d d dZ d dZed dZ y)PollinationsImageProviderz3Wraps the Pollinations.ai API for image generation. pollinationsNc|xstjdd|_t|_tjdd|_t tjdd|_t tjdd|_|j tvr-tjd|j td|_y y ) zInitialize the Pollinations image provider. Args: api_key: Optional API key. If not provided, uses POLLINATIONS_API_KEY env var. Can work without key for limited usage. POLLINATIONS_API_KEYPOLLINATIONS_IMAGE_MODELrPOLLINATIONS_IMAGE_WIDTH1024POLLINATIONS_IMAGE_HEIGHTz5Unknown image model '%s', using 'flux'. Available: %sN) osgetenvapi_keyPOLLINATIONS_BASE_URLbase_urlmodelintwidthheight IMAGE_MODELSloggerwarning)selfrs N/root/.openclaw/workspace/visionaryfx/providers/pollinations_image_provider.py__init__z"PollinationsImageProvider.__init__3sG")),BB"G - YY96B #=vFG "))$?HI ::\ ) NNG    DJ *c t|}|jjdd|xs |j}|xs |j}|xs |j }t |d} |jd| } ||||t|j|t| jd}|dvr!| |d<t| j|d <|jr|j|d <d d i}|jrd |j|d<tjd||||tjd|ddtjd5}|j!| ||}|j#|j%|j&dddtjd|t)j&|S#1swY5xYw)a.Generate an image from prompt and save to output_path. Args: prompt: Text description of the image to generate output_path: Where to save the generated image width: Image width in pixels (default: 1024) height: Image height in pixels (default: 1024) model: AI model to use (default: from env or 'flux') seed: Random seed for reproducibility (-1 for random) enhance: Let AI improve the prompt negative_prompt: What to avoid in the image safe: Enable safety content filters quality: Image quality level (gptimage only): low, medium, high, hd transparent: Generate with transparent background (gptimage only) Returns: Path to the saved image file T)parentsexist_okr)safez/image/)rr!r"seedenhancenegative_promptr-)r rquality transparentkeyAcceptzimage/*zBearer Authorizationu=Generating image via Pollinations: model=%s size=%dx%d → %sz Prompt: %sNd)timeout)paramsheaderszImage saved: %s (%d bytes))rparentmkdirr!r"rrrstrlowerrr$infodebughttpxClientgetraise_for_status write_bytescontentlen)r&prompt output_pathr!r"rr.r/r0r-r1r2encoded_prompturlr9r:clientresponses r'_generate_syncz(PollinationsImageProvider._generate_syncHs@;'    =#&4;;#vB/w~&677|))+.IOO%  2 2 'F9 $' $4$:$:$    s &/-/c*tjS)z&Return list of available image models.)r#copyr)r' list_modelsz%PollinationsImageProvider.list_modelss  ""r))N)r str | None) NNNFzworst quality, blurryFmediumF)rHr=rIz str | Pathr! int | Noner"r[rrXr.r r/boolr0r=r-r\r1r=r2r\returnr)rHr=rIrrRrr]r)r]z list[str]) __name__ __module__ __qualname____doc__namer(rNrS staticmethodrWrVr)r'rr.s= D 2!! 6!XX X X  X  XXXXXXX Xt          ##r)r)ra __future__rrPloggingrpathlibrtypingr urllib.parserrAproviders.portsr getLoggerr^r$rr#rrVr)r'rksT #  )   8 $5  B# B#r)