Update handler.py

handler.py

@@ -102,35 +102,25 @@ def check_safety(x_image):
 
 def upload_image(image_base64):
     try:
-        decode_time = TimeMesurer()
-        
         image_bytes = base64.b64decode(image_base64)
         pil_image = Image.open(BytesIO(image_bytes))
 
-        decode_time.log_time_end('decode_time');
-
-        check_safety_time = TimeMesurer()
         # check safety of the image
         is_nsfw = check_safety(pil_image)
-        check_safety_time.log_time_end('check_safety_time');
         
         postfix = ''
         if is_nsfw:
             postfix = '_nsfw'
 
-        webp_time = TimeMesurer()
         webp_image = BytesIO()
         pil_image.save(webp_image, format="WebP", lossless=convert_is_lossless, quality=convert_quality, method=convert_method)
         webp_image.seek(0) # reset the BytesIO object to the beginning
-        webp_time.log_time_end('webp_time');
 
         image_name = f'generated/{uuid.uuid4()}{postfix}.webp'
         blob = bucket.blob(image_name)
 
-        upload_time = TimeMesurer()
         blob.upload_from_file(webp_image, content_type='image/webp')
         blob.make_public()
-        upload_time.log_time_end('total_upload_time');
 
         return f'{images_base_url}/{image_name}'
     except Exception as e:
@@ -202,7 +192,7 @@ def handler(event):
         if images_field in result:
             update_progress_status(job_id=job_id, progress_status="UPLOADING_IMAGES")
             result[images_field] = upload_images_in_parallel(result[images_field])
-        upload_time.log_time_end("upload_time")
+        upload_time.log_time_end("total_upload_time")
 
         if image_field in result:
             update_progress_status(job_id=job_id, progress_status="UPLOADING_IMAGES")