o X h} @sbddlZddlmZddlmZddlmZddlmZddlm Z ddl m Z Gdd d Z dS) N)datetime) AsyncSession) Prediction)save_input_images_async)logger) UploadFilec@s0eZdZdedefddZdedefddZd S) SavingInputImagedbidcCs||_||_dS)N)r r )selfr r r M/var/www/html/construction_image-detection-poc/src/components/input_images.py__init__ s zSavingInputImage.__init__ before_image after_imagec std}tjd|}ztj|ddtd|tj|d}tj|d}z)t |d}| | IdHWdn1sIwYtd |Wnt ym}z t d t d |d}~wwz)t |d} | | IdHWdn1swYtd |Wnt y}z t d t d |d}~wwzt|IdH\} } td | d| Wnt y}z t dt d|d}~ww|j} z,t| | | iddd} |j| |jIdH|j| IdHtd| jWnt y!}z t dt d|d}~wwd| j| | ||dWSt yG}zt dt dt|d}~ww)Nz %Y%m%d_%H%M%S Input_ImagesT)exist_okz!Created folder for input images: z before.jpgz after.jpgwbzSaved 'before' image at: zFailed to save 'before' image.zSaved 'after' image at: zFailed to save 'after' image.zGenerated URLs - Before: z , After: zFailed to generate image URLs.pending)user_idbefore_image_urlafter_image_url text_resultsoutput_image_urlstatuszSaved prediction with ID: z*Failed to save prediction to the database.zDatabase operation failed.zImages uploaded successfully!)message prediction_id before_url after_url before_path after_pathz-Error occurred while processing input images.zError while saving images: )rnowstrftimeospathjoinmakedirsrinfoopenwriteread Exception exceptionrr rr addcommitrefreshstr)r rr timestamp folder_pathr r!bfeafrrr predictionr r r saving_input_images s            z$SavingInputImage.saving_input_imagesN)__name__ __module__ __qualname__rintrrr8r r r r r sr) r$rsqlalchemy.ext.asynciorsrc.models.predictionrsrc.utils.files_utilsrsrc.common.loggerrfastapirrr r r r s