Xj ddlmZddlZddlmZddlZddlmZddlm Z m Z m Z ddl m Z mZmZddd d d d d dddddd ZdZGddZy)) annotationsN)Path)OpenAI)Image ImageFilter ImageStat)retrystop_after_attemptwait_exponentialzPfuturistic neural interfaces, holographic model cards, luminous compute clusterszKcinematic product render, exploded chip view, studio lighting, macro detailzWfunding chart holograph, modern founder office, growth dashboards, premium pitch energyzPdark SOC room, encrypted code rain, abstract lock geometry, red alert UI accentszKfuturistic lab, scientific diagrams as 3D objects, clean research aestheticzMrealistic IDE on ultrawide monitor, developer hands, syntax highlighting glowz>cinematic humanoid robot, industrial atelier, volumetric lightzBmedical imaging holograms, clean clinical tech, soft cyan lightingzIfintech dashboard glass panels, trading floor blur, premium navy and goldz:isometric cloud topology, glowing nodes, data center bokehz>collaborative coding space, terminal windows, community energyzGkeynote stage lighting, product silhouette, premium brand-safe abstract) aihardwarestartupcyberresearch programmingrobotics healthcarefinancecloud opensourcebigtech)dall-e-3z gpt-image-1ceZdZdZdddZdddddd dd Zddd Zeed e d d ddddZ ddZ ddZ y)ImageGeneratoruAMarketing-quality stills via OpenAI Images (DALL·E 3 preferred).cBt||_||_||_y)N)api_key)rclientpreferred_modelmodel)selfrrs ;/var/www/html/content_generation/src/generators/ai_image.py__init__zImageGenerator.__init__!sW- $ r editorial_3d dark_premiumN)categoryillustration_stylebackground_stylepalettehookc tj|td}|rdj|ddnd}d|d|d|d|d |xs|d dd S) Nr z, zpremium brand colorsuGGenerate HERO ARTWORK ONLY (not a final social post). Ultra-premium social media campaign visual for a top AI/tech media brand. Quality bar: Apple keynote / Stripe / Linear / Nvidia campaign — NOT Canva, NOT poster, NOT meme. NO readable text, NO logos of real companies, NO watermark, NO clipart icons floating in space. Scene must have strong focal subject + foreground + middle ground + background depth. Use volumetric lighting, atmospheric depth/fog, rim lighting, reflections, cinematic shadows, HDR feel. Rich environmental storytelling with realistic textures and layered objects. Reserve natural 20-30% clean composition space for future text overlay (not a giant empty gradient). The clean zone should be subtle and integrated into composition. If human faces appear, keep them realistic and non-distorted. Illustration style: z. Background: z . Palette: z. Visual language: z . Story: z. Include concrete subjects: devices, UI mockups, 3D objects, environments, data viz metaphors. Avoid flat framing; compose like a premium campaign hero shot.<)CATEGORY_VISUALSgetjoin) r base_promptr'r(r)r*r+catcolorss r! build_promptzImageGenerator.build_prompt&s""8-=d-CD+272A;'8N #$6"6nEUDVWx23%8!)T*+M M 4! r#c|jjdrdddj|dSdddj|dS)u#Map creative aspect → model size.dall-e 1024x1792 1792x1024)portrait landscape 1024x1024 1024x1536 1536x1024)r startswithr0)r aspects r!aspect_to_sizezImageGenerator.aspect_to_sizeDsE ::  * ++FJJ6S^_ _'kBFFv{[[r#)minmaxT)stopwaitreraisec|jjddg}|jgtD]}||vs|j |d}|D]} ||_|xs|j |} |jdrhd} nhd} | | vr| nd} ||dd| dd } |jdrd | d <|j| |jd }|j|r|cSt| } | d dzdzdd| d <|j| |}|j|r|cStdtd|d|#t$r } | }Yd} ~  d} ~ wwxYw)NT)parentsexist_okr7>r<r8r9>autor<r=r>r<r.)rpromptsizenhdqualityz.pngrNz@ Enforce premium cinematic composition, no flat poster framing, zRclear focal subject, layered depth, rich textures, and subtle negative space only.z)Generated image failed quality gate twicez#Image generation failed for models z: )parentmkdirrIMAGE_MODEL_FALLBACKSappendrrAr?_generate_once with_suffix_passes_quality_gatedict RuntimeError Exception)r rNout_pathrOr@modelsmlast_errruse_sizeallowedkwargsstricterexcs r!generatezImageGenerator.generateKsdT:&&?)>?A a @&*E# " >4#6#6v#>##H-EGMG'/7':8 #$Udm$   ##H-(,F9%..vx7K7KF7ST,,X6#O <8$XYjk4 &"  ..xB,,X6#O"#NOOCJ@8*UVV  s&BE +AE . E  E!EE!c|jjjdi|}|jd}t |dd}t |dd}|r&|j t j||S|rDtj|d}|j|j |j|Std)Nrb64_jsonurlZ)timeoutz+Image API returned neither url nor b64_json) rimagesrfdatagetattr write_bytesbase64 b64decoderequestsr0raise_for_statuscontentr[)r rcr]resultitemb64riresps r!rWzImageGenerator._generate_once{s,##,,6v6{{1~dJ-dE4(   !1!1#!6 7 <<R0D  ! ! #   .LM Mr#c tj|jd}|jd}t j |}|j r|j dnd}|jtj}t j |}|jr|jdnd}|j\} } |jdd| | dzf|jd| dz| d| zdzf|jdd| zdz| | fg} | D cgc]$} t j | j d&} } td| D}| d z| d zd| zd zd| zd zf\}}}}|j||||f}t j |j d}|d k\xr|d k\xr |d k\xr|d kS#t$rYywxYwcc} w)z Lightweight quality gate to reduce flat/poster-like generations. Reject if image is too low-detail, low-contrast, or overly uniform. RGBFLrgrBc3,K|] }|dks dyw)rMNrl).0ss r! z6ImageGenerator._passes_quality_gate..s?*QBq*s r-rM)ropenconvertr\rStatstddevfilterr FIND_EDGESmeanrOcropsum)r image_pathimggraystatstdedgesest edge_meanwhthirdst third_stdstoo_flat_regionscx0cy0cx1cy1center center_stds r!rYz#ImageGenerator._passes_quality_gates  **Z(007C{{3~~d# $ dkk!n K223nnU##&88CHHQK xx1 IIq!QQ' ( IIq!q&!a!e\2 3 IIq1q5Q,1- .  rsc" --@@ ]]h _]bPVZ IRX  4S S r#