debug

Author: zhchaoxing

modules/api/api.py

@@ -133,7 +133,7 @@ def decode_base64_to_image(encoding):
 user_input_data = {}
 
 
-def set_img_exif_dict():
+def set_img_exif_dict(image_id="img_id_1"):
     print(f"log@{datetime.datetime.now().strftime(f'%Y%m%d%H%M%S')} ***&&& set_img_exif_dict Writing exif")
     title = "AIRI"
     date_taken = "2001:01:01 01:01:01"
@@ -146,10 +146,12 @@ def set_img_exif_dict():
     user_id = "AIRI tester"
     if "user_id" in user_input_data:
         user_id = user_input_data['user_id']
-    keywords = "Generated in AIRI platform. https://airilab.com"
-    description = "An image processed by the AIRI platform."
+    generation_id = "gen_id_1"
+    if "generation_id" in user_input_data:
+        generation_id = user_input_data['generation_id']
+    keywords = "Generated in AIRI platform. https://airilab.com . Generation ID: {generation_id}, Image ID: {image_id}"
+    description = f"An image processed by the AIRI platform. Generation ID: {generation_id}, Image ID: {image_id}"
     software = "AIRI Platform v1.0"
-    # imageid = "imageid?"
     # imagenum = "imagenum?"
     # seed = "seed?"
     exif_dict = {
@@ -231,12 +233,12 @@ def encode_pil_to_base64(image):
             # })
 
             # Convert dict to bytes
-            exif_bytes = piexif.dump(set_img_exif_dict())
+            exif_bytes = piexif.dump(set_img_exif_dict()) # Doesn't work here even though the exif is confirmed to have been added here
 
             if opts.samples_format.lower() in ("jpg", "jpeg"):
                 image.save(output_bytes, format="JPEG", exif=exif_bytes, quality=opts.jpeg_quality)
-                print(f"log@{datetime.datetime.now().strftime(f'%Y%m%d%H%M%S')} ***&&& encode_pil_to_base64")
-                print(Image.open(output_bytes).getexif())  # Print the EXIF data
+                # print(f"log@{datetime.datetime.now().strftime(f'%Y%m%d%H%M%S')} ***&&& encode_pil_to_base64")
+                # print(Image.open(output_bytes).getexif())  # Print the EXIF data
             else:
                 image.save(output_bytes, format="WEBP", exif=exif_bytes, quality=opts.jpeg_quality)
 
@@ -948,30 +950,24 @@ def post_invocations(self, b64images, quality):
                 key = key[ : -1]
             images = []
             for b64image in b64images:
+                image_id = datetime.datetime.now().strftime(f"%Y%m%d%H%M%S-{uuid.uuid4()}")
+                suffix = opts.samples_format.lower()
                 img = decode_to_image(b64image)
-                if opts.samples_format.lower() in ("jpg", "jpeg", "webp"):
-                    print(f"log@{datetime.datetime.now().strftime(f'%Y%m%d%H%M%S')} ***&&& post_invocations_before")
-                    # Create a file-like object from the image object
-                    # image_file = BytesIO(img.tobytes())
-                    # image = Image.open(image_file)
-                    # print(image.getexif())  # Print the EXIF data
-                    print(img.getexif())  # Print the EXIF data
+                if opts.samples_format.lower() in ("jpg", "jpeg"):
+                    # print(f"log@{datetime.datetime.now().strftime(f'%Y%m%d%H%M%S')} ***&&& post_invocations_before")
+                    # print(img.getexif())  # Print the EXIF data
 
                     # Save the image with the EXIF data
-                    exif_bytes = piexif.dump(set_img_exif_dict())
-                    # image.save(img, format="JPEG", exif=exif_bytes, quality=opts.jpeg_quality)
+                    exif_bytes = piexif.dump(set_img_exif_dict(image_id=image_id))
                     with io.BytesIO() as output_bytes:
                         img.save(output_bytes, format="JPEG", exif=exif_bytes, quality=opts.jpeg_quality)
                         output_bytes.seek(0)  # Go back to the start of the BytesIO object
-                        img_after = Image.open(output_bytes)
-                        print(f"log@{datetime.datetime.now().strftime(f'%Y%m%d%H%M%S')} ***&&& post_invocations_after")
-                        print(img_after.getexif())  # Print the EXIF data
+                        # img_after = Image.open(output_bytes)
+                        # print(f"log@{datetime.datetime.now().strftime(f'%Y%m%d%H%M%S')} ***&&& post_invocations_after")
+                        # print(img_after.getexif())  # Print the EXIF data
 
                 bytes_data = export_pil_to_bytes(img, quality)
                 # bytes_data = export_pil_to_bytes(decode_to_image(b64image), 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(
                     Body=bytes_data,
                     Bucket=bucket,