Image compresson configured

Author: MySecondLanguage

nxtbn/filemanager/api/dashboard/serializers.py

@@ -1,21 +1,19 @@
 from django.utils.translation import gettext_lazy as _
 from rest_framework import serializers
+from nxtbn import settings
 from nxtbn.filemanager.models import Image, Document
 
 
 from PIL import Image as PILImage
 from io import BytesIO
 from django.core.files.base import ContentFile
 
+
 class ImageSerializer(serializers.ModelSerializer):
     class Meta:
         model = Image
         fields = "__all__"
-        read_only_fields = (
-            "id",
-            "created_by",
-            "last_modified_by",
-        )
+        read_only_fields = ("id", "created_by", "last_modified_by")
 
     def create(self, validated_data):
         request = self.context["request"]
@@ -25,6 +23,7 @@ def create(self, validated_data):
         # Optimize the image before saving
         if "image" in validated_data:
             validated_data["image"] = self.optimize_image(validated_data["image"])
+            validated_data["image_xs"] = self.optimize_image(validated_data["image"], max_size_kb=1, format="png", max_dimension=50)
 
         return super().create(validated_data)
 
@@ -39,28 +38,38 @@ def update(self, instance, validated_data):
         return super().update(instance, validated_data)
 
     @staticmethod
-    def optimize_image(image_file, max_size_kb=200, format="WEBP"):
+    def optimize_image(image_file, max_size_kb=settings.IMAGE_COMPRESS_MAX, format="WEBP", max_dimension=800):
         """
         Optimize an image by resizing and converting it to the specified format while maintaining the aspect ratio.
         Ensures the image file size is below max_size_kb.
         """
         img = PILImage.open(image_file)
-        img = img.convert("RGB")  # Ensure compatibility for formats like PNG with alpha
+        img = img.convert("RGB")  # Convert to RGB for compatibility
+
+        # Adjust dimension based on image type (smaller for PNG to ensure < 2023 bytes)
+        if format.lower() == "png":
+            max_dimension = 64  # Reduce size for PNG to keep it small
 
-        # Resize the image to fit within a reasonable dimension (adjust if needed)
-        max_dimension = 800
         img.thumbnail((max_dimension, max_dimension), PILImage.Resampling.LANCZOS)
 
-        # Save the image to a buffer
+        # Use a buffer to store the optimized image
         buffer = BytesIO()
-        img.save(buffer, format=format, optimize=True, quality=85)
-        buffer.seek(0)
 
-        # Check if the image exceeds the maximum size, reduce quality iteratively
-        while buffer.tell() > max_size_kb * 1024:
-            buffer.seek(0)
-            img.save(buffer, format=format, optimize=True, quality=max(10, 85 - 10))
-            buffer.seek(0)
+        if format.lower() == "png":
+            # Reduce colors using quantization (important for PNG)
+            img = img.convert("P", palette=PILImage.ADAPTIVE, colors=32)  # Limit colors to 32
+            img.save(buffer, format="PNG", optimize=True)
+        else:
+            quality = 85
+            img.save(buffer, format=format, optimize=True, quality=quality)
+
+            # Reduce quality iteratively for WebP/JPEG if size is too large
+            while buffer.tell() > max_size_kb * 1024 and quality > 10:
+                quality -= 5
+                buffer = BytesIO()  # Reset buffer
+                img.save(buffer, format=format, optimize=True, quality=quality)
+
+        buffer.seek(0)
 
         # Return the new image as a ContentFile
         return ContentFile(buffer.read(), name=f"{image_file.name.split('.')[0]}.{format.lower()}")

nxtbn/filemanager/migrations/0003_image_image_sm.py

@@ -0,0 +1,18 @@
+# Generated by Django 4.2.11 on 2025-02-20 15:56
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ("filemanager", "0002_initial"),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name="image",
+            name="image_sm",
+            field=models.ImageField(blank=True, null=True, upload_to=""),
+        ),
+    ]

nxtbn/filemanager/migrations/0004_rename_image_sm_image_image_xs.py

@@ -0,0 +1,18 @@
+# Generated by Django 4.2.11 on 2025-02-20 16:00
+
+from django.db import migrations
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ("filemanager", "0003_image_image_sm"),
+    ]
+
+    operations = [
+        migrations.RenameField(
+            model_name="image",
+            old_name="image_sm",
+            new_name="image_xs",
+        ),
+    ]

nxtbn/filemanager/models.py

@@ -9,6 +9,7 @@ class Image(AbstractBaseModel):
     last_modified_by = models.ForeignKey(User, on_delete=models.PROTECT, related_name='image_modified', null=True, blank=True)
     name = models.CharField(max_length=255)
     image = models.ImageField()
+    image_xs = models.ImageField(null=True, blank=True)
     image_alt_text = models.CharField(max_length=255)
 
 

nxtbn/settings.py

@@ -404,6 +404,7 @@ def get_env_var(key, default=None, var_type=str):
 # ============================
 # NXTBN Specific Configuration
 # ============================
+IMAGE_COMPRESS_MAX  = get_env_var("IMAGE_COMPRESS_MAX", default=200, var_type=int)  # in KB
 
 PLUGIN_BASE_DIR = 'nxtbn.plugins.sources'