h-?tdZddlZddlZddlZddlZddlmZddlmZGddZdZ e dk(re yy) z Generate professional stock images for CleanKitchens using DALL-E 3 Industrial, realistic style with caution tape, commercial kitchens, violations N)OpenAI)datetimec<eZdZdZdZdZdZdZdZdZ dZ y ) StockImageGeneratorcttjd|_d|_d|_d|_dddd|_d|_y) NOPENAI_API_KEY)api_keyzJ/var/www/twin-digital-media/public_html/_sites/cleankitchens/assets/images 1792x1024hdr) generatedfailed total_costzProfessional photography, industrial commercial kitchen setting, dramatic lighting, high contrast, cinematic quality, photorealistic, no people, no text, no logos) rosgetenvclientbase_dir image_sizequalitystats style_baseselfs h/var/www/twin-digital-media/public_html/_sites/cleankitchens/production/scripts/generate_stock_images.py__init__zStockImageGenerator.__init__sPRYY/?%@A e &   cgd}|D]Q}tjj|j|}tj|dt d|Sy)z%Create directory structure for images) violationschainscuisine summariesT)exist_okzCreated directory: N)rpathjoinrmakedirsprint)r categoriescategoryr"s rcreate_directoriesz&StockImageGenerator.create_directories$sLE " 0H77<< x8D KKt , 'v. / 0rc d|jd|jd|jd|jgd|jd|jd|jd|jgd |jd |jd |jd |jgd |jd|jd|jd|jgd|jd|jd|jd|jgd|jd|jd|jd|jgd|jd|jd|jd|jgd|jd|jd|jd |jgd!|jd"|jd#|jd$|jgd%|jd&|jd'|jd(|jd)|jgd*|jd+|jd,|jd-|jgd. S)/z Get prompts for violation imagesztDark industrial kitchen corner with visible mouse droppings near food prep area, yellow caution tape in foreground, z\Commercial kitchen floor with rat shadows and gnawed food packaging, warning signs visible, z]Stainless steel kitchen shelving with rodent evidence, pest control boxes, dramatic shadows, zeRestaurant storage area with compromised food containers showing rodent damage, industrial lighting, z\Commercial kitchen dark corner with cockroach silhouettes on stainless steel, caution tape, z[Industrial dishwashing area with pest evidence, dramatic overhead lighting, warning signs, zZRestaurant prep station with insect traps and warning indicators, high contrast lighting, zbDark commercial kitchen crevices with pest activity indicators, professional inspection lighting, zfCommercial refrigerator with digital thermometer showing danger zone temperature, red warning lights, zdIndustrial kitchen heat lamp station with temperature violation indicators, steam and condensation, zVWalk-in cooler with frost buildup and temperature gauge in red zone, caution signage, zVStainless steel food warming station with visible temperature problems, warning tape, zaGrimy commercial kitchen surfaces with visible buildup, harsh inspection lighting, caution tape, zSDirty industrial dishwashing area with standing water, overhead dramatic lighting, zNNeglected restaurant prep area with accumulated debris, warning signs posted, zPCommercial kitchen floor drains with visible contamination, industrial setting, zaDark commercial kitchen corner with visible mold growth on walls, moisture damage, warning tape, zSIndustrial refrigerator seals with black mold, close-up detail, dramatic lighting, zTRestaurant ceiling tiles with water damage and mold stains, looking up perspective, zQWalk-in cooler with mold on surfaces, condensation visible, industrial lighting, z[Commercial cutting boards with raw and cooked foods dangerously close, warning indicators, zXIndustrial kitchen prep area with improper food storage, color-coded violation markers, zTRestaurant work station showing contamination risks, caution tape sectioning areas, zWStainless steel surfaces with mixed food preparation hazards, overhead warning lights, zXBroken commercial sink in industrial kitchen, caution tape across it, dramatic shadows, zUEmpty soap dispensers and paper towel holders in restaurant washroom, warning signs, zSBlocked handwashing station with supplies stored in sink, violation notice posted, zTIndustrial kitchen with missing handwashing supplies, inspection checklist visible, zWDamaged commercial kitchen ceiling with exposed infrastructure, caution tape barriers, zPCracked restaurant floor tiles with safety hazards marked, industrial lighting, zGBroken industrial kitchen equipment with warning tags, dramatic angle, zVRestaurant wall damage with exposed areas, temporary barriers, construction lighting, zKCommercial floor drain backup in industrial kitchen, warning cones placed, zYRestaurant basement with water damage indicators, caution tape barriers, harsh lighting, zGIndustrial kitchen with plumbing issues visible, warning signs posted, zUCommercial dishwashing area with drainage problems, standing water, warning markers, zYRestaurant entrance with official red closure notice on door, dramatic evening lighting, zQCommercial kitchen with red tags on all equipment, caution tape across entrance, zPLocked restaurant doors with health department closure signs, reflective glass, zODark restaurant interior visible through window with closure notice prominent, zXYellow caution tape across restaurant entrance with official notices, dramatic shadows, zcWide angle industrial kitchen with multiple violation indicators, caution tape, dramatic lighting, zaCommercial restaurant kitchen under inspection, warning signs visible, professional photography, z]Stainless steel commercial kitchen with health code violation markers, industrial aesthetic, zPRestaurant prep area with inspection tags and warnings, high contrast lighting, rodentroach temperature cleanlinessmoldcross_contamination handwashing structuralsewageclosuregeneralrrs rget_violation_promptsz)StockImageGenerator.get_violation_prompts-s0GHLHWHWGXYnoso~o~nAoptpppABwx|yHyHxIJ ooso~o~nAmnrn}n}m~lmqm|m|l}~tuyvEvEuFG yy}zIzIyJKvw{xGxGwHIhimixixhyzhimixixhyz ttxuDuDtEFefjfufuevw`aeapap`qrbcgcrcrbst ttxuDuDtEFefjfufuevwfgkgvgvfwxcdhdsdsctu nnrn}n}m~jkokzkzj{|fgkgvgvfwxijnjyjyiz{ $kkokzkzj{|ghlhwhwgxyefjfufuevwfgkgvgvfwx jjnjyjyiz{bcgcrcrbstYZ^ZiZiYjkhimixixhyz ^^b^m^m]noklpl{l{k|}YZ^ZiZiYjkghlhwhwgxy llpl{l{k|}cdhdsdsctubcgcrcrbstabfbqbqarsjkokzkzj{| vvzwFwFvGHstxuDuDtEFoptpppABbcgcrcrbst }D D rcd|jd|jd|jd|jgd|jd|jd|jd|jgd |jd |jd |jd |jgd |jd|jd|jd|jgdS)z'Get prompts for chain restaurant imagesz^Generic fast food burger restaurant kitchen, industrial fryer area, stainless steel surfaces, zWCommercial burger grill station with ventilation hoods, professional kitchen lighting, zCFast food prep area with burger assembly line, industrial setting, zMChain restaurant freezer with burger supplies, commercial equipment visible, zNCommercial coffee shop equipment area, espresso machines, industrial counter, zQChain coffee store prep area with syrups and supplies, stainless steel surfaces, zOIndustrial coffee roasting equipment in commercial setting, dramatic lighting, zHCoffee chain storage area with beans and supplies, commercial shelving, zIIndustrial pizza oven in commercial kitchen, high temperature equipment, zOChain pizzeria prep station with dough and toppings, stainless steel surfaces, zFCommercial pizza kitchen with multiple ovens, industrial ventilation, zFPizza chain refrigeration units with ingredients, commercial storage, zGIndustrial deep fryer station for chicken, commercial kitchen setting, zGChain restaurant chicken prep area, breading station, stainless steel, zACommercial pressure fryer equipment, industrial chicken kitchen, z:Fast food chicken storage freezers, commercial equipment, ) burger_chain coffee_chain pizza_chain chicken_chainr6rs rget_chain_promptsz%StockImageGenerator.get_chain_promptsusXqqurArAqBCijnjyjyiz{UVZVeVeUfg_`d`o`o_pq aaeapap`qrcdhdsdsctuabfbqbqarsZ[_[j[jZkl \\`\k\k[lmabfbqbqarsXY]YhYhXijXY]YhYhXij ZZ^ZiZiYjkYZ^ZiZiYjkSTXTcTcSdeLT__L]^ '  rc$d|jd|jd|jd|jd|jgd|jd|jd|jd |jd |jgd S) zGet prompts for summary imagesz\Clipboard with inspection checklist on stainless steel surface, dramatic overhead lighting, zXHealth inspection grade cards arranged on industrial surface, professional photography, zFDigital tablet showing inspection data in commercial kitchen setting, zMStack of violation notices on industrial kitchen counter, dramatic lighting, zNInspector's toolkit and reports on stainless steel table, professional angle, zIMultiple inspection reports spread on industrial surface, overhead view, zOWeek calendar with violation markers in commercial setting, dramatic lighting, zPSeries of health grade cards showing weekly progression, industrial background, zNInspection clipboard with weekly summary charts, commercial kitchen backdrop, zIDigital display showing weekly violation statistics, industrial setting, ) daily_summaryweekly_summaryr6rs rget_summary_promptsz'StockImageGenerator.get_summary_promptssooso~o~nAjkokzkzj{|XY]YhYhXij_`d`o`o_pq`aeapap`qr \\`\k\k[lmabfbqbqarsbcgcrcrbst`aeapap`qr[\`\k\k[lm   rc b td|d|d|jjjd||j|j d}|j dj}tj|}|jdk(r|d vrd }nd |vrd }n d |vrd}nd}|d|d}tjj|j||} t| d5} | j!|j"dddtd| |j$dxxdz cc<|j$dxxdz cc<|||| |dS t)j*dy#1swYdxYw#t&$r5} td|d|d| |j$dxxdz cc<Yd} ~ yd} ~ wwxYw)z&Generate a single image using DALL-E 3z Generating _z...zdall-e-3)modelpromptsizernrr*r_chainr_summaryr r5z.jpgwbNu ✓ Saved: r rg{Gz?)r'filename subfolderfilepathrFu✗ Failed to generate z: r )r%rimagesgeneraterrdataurlrequestsget status_coderr"r#ropenwritecontentr Exceptiontimesleep) rrFr'indexresponse image_url img_responserNrMrOfes rgenerate_imagez"StockImageGenerator.generate_images3  Kz5'5 6{{))22 __  3H! a(,,I$<< 2L''3. ??!-I) (I8+ +I )I'Zqt477<< y(K(D)2QGGL0012 H:./ ;'1,' <(D0(!) (!* ($ 1/L 1 +22  +H:QugRsC D JJx A %  s1C#E0%E$A E0$E-)E00 F.9+F))F.cg}td|dt|dD],\}}|j|||}|s|j|.|S)z"Generate all images for a categoryz === Generating z images ===rD)r% enumeraterdappend)r category_namepromptsresultsirFresults rgenerate_categoryz%StockImageGenerator.generate_categorys] !- <="7A. 'IAv((BFv& ' rcbtdtd|jtd|j|jg}|j }|j D](\}}|j ||}|j|*|j}|j D](\}}|j ||}|j|*|j}|j D](\}}|j ||}|j|*tjj|j|d}tjj!|j"d} t%| d5} t'j(|| dd d d td td |jd d td|jdd td|jddtd| |S#1swYxxYw)zGenerate all stock imagesz0Starting CleanKitchens stock image generation...z Image size: z Quality: ) generated_atrrQzgeneration_metadata.jsonwrP)indentNz === Generation Complete ===z Generated: r z imageszFailed: r z Total cost: $rz.2fzMetadata saved to: )r%rrr(r7itemsrmextendr=rArnow isoformatrrr"r#rrXjsondump) r all_resultsviolation_promptsr'rirj chain_promptssummary_promptsmetadata metadata_pathrbs rgenerate_all_imagesz'StockImageGenerator.generate_all_imagess @A T__-./ $,,() ! !668!2!8!8!: ( Hg,,Xw?G   w ' ( ..0 !.!4!4!6 ( Hg,,Xw?G   w ' ( 224!0!6!6!8 ( Hg,,Xw?G   w ' ( %LLN446ZZ!   T]]4NO - % - IIh! , - -/ DJJ{34G<= H-.g67 djj6s;<= #M?34 - -s H%%H.N) __name__ __module__ __qualname__rr(r7r=rArdrmr~rrrrs.*0F P : &8t .rrctjdstdtdyt}tdt d}|j dk(r|j ytdy) zMain executionrz2Error: OPENAI_API_KEY environment variable not setz*Run: export OPENAI_API_KEY='your-key-here'NzrsD  LL\$ zFr