debug

Author: zhchaoxing

modules/api/api.py

@@ -948,9 +948,18 @@ def post_invocations(self, b64images, quality):
                 key = key[ : -1]
             images = []
             for b64image in b64images:
-                bytes_data = export_pil_to_bytes(decode_to_image(b64image), quality)
-                print(f"log@{datetime.datetime.now().strftime(f'%Y%m%d%H%M%S')} ***&&& post_invocations")
-                print(Image.open(BytesIO(bytes_data)).getexif())  # Print the EXIF data
+                print(f"log@{datetime.datetime.now().strftime(f'%Y%m%d%H%M%S')} ***&&& post_invocations_before")
+                img = decode_to_image(b64image)
+                print(Image.open(img).getexif())  # Print the EXIF data
+
+                if opts.samples_format.lower() in ("jpg", "jpeg", "webp"):
+                    exif_bytes = piexif.dump(set_img_exif_dict())
+                    img.save(img, format="JPEG", exif=exif_bytes, quality=opts.jpeg_quality)
+                    print(f"log@{datetime.datetime.now().strftime(f'%Y%m%d%H%M%S')} ***&&& post_invocations_after")
+                    print(Image.open(img).getexif())  # Print the EXIF data
+                    
+                bytes_data = export_pil_to_bytes(img, quality)
+
                 image_id = datetime.datetime.now().strftime(f"%Y%m%d%H%M%S-{uuid.uuid4()}")
                 suffix = opts.samples_format.lower()
                 shared.s3_client.put_object(