Update utils.py

utils.py

@@ -5,7 +5,7 @@ import numpy as np
 import json
 import torch
 import uuid
-from PIL import Image, PngImagePlugin
+from PIL import Image
 from datetime import datetime
 from dataclasses import dataclass
 from typing import Callable, Dict, Optional, Tuple
@@ -162,21 +162,26 @@ def preprocess_image_dimensions(width, height):
 def save_image(image, metadata, output_dir, is_colab):
     if is_colab:
         current_time = datetime.now().strftime("%Y%m%d_%H%M%S")
-        filename = f"image_{current_time}.png"   
+        filename = f"image_{current_time}.jpg"   
     else:
-        filename = str(uuid.uuid4()) + ".png"
+        filename = str(uuid.uuid4()) + ".jpg"
     os.makedirs(output_dir, exist_ok=True)
     filepath = os.path.join(output_dir, filename)
-    metadata_str = json.dumps(metadata)
-    info = PngImagePlugin.PngInfo()
-    info.add_text("metadata", metadata_str)
-    image.save(filepath, "PNG", pnginfo=info)
-    return filepath
     
+    # Lưu metadata dưới dạng tệp văn bản đi kèm
+    metadata_str = json.dumps(metadata)
+    metadata_filepath = os.path.join(output_dir, f"{filename}.json")
+    with open(metadata_filepath, 'w') as f:
+        f.write(metadata_str)
     
+    # Lưu hình ảnh dưới định dạng JPEG
+    image.save(filepath, "JPEG")
+    return filepath
+
+
 def is_google_colab():
     try:
         import google.colab
         return True
     except:
-        return False
+        return False