o Z_h< @sxddlmZmZddlmZmZmZddlmZddl Z eddddZ Gd d d eZ eZ e d d e fd dZdS))FastAPI HTTPException) BaseModelField validator)ProfessionalImageGeneratorNzImage Generation APIzJAPI for generating images using Pollinations.ai - All Categories Supportedz3.0.0)title descriptionversionc@seZdZUedddZeed<edddZeed<edddZeed<ed d dZ e ed <ed d dZ e ed<edddZ e ed<e dddZe dddZe d dddZGdddZdS) ImageRequest.z$The text prompt for image generation)r promptzCategory/style for the imagecategoryz'Model name: flux, nano-banana, or turbo model_namezImage width in pixelswidth8zImage height in pixelsheightz$Number of images to generate (max 3) image_countcCs(gd}||vrtdd||S)N)fluxz nano-bananaturbozmodel_name must be one of: z, ) ValueErrorjoin)clsvallowed_modelsrH/var/www/eduai.edurigo.com/generate_image/production/image_generation.pyvalidate_model_namesz ImageRequest.validate_model_namecCs(d|kr dkstdtd|S)Nrz#image_count must be between 1 and 3rrrrrrvalidate_image_counts z!ImageRequest.validate_image_countcCs|dkrtd|S)Nrz*width and height must be positive integersr r!rrrvalidate_dimensions!sz ImageRequest.validate_dimensionsc@s"eZdZddddddddiZd S) zImageRequest.ConfigexampleziGenerate a professional online training session with diverse learners interacting through virtual screens realisticrrrrr r rrrrN)__name__ __module__ __qualname__json_schema_extrarrrrConfig'sr+N)r'r(r)rr str__annotations__r rrintrrrrr"r#r+rrrrr s    r z/generate-imagerequestc sz"tj|j|j|j|j|j|j d}|st dddd|iWSt y+t yA}z t ddt |dd}~ww)a Generate images based on the provided parameters. - **prompt**: The text prompt for image generation - **category**: Category/style for the image (accepts any category type) - **model_name**: Model to use (flux, nano-banana, turbo) - **width**: Image width in pixels - **height**: Image height in pixels - **image_count**: Number of images to generate (1-3) Returns a list of generated images with names and base64 data. r&izImage generation failed) status_codedetailimageszError generating image: N) generatorgenerate_multiple_imagesr r lowerstriprrrrr Exceptionr,)r/r2errrgenerate_image6s&    r9)fastapirrpydanticrrrpollinations_image_generatorrbase64appr r3postr9rrrrs (