o Z_h @sHdZddlZddlZddlZddlZddlZddlZddlmZddl m Z ddl m Z ddl Z ddl Z zddlmZWn&eybedddlZee jdd d d gddlmZed Ynwed eded eded edZdZdZdZdZdgdddddgdddddgdd d!dd"gd#d$d%dd&gd'd(d)dd*gd+d,d-dd.Zid/d0d1d0d2d0d3d0d4d0d5d6d7d6d8d6d9d6d:d6d;d<d=d<d>d<d?d<d@d<dAd<dBdCidDdCdEdCdFdCdGdHdIdHdJdHdKdLdMdLdNdLdOdLdPdLdQdLdRdLdSdLdTdLdUdLdVdLidWdLdXdLdYdLdZdLd[dLd\dLd]dLd^dLd_dLd`dLdad0dbd6dcd6ddd0ded0dfd6dgd6idhd0did0djd0dkdLdld0dmd0dnd0dod0dpd0dqdLdrdLdsdLdtdLdudLdvdLdwdLdxdLidydCdzdLd{dLd|dLd}dLd~dLdd0dd0dd<dd<dd0dd6dd6dd<dd6dd6dd0idd0dd0dd0dd6dd6dd6ddLddLddLddLddLddLdd0ddLddLddLddLiddLddLddLddLdd0ddLddHddLdd0ddLddLdd0dd0dd0dd0dd0dd6d6d6dHdZidZdd[dd^dd{ddUddTddvddaddhddlddfddddddqddtddzdddd_diZddZdd„ZddĄZddƄZ ddȄZ!GddʄdʃZ"dd̄Z#dd΄Z$ddЄZ%dd҄Z&ddԄZ'e(dkre'dSdS)u PROFESSIONAL IMAGE GENERATOR - ALL CATEGORIES VERSION ====================================================== Features: ✅ ALL Category Types Supported - No Restrictions ✅ Auto Model Selection Based on Category Keywords ✅ 100+ Category Types Enabled ✅ Intelligent Category Mapping ✅ 1920x1080 Full HD Resolution ✅ Professional Quality Output Author: Professional Image Generation System Date: 2025 N)Path)datetime)BytesIO)Imageu📦 Installing Pillow...z-mpipinstallpillowu✅ Pillow installed! P================================================================================z:PROFESSIONAL IMAGE GENERATOR - ALL CATEGORIES VERSION V3.0uH✅ Full HD 1920x1080 | ALL Categories Supported | Smart Model Selectionz%https://image.pollinations.ai/prompt/professional_imagesii8Tz flux-realism) realisticphotorealistichyperrealisticphoto photographyzultra photorealistic, professional photography, extremely detailed, sharp focus, perfect lighting, 8K UHD, RAW photo, high quality, film grain, Fujifilm XT3, professional color grading, award winning photographyzWcartoon, anime, illustration, painting, drawing, art, sketch, 3d render, deformed, ugly)modelkeywordssuffixnegativezflux-3d)3dcgirenderrendered 3d-renderoctanez3D render, octane render, unreal engine 5, ray tracing, ultra detailed 3D model, CGI, volumetric lighting, ambient occlusion, subsurface scattering, physically based rendering, 8K resolution, studio lightingzA2d, flat, cartoon, anime, painting, sketch, low poly, low qualityz flux-cablyai) cartoonisticcartoonanimated animationdisneypixartoonzcartoon style, disney pixar style, animated, colorful, clean lines, vibrant colors, professional animation, studio quality, character design, expressive, polished, high quality cartoonz7realistic, photo, 3d render, dark, gritty, horror, uglyz flux-cartoon)comic comic-book graphic-novelpop-artgraphiccomicszcomic book style, comic art, graphic novel, bold lines, vibrant colors, dynamic composition, pop art, professional comic illustration, ink and color, halftone dotsz0realistic, photo, 3d render, blurry, low qualityz flux-anime)animemanga anime-style manga-stylezjapanese-animezanime style, manga art, japanese animation, cel shaded, vibrant colors, expressive eyes, detailed hair, professional anime artwork, studio quality, beautiful lightingzprofessional portrait, studio lighting, shallow depth of fieldz4beautiful landscape, scenic view, nature photographyzIproduct shot, clean background, professional lighting, commercial qualityz@architectural photography, professional composition, clean lineszEfashion photography, haute couture, editorial style, magazine qualityz4vintage style, retro aesthetic, aged look, nostalgicz3abstract art, artistic composition, creative designz4minimalist design, clean composition, simple elegantz1dark atmosphere, moody lighting, dramatic shadowsz.pixel art style, retro gaming, 8-bit aestheticz@steampunk aesthetic, Victorian era, brass and copper, mechanicalcs(|}|ddddtvrtStvrStD]\}}|dD]}|vs4|vr:|Sq*q"tfdddDrIdStfd dd DrVd Stfd dd DrcdStfdddDrpdStfdddDr}dStfdddDrdStd|ddS)z Intelligently detect which main category to use based on input Args: category: Input category string Returns: Main category name  -_rc3|]}|vVqdSN.0wordcategory_normalizedrT/var/www/eduai.edurigo.com/generate_image/production/pollinations_image_generator.py Tz'detect_category_type..)realrr c3rrrrrrrrVr)rrrrc3rrrrrrrrXr)ranimrrc3rrrrrrrrZr)r!r(r!c3rrrrrrrr\r)r'r(r'c3rrrrrrrr^r)cinemar.r/r,u ℹ️ Unknown category 'z', using 'cinematic' as default)lowerstripreplaceEXTENDED_CATEGORIESMAIN_CATEGORIESitemsanyprint)categorycategory_lowermain_catconfigkeywordrrrdetect_category_type8s4   rc Cs|dddd}t|}t|}d}|d}d}|tvr)dt|}d}|d|d||d|} |d krD| d 7} | S|d krN| d 7} | S|d krX| d7} | S|dkrb| d7} | S|dkrl| d7} | S|dkrt| d7} | S)z Enhance prompt based on category with intelligent additions Args: prompt: Original user prompt category: Selected category Returns: Enhanced prompt string rrrzLmasterpiece, best quality, ultra detailed, professional, perfect compositionr, z>perfect anatomy, perfect perspective, golden ratio compositionr zB, shot on Canon EOS R5, 85mm lens, f/1.4, professional photographyrzD, rendered in Octane, global illumination, HDRI lighting, 8K qualityr,z4, cinematic composition, film grain, anamorphic lensr'z#, trending on pixiv, studio qualityrz), Disney animation studios, Pixar qualityr!z), professional comic artist, Marvel style)rrrrrCATEGORY_ENHANCEMENTS) promptrr main_categorycategory_configquality_prefix style_suffixextra_enhancement technicalenhancedrrrenhance_prompt_for_categoryfs8   rcCst|}t|ddS)z Get negative prompt based on category Args: category: Category name Returns: Negative prompt string rr)rrgetrrrrrget_negative_prompts rcCst|}t|dS)z Get the best model for a given category Args: category: Category name Returns: Model name r)rrrrrrget_model_for_categorys rc Csrt|}ddddddd}||d\}}tt|d d d }td d }||||}||} | S)z Generate optimized seed based on category and prompt Args: category: Category name prompt: Prompt text Returns: Optimized seed value ))r')r:)r N)ra)r0ur2)rrNrr) rrinthashlibmd5encode hexdigestrandomrandint) rrr seed_rangesmin_seedmax_seed prompt_hashrandom_component base_seedoptimized_seedrrrget_optimized_seeds   rc@sNeZdZdZddZddZdddZdd d Zdd dZddZ ddZ dS)ProfessionalImageGeneratorzBProfessional image generation system supporting ALL category typescCs:d|_|tdtdtdtdtddS)Nru*✅ ProfessionalImageGenerator initializedu📁 Output folder: /u/🎨 ALL Categories Supported - No Restrictionsu(🤖 Intelligent Model Selection Enabled)generation_countcreate_output_foldersr OUTPUT_FOLDERselfrrr__init__s  z#ProfessionalImageGenerator.__init__cCsftt}|jddtD] }||jddq|djdd|djdd|djdddS)z(Create organized output folder structureT)exist_okothermetadatatestsN)rrmkdirrkeys)r base_pathrrrrrs  z0ProfessionalImageGenerator.create_output_foldersNc CsH|durt}|dur t}|jd7_|}t|}|}t||}t|} t||} t t j |} |||| ddt tdd} | rL| | d<td|jtd|d|td |td |d |td | td |dddtdzet j| | dd} | jdkrtt| j}|jdkr|d}t}|j|dddd|d|}t|d}tdtd|ddtd|WStd | jtd!| jddWdSt jjytd"YdSt y#}ztd#t |ddl!}|"WYd}~dSd}~ww)$a Generate single image and return bytes for FastAPI Accepts ANY category type - no restrictions Args: prompt: The text prompt category: ANY category name (used for prompt enhancement) model_name: Specific model to use (flux, nano-banana, turbo) width: Image width (default 1920) height: Image height (default 1080) Returns: Image bytes or None if failed Ntrue)widthheightrseednologoenhancesafenofeednegative_promptu 🎨 Generating Image #z Category: u → Main: z Model: z Resolution: xz Seed: z Prompt: <z...u$ ⏳ Processing (15-30 seconds)...x)paramstimeoutRGBPNGdF)formatqualityoptimizeru ✅ SUCCESSu 📊 Size: z.2fz MBu 🎯 Quality: Professional HDu ❌ API Error: HTTP z Response: u4 ❌ Timeout: Request took longer than 120 secondsu ❌ Generation Failed: )# QUALITY_WIDTHQUALITY_HEIGHTrrrrrrrAPI_URLrequestsutilsquotestrSAFETY_ENABLEDrupperr status_coderopenrcontentmodeconvertsaveseekgetvaluelentext exceptionsTimeout Exception traceback print_exc)rrr model_namerrrrenhanced_promptrrurlrresponseimageoutput image_bytes image_size_mberrrrgenerate_single_image_bytesst          z6ProfessionalImageGenerator.generate_single_image_bytesrc Cs6g}t|D]}td|dd||j|||||d} | dur,td|dqtddd} t|d | dd } | dd } d |  d d  dd dd} |d | d | d |dd | }t| d}|||dtd|||dkrtdq|S)a Generate multiple images and return list with names and base64 data Args: prompt: The text prompt category: Category name for enhancement model_name: Model to use width: Image width height: Image height image_count: Number of images to generate Returns: List of dicts with 'name' and 'base64' keys u 📸 Generating image rr)rrr rrNu ❌ Failed to generate image %Y%m%d_%H%M%S_%fr,r.zutf-8)namebase64u ✅ Generated: )rangerrrnowstrftimerrrrsplitjoinrrr b64encodedecodeappendtimesleep)rrrr rr image_countimagesir timestampr prompt_words prompt_brief image_name base64_datarrrgenerate_multiple_imagesYs8 $&$  z3ProfessionalImageGenerator.generate_multiple_imagesr+c Cs|durt}|dur t}||||||}|durdddSt|}|tvr)|}nd}tddd} t |  dd} | dd } d | d d d d dd} |d | d | d | d} tt|| }t|d }||Wdn1swYt|jd|t||t|d||t||d|t|dt||dd}ttd| dd}t|d}tj||ddWdn1swYtd|td|dt|t||dS) a' Generate image and save to disk Args: prompt: The text prompt category: Category name model_name: Model to use width: Image width height: Image height Returns: Dictionary with result info NFzImage generation failed)successerrorrrrrrrrrrz.pngwb)r*generation_number)originalrrrr)rrr resolution file_size_mbr)generation_info prompt_datatechnical_detailsrz.jsonw)indentu 💾 Saved: u 📄 Metadata: T)r0path metadata_pathr)rrrrrrrrrrrrr r!rrrrrwrite isoformatrrrrrrjsondumprr)rrrr rrrrfolderr*rr+r,filename output_pathfrr?rrrgenerate_and_save_images\ &      z2ProfessionalImageGenerator.generate_and_save_imagecCs(ddtDtttttddS)z.Get information about all supported categoriescSs.i|]\}}||d|dd|ddqS)r descriptionrr)rrIrr)rrrrrr s zFProfessionalImageGenerator.get_all_categories_info..zdAll category types are supported. Unknown categories will be intelligently mapped to the best model.)main_categoriesextended_categoriestotal_supportedmessage)rrrrrrrrget_all_categories_infosz2ProfessionalImageGenerator.get_all_categories_infocCs tdtdt|dtdtd|tdg}t|dD]4\}}td|dt|d |td |||}|||d |t|krXtd td q$tdd|D}tdtdtdtd|dt|tdt||dt|td|S)z Test generation with multiple categories Args: prompt: Prompt to use categories_list: List of categories to test Returns: List of results Q ================================================================================u 🧪 TESTING z CATEGORIESr u 📝 Prompt: r [rz ] Testing: <------------------------------------------------------------)rresult ⏳ Waiting 2 seconds...r<css$|] }|dddrdVqdS)rTr0FrNrJrrrrrrs"zFProfessionalImageGenerator.test_multiple_categories..u✅ TEST COMPLETE Success: zFailed: )rr enumeraterHr$r%r&sum)rrcategories_listresultsidxrrT successfulrrrtest_multiple_categoriess4     z3ProfessionalImageGenerator.test_multiple_categories)NN)NNr)r+NN) __name__ __module__ __qualname____doc__rrrr/rHrPr_rrrrrs  h 8N rc CsVtdtdtdt}dddddddd dd d dd d ddddddddddddddddg }g}t|dD]C\}}td|dt|d|dtd|dtd|j|d|dd}|||t|krtd td!q@t d"d#|D}td$td%tdtd&|dt|tdd'S)(zTest most popular categoriesr zTESTING POPULAR CATEGORIESz!A beautiful sunset over mountainsr rrzA futuristic robot characterrzA cute cartoon dog playingrzA superhero in action poser!zAn anime girl with blue hairr'zAn epic battle scener,zA magical fantasy castler9zA cyberpunk city at nightr<zA professional portrait photor?z A watercolor painting of flowersr7rrRr] rzPrompt: rrSrUr<css |] }|ddrdVqdS)r0FrNrJrVrrrrHsz*test_popular_categories..rQu$✅ POPULAR CATEGORIES TEST COMPLETErXN) rrrYrrrHr$r%r&rZ) generator test_casesr\r]testrTr^rrrtest_popular_categories sF &    ricCs:tdtdtdt}gd}d}|||}|S)z6Test various category types to demonstrate flexibilityr zTESTING ALL CATEGORY TYPES)r rrr!r'r,r9r:r<r7r6rRr?rErIrCrVrZr_raz'A majestic dragon in a mystical settingrrr_)rftest_categoriesrr\rrrtest_all_category_typesPs rlc Cstdtdtdt}dddddddd dg}t|d D]-\}}td |d t|d |d|j|d|dd}|t|krOtdq"tdtdtddS)z"Quick test with 3 different stylesr zQUICK TEST - 3 CATEGORIESzA red sports carr rdzA cute robot characterrzA magical forest scener9rrRrrerrr<rQu✅ QUICK TEST COMPLETEN)rrrYrrrHr%r&)rfrgr]rhrTrrr quick_tesths(&   rmcCsJtdtdtdtdtdt}gd}d}|||}|S)z*Test with completely custom category namesr zTESTING CUSTOM CATEGORY NAMESz7Testing system's ability to handle any category name...) zepic-battle-scenezprofessional-headshotzgame-character-designzbook-cover-artz movie-posterz album-coverzt-shirt-designz logo-designzchildren-book-illustrationzscientific-diagramz"A creative and professional designrj)rfcustom_categoriesrr\rrrtest_custom_categoriess roc CstdtdtdtdtdtdtdtdtdtdtD]\}}td |d |d q,tdtd tdgd }td|tdtdtdtdtdtdtdtdtdtdtdtdtd}|dkrtn|dkrtn|dkrt n|dkrt n|dkrtd}td}t }| ||n|d kr6t }| }tdtd!tdtd"t|d#|d#D] }td |qtd$t|d%t|d%}td&t|d'D]} td(d|| | d'qtd)|d*tdn|d+krAtd,d-Std.d-Stdtd/td0td1tdd-S)2z#Main function with interactive menurQz5PROFESSIONAL IMAGE GENERATOR - ALL CATEGORIES VERSIONr u! ✅ ALL CATEGORY TYPES SUPPORTEDu0✅ NO RESTRICTIONS - ANY CATEGORY NAME ACCEPTEDu✅ INTELLIGENT MODEL SELECTIONu✅ 100+ CATEGORY MAPPINGSzAVAILABLE MAIN CATEGORIESu • z: rzEXAMPLE SUPPORTED CATEGORIES)r rrr!r'r,r9r:r<r7r6rRr?rErVrZr_rarIrCr=rUr[rMrPr^r(rrrz4 ...and MANY MORE! System accepts ANY category name.z TESTING MENUu1️⃣ Quick Test (3 images)u,2️⃣ Test Popular Categories (10 images)u,3️⃣ Test All Category Types (20 images)u+4️⃣ Test Custom Categories (10 images)u!5️⃣ Custom Single Generationu&6️⃣ Show All Supported Categoriesu 7️⃣ Exitz Select option (1-7): 12345z Enter prompt: zEnter category (any name): 6zALL SUPPORTED CATEGORIESz Main Categories: rLz Extended Categories: rMrz  rO7u 👋 Goodbye!Nu ❌ Invalid optionu🎉 OPERATION COMPLETEu📁 Check images in: r)rrrrr!inputrrmrirlrorrHrPrrlistrr) rrexample_categorieschoicerrrfinfor[r)rrrmains    $   r~__main__))rcrrBr%rrrpathlibrriorsysrePILr ImportErrorr subprocess check_call executablerrrrrrrrrrrrrrrirlrmror~r`rrrrsD            3      #$%&'*+,-./01256789:;>?@ABEFGHIJMNOPQTUVWX[\]^abcdehijklopqrsvwxy|}~    !"#$%&,     .3#J0" U