o rr)r#totalrEr$r$r% get_statss  zImageCache.get_statscCs|jd|_d|_dS)Nr)rclearr r!r"r$r$r%rHs  zImageCache.clearN)__name__ __module__ __qualname____doc__r&strintr2rbytesr<r@rr rGrHr$r$r$r%rWs" rc@seZdZdZddZdededefddZdefd d Z d e d ed ede e fddZ dededededede e f ddZ dededededede e f ddZdededededede e f ddZdededededede e f ddZdededededede e f ddZdededededede e f dd Zdededededed!edeefd"d#Zdeeeffd$d%Zd&d'Zdeeeeffd(d)Zdeeeffd*d+Zd,S)-MultiProviderImageGeneratorz)Multi-provider with working free servicesc Cst|_d|_dd|jdddd|jdddd |jddd d |jddd d |jddg|_t dt dt dd|jDdS)NrzPollinations-Mainr4T)nameprioritymethodrCzPollinations-Directr zPollinations-APIzPollinations-Alt1zPollinations-Alt2u+✅ MultiProviderImageGenerator initializedz Active Providers: cSg|]}|dr|qSrCr$.0pr$r$r% z8MultiProviderImageGenerator.__init__..) rrgeneration_count_pollinations_main_pollinations_direct_pollinations_api_pollinations_alt1_pollinations_alt2 providersr9r>r"r$r$r%r&s<!"z$MultiProviderImageGenerator.__init__r'r(r+cCs@|}t|td}|d|dd}|d}||fS)zEnhance promptrz, rz, masterpiece, best qualityr)lowerstripCATEGORY_ENHANCEMENTSr<)r#r'r( enhancementenhancedrr$r$r%enhance_prompts z*MultiProviderImageGenerator.enhance_promptcCs tddS)zGenerate random seedii?B)randomrandintr"r$r$r% generate_seeds z)MultiProviderImageGenerator.generate_seed image_data target_width target_heightc Csz7tt|}|j||fkr|||ftjj}|jdkr#|d}t}|j |dt d| d| WSt yQ}ztd|WYd}~dSd}~ww)zProcess image dataRGBPNG)formatqualityrz Image processing error: N)ropenrrDresize ResamplingLANCZOSmodeconvertsave IMAGE_QUALITYseekgetvalue Exceptionr9)r#rnrorpimgoutputer$r$r%_process_images    z*MultiProviderImageGenerator._process_imagerr)r*seedc s>zdtj|}|||dddd}t4IdH[}|j||tjtdd4IdH3} | jdkrV| IdH} | | ||WdIdHWdIdHWSWdIdHn 1IdHsfwYWdIdHWdS1IdHs}wYWdSt y} zt d | WYd} ~ dSd} ~ ww) zMain Pollinations endpoint%https://image.pollinations.ai/prompt/trueflux)r)r*rnologoenhancemodelNrFparamstimeoutz Pollinations-Main: urllibparsequoteaiohttp ClientSessionr< ClientTimeoutREQUEST_TIMEOUTstatusreadrrr9 r#r'rr)r*rurlrsessionresponser5rr$r$r%r_s8 "  *z.MultiProviderImageGenerator._pollinations_mainc s:zdtj|}|||dd}t4IdH[}|j||tjtdd4IdH3} | jdkrT| IdH} | | ||WdIdHWdIdHWSWdIdHn 1IdHsdwYWdIdHWdS1IdHs{wYWdSt y} zt d| WYd} ~ dSd} ~ ww) z"Direct Pollinations image endpointzhttps://pollinations.ai/p/r)r)r*rrNrrrz Pollinations-Direct: rrr$r$r%r`s4"  *z0MultiProviderImageGenerator._pollinations_directc sRzd}|dtj|}|||dddd}|r||d<t4IdH[} | j||tjtd d 4IdH3} | jd kr`| IdH} | | ||WdIdHWdIdHWSWdIdHn 1IdHspwYWdIdHWdS1IdHswYWdSt y} zt d | WYd} ~ dSd} ~ ww) zPollinations API endpointz$https://image.pollinations.ai/prompt/rrfalse)r)r*rrrrrNrrrz Pollinations-API: r) r#r'rr)r*rrfull_urlrrrr5rr$r$r%ras> "  *z-MultiProviderImageGenerator._pollinations_apic sFzd}|dtj|d|d|d|d }t4IdHZ}|j|tjtdd 4IdH3} | jd krZ| IdH} | | ||WdIdHWdIdHWSWdIdHn 1IdHsjwYWdIdHWdS1IdHswYWdSt y} zt d | WYd} ~ dSd} ~ ww) z!Alternative Pollinations format 1zhttps://image.pollinations.aiz/prompt/z?width=z&height=z&seed=z &nologo=trueNr)rrz Pollinations-Alt1: r) r#r'rr)r*rbase_urlrrrr5rr$r$r%rb9s,*   *z.MultiProviderImageGenerator._pollinations_alt1c s<zdtj|}|||ddd}t4IdH[}|j||tjtdd4IdH3} | jdkrU| IdH} | | ||WdIdHWdIdHWSWdIdHn 1IdHsewYWdIdHWdS1IdHs|wYWdSt y} zt d | WYd} ~ dSd} ~ ww) z!Alternative Pollinations format 2rturbor)r)r*rrrNrrrz Pollinations-Alt2: rrr$r$r%rcJs6"  *z.MultiProviderImageGenerator._pollinations_alt2 model_namec s|jd7_|j||||}|r|S|||\}}|} td|jtd|dddtd|d|d | d d |jD} tt| d d ddD]r\} } | d} td| dt | d| z3| d||||| IdH}|rt |d}td|dd|j ||||||WStdWnt y}z td|WYd}~nd}~ww| t | krt tIdHqUtddS)zGenerate with fallbackr4u 🎨 Generation #z Prompt: Nrz...z Size: xz | Seed: cSrWrXr$rYr$r$r%r\yr]zFMultiProviderImageGenerator.generate_with_fallback..cS|dSNrRr$rr$r$r%{zDMultiProviderImageGenerator.generate_with_fallback..r:rQz [rz] rSu ✅ SUCCESS - rAz MBu ⚠️ No data returnedu ❌ Error: u ❌ All providers failed)r^rr<rjrmr9rd enumeratesortedr>r@rasynciosleep RETRY_DELAY)r#r'r(rr)r*cachedrirractiveidxproviderrQresultsize_mbrr$r$r%generate_with_fallbackfs>    z2MultiProviderImageGenerator.generate_with_fallback image_countc sg}tddtd|dtdt|D]}td|dd|||||||IdH} | rtd dd } t|d | dd } d | dd  dddd} |d | d | d |dd | } t| d}|| |dt| dddtd| n|d|dddddtd||dkrtdIdHqtdd|D}tddtd|d|tdd|S)zGenerate multiple images zP================================================================================z GENERATING z IMAGE(S)u 📸 Image r4rNz%Y%m%d_%H%M%S_%fr,rT,zutf-8TrrA)rQbase64successru ✅ failed_FzGeneration failed)rQrrerroru ❌ Failedcss|] }|drdVqdS)rr4Nr$)rZrr$r$r% szGMultiProviderImageGenerator.generate_multiple_images..u✅ Complete: )r9rangerrr7strftimer-md5r/r0joinsplitrereplacer b64encodedecodeappendr>rrsum)r#r'r(rr)r*rimagesir timestamphash_valbriefrQb64 successfulr$r$r%generate_multiple_imagessH  $*$  z4MultiProviderImageGenerator.generate_multiple_imagescCs |jSN)rrGr"r$r$r%get_cache_statss z+MultiProviderImageGenerator.get_cache_statscCs|jdSr)rrHr"r$r$r% clear_cachesz'MultiProviderImageGenerator.clear_cachecCsddt|jdddDS)NcSs$g|]}|d|d|ddqS)rQrRrC)rQrRrCr$rYr$r$r%r\szAMultiProviderImageGenerator.get_provider_info..cSrrr$rr$r$r%rrz?MultiProviderImageGenerator.get_provider_info..r)rrdr"r$r$r%get_provider_infosz-MultiProviderImageGenerator.get_provider_infocCs(|j|tdd|jD|dS)NcSrWrXr$rYr$r$r%r\r]z9MultiProviderImageGenerator.get_stats..)total_generations cache_statsactive_providersrd)r^rr>rdrr"r$r$r%rGs z%MultiProviderImageGenerator.get_statsN)rIrJrKrLr&rMtuplerjrNrmrOrrr_r`rarbrcrrdictrrr rrrrGr$r$r$r%rPs>) &&&&& / 5 rP)"rLrrrequestsrr-rkr3iorPILrrrtypingrrrr collectionsr urllib.parser DEFAULT_WIDTHDEFAULT_HEIGHTr|rrr6rrrgrrPr$r$r$r%sb     (A