Update main.py

main.py

@@ -8,6 +8,7 @@ import requests
 import boto3
 import base64
 from concurrent.futures import ThreadPoolExecutor, as_completed
+from meta_data import meta_data_helper_function
 
 load_dotenv()
 
@@ -79,13 +80,14 @@ def init_s3():
     )
 
 def upload_to_r2(image_bytes):
+    meta_data_img = meta_data_helper_function(image_bytes)
     s3_client = init_s3()
     filename = f"{uuid4().hex}.png"
     file_key = f"infinityverse/{filename}"
     s3_client.put_object(
         Bucket=os.getenv("R2_BUCKET_NAME"),
         Key=file_key,
-        Body=image_bytes,
+        Body=meta_data_img,
         ContentType="image/png",
     )
     return f'{os.getenv("NEW_BASE").rstrip("/")}/{file_key}'