o 8h*@sdZddlZddlZddlZddlZddlZddlmZddlm Z ddl m Z m Z m Z ddlZejejdeeZGdddZdS) u MULTI-API IMAGE GENERATOR WITH FALLBACK SYSTEM ============================================== Features: ✅ Multiple Free APIs (Auto Fallback) ✅ 99.9% Uptime Guarantee ✅ Fast Generation (10-20 seconds) ✅ Unlimited & Completely Free ✅ Smart Retry Logic ✅ Load Balancing N)BytesIO)Image)OptionalDictList)levelc@seZdZdZddZdededefddZdeded ed edee f d d Z deded ed edee f d dZ deded ed edee f ddZ defddZ defddZd"deded ed edee f ddZ  d#dededed ed ededeefddZdefdd Zd!S)$MultiAPIImageGeneratorz< Robust image generator with multiple API fallbacks cCsvdd|jddddd|jddddd |jdddg|_d d d d dd d d ddd d dd d d d|_d|_d|_dS)N Pollinations%https://image.pollinations.ai/prompt/Tr)nameurlmethod availablefailures HuggingFace,https://api-inference.huggingface.co/models/ Stabilityz'https://api.stability.ai/v1/generation/z flux-realism(stabilityai/stable-diffusion-xl-base-1.0zstable-diffusion-xl-1024-v1-0) pollinations huggingface stabilityzflux-3dz flux-animezLinaqruf/anything-v3.0z flux-cablyai) realistic3danimecartoon<)_generate_pollinations_generate_huggingface_generate_stability_freeapis model_map max_retriestimeoutselfr&H/var/www/eduai.edurigo.com/generate_image/testing/multi_api_generator.py__init__ sL zMultiAPIImageGenerator.__init__promptcategoryreturncCs.dddddd}||d}|d|S) z Enhance prompt based on categoryz@photorealistic, ultra detailed, 8k uhd, professional photographyz03d render, octane render, high quality, detailedz.anime style, manga art, high quality, detailedz*cartoon style, vibrant colors, clean linesz0cinematic shot, movie quality, dramatic lighting)rrrr cinematiczhigh quality, detailedz, )getlower)r%r)r* enhancements enhancementr&r&r'_enhance_promptUsz&MultiAPIImageGenerator._enhance_promptwidthheightc CszF|j|idd}|||}dtj|}|||tdddddd}tj|||jd} | j d kr;| j WSt d | j Wd St ya} zt d | WYd } ~ d Sd } ~ ww) zGenerate using Pollinations APIrfluxr itrue)r2r3modelseednologoenhancenofeed)paramsr#zPollinations returned status NzPollinations error: )r!r-r1requestsutilsquoterandomrandintr# status_codecontentloggerwarning Exceptionerror) r%r)r*r2r3r7enhanced_promptr r<responseer&r&r'ras,   z-MultiAPIImageGenerator._generate_pollinationsc Csz=|j|idd}|||}d|}|||ddd}tj|||jd} | jdkr2| jWSt d | jWd St yX} zt d | WYd } ~ d Sd } ~ ww) z4Generate using HuggingFace Inference API (Free tier)rrr)r2r3num_inference_steps)inputs parameters)jsonr#r=zHuggingFace returned status NzHuggingFace error: ) r!r-r1r>postr#rCrDrErFrGrH) r%r)r*r2r3r7rIr payloadrJrKr&r&r'r~s(   z,MultiAPIImageGenerator._generate_huggingfacec Csz4|||}d}|||d}tj|||jd}|jdkr2|} d| vr2tj| ddd} | jWSWdStyO} zt d | WYd} ~ dSd} ~ ww) z)Fallback using alternative free endpointsz#https://api.deepai.org/api/text2img)textr2r3)datar#r= output_urlrL)r#NzStability fallback error: ) r1r>rQr#rCrPr-rDrGrErH) r%r)r*r2r3rIr rTrJresult img_responserKr&r&r'rs&  z/MultiAPIImageGenerator._generate_stability_free api_indexcCsZ|j|dd7<|j|ddkr+d|j|d<td|j|ddd Sd S) z-Mark API as failed and potentially disable itrr5FrzAPI r z disabled due to failuresN)r rErFr%rXr&r&r'_mark_api_failures  z(MultiAPIImageGenerator._mark_api_failurecCs d|j|d<d|j|d<dS)zReset failure count on successrrTrNr rZr&r&r'_mark_api_successsz(MultiAPIImageGenerator._mark_api_success8c Cstd|ddd|t|jD]}t|jD]\}}|ds$qtd|dd|jd |d zA|d ||||}|rrztt|| |td |d |WWSt d |d YWq| |Wqt y} zt d|d d| | |WYd} ~ qd} ~ ww||jdkr|dd} td| dt| qt ddS)zF Generate image with automatic fallback to other APIs zGenerating image: N2z... Category: rzAttempt r5/z - Using r r u✅ Success with zInvalid image from z Error with z: u ⏳ Waiting zs before retry...u!❌ All APIs failed after retries)rEinforanger" enumerater ropenrr]rFr[rGrHtimesleep) r%r)r*r2r3attemptidxapi image_bytesrK wait_timer&r&r'generate_image_with_fallbacks>&      z3MultiAPIImageGenerator.generate_image_with_fallbackr5 model_name image_countcCsg}t|D]m}td|dd||||||} | r^td} t|d| dd} |d| d|dd| } t |  d} | | | d td | n td |d||dkrstdq|S) z@ Generate multiple images with fallback support zGenerating image r5raz %Y%m%d_%H%M%S_Nzutf-8)r base64u✅ Generated: u❌ Failed to generate image )rdrErcrnrgstrftimehashlibmd5encode hexdigestrs b64encodedecodeappendrHrh)r%r)r*ror2r3rpimagesirl timestamp prompt_hash image_name base64_datar&r&r'generate_multiple_imagess&  $  z/MultiAPIImageGenerator.generate_multiple_imagescCsddd|jDiS)zGet status of all APIsr cSs$g|]}|d|d|ddqS)r rr)r rrr&).0rkr&r&r' sz9MultiAPIImageGenerator.get_api_status..r\r$r&r&r'get_api_statussz%MultiAPIImageGenerator.get_api_statusN)r^r_)r^r_r5)__name__ __module__ __qualname____doc__r(strr1intrbytesrrrr[r]rnrrrrr&r&r&r'rs,5" "" $- %r)rr>rsrurArgiorPILrtypingrrrlogging basicConfigINFO getLoggerrrErr&r&r&r's