o hi @sddlmZmZddlmZmZmZddlmZeddddZ Gdd d eZ eZ e d d e fd d Z e dddZdS))FastAPI HTTPException) BaseModelField validator)LocalImageGeneratorzLocal Image Generation APIu@🚀 GPU-based Self-Hosted Image Generator (returns base64 only)z5.0.0)title descriptionversionc@seZdZUedddZeed<edddZeed<edddZeed <ed d dZ e ed <ed d dZ e ed<edddZ e ed<e d ddZe dddZdS) ImageRequest.zPrompt for image generation)r promptzVisual style or themecategorysdxlzModel: sdxl, sd15, realistic model_nameizImage width in pixelswidthzImage height in pixelsheightzNumber of images (max 3) image_countcCs"gd}||vrtd||S)N)rsd15 realisticzmodel_name must be one of:  ValueError)clsvalueallowed_modelsrE/var/www/eduai.edurigo.com/generate_image/testing/image_generation.pyvalidate_model_namesz ImageRequest.validate_model_namecCs(d|kr dkstdtd|S)Nrz#image_count must be between 1 and 3r)rrrrrvalidate_image_counts z!ImageRequest.validate_image_countN)__name__ __module__ __qualname__rr str__annotations__r rrintrrrrrrrrrr s  r z/generate-imagerequestc sz'tj|j|j|j|j|j|jd}|stddddt |dd|idWSt y?}z tdd t |dd }~ww) z= Generate one or more base64 images using local GPU. )r r rrrrzImage generation failed. status_codedetailsuccessz! image(s) generated successfully.images)statusmessagedatazError generating image: N) generatorgenerate_multiple_imagesr r rrrrrlen Exceptionr#)r&r,errrgenerate_image+s(  r5z/healthc sFz t}d|dWSty"}z tddt|dd}~ww)z#Check if GPU and model are healthy.r+)r-r/r'zHealth check failed: r(N)r0 check_healthr3rr#) health_statusr4rrr health_checkFs r8N)fastapirrpydanticrrrlocal_image_generatorrappr r0postr5getr8rrrrs