Update files

utils/image_processing.py

@@ -2,33 +2,70 @@ import os
 from wand.image import Image
 from wand.color import Color
 
-def set_canvas_size(width, height):
-    global canvas_width, canvas_height
-    canvas_width = int(width)
-    canvas_height = int(height)
 
-def resize_image(image_path, output_path, additional_name):
+class ImageProcessor:
+    def __init__(
+        self, canvas_width=900, canvas_height=900, background_color="transparent"
+    ):
+        """
+        Initialize the ImageProcessor with default values.
+        """
+        self.canvas_width = canvas_width
+        self.canvas_height = canvas_height
+        self.background_color = background_color
 
-    # Normalize the paths to ensure consistency
-    image_path = os.path.normpath(image_path)
-    output_path = os.path.normpath(output_path)
+    def set_canvas_size(self, width, height):
+        """
+        Set the canvas size.
+        """
+        self.canvas_width = int(width)
+        self.canvas_height = int(height)
 
-    with Image(filename=image_path) as img:
-        img.transform(resize=f'{canvas_width}x{canvas_height}>')
-        
-        x_offset = int((canvas_width - img.width) / 2)
-        y_offset = int((canvas_height - img.height) / 2)
-        
-        with Image(width=canvas_width, height=canvas_height, background=Color('transparent')) as canvas:
-            canvas.composite(img, left=x_offset, top=y_offset)
-            # Create a new filename
-            new_filename = os.path.splitext(os.path.basename(output_path))[0]
-            if additional_name:
-                new_filename += " - " + additional_name.strip()
-            new_filename += os.path.splitext(output_path)[1]
-            # Construct the final output path
-            final_output_path = os.path.join(os.path.dirname(output_path), new_filename)
-            # Save the image to the final output path
-            canvas.save(filename=final_output_path)
-            print(f"Saved to: {final_output_path}")
-set_canvas_size(900, 900)
+    def set_background_color(self, color):
+        """
+        Set the background color.
+        """
+        self.background_color = Color(color)
+
+    def resize_image(self, image_path, output_path, additional_name=None):
+        """
+        Resize and process the image.
+        """
+        # Normalize the paths to ensure consistency
+        image_path = os.path.normpath(image_path)
+        output_path = os.path.normpath(output_path)
+        print(image_path)
+        print(output_path)
+        with Image(filename=image_path) as img:
+            img.transform(resize=f"{self.canvas_width}x{self.canvas_height}>")
+
+            x_offset = int((self.canvas_width - img.width) / 2)
+            y_offset = int((self.canvas_height - img.height) / 2)
+
+            with Image(
+                width=self.canvas_width,
+                height=self.canvas_height,
+                background=self.background_color,
+            ) as canvas:
+                canvas.composite(img, left=x_offset, top=y_offset)
+                # Create a new filename
+                new_filename = os.path.splitext(
+                    os.path.basename(output_path))[0]
+                if additional_name:
+                    new_filename += " - " + additional_name.strip()
+                new_filename += os.path.splitext(output_path)[1]
+                # Construct the final output path
+                final_output_path = os.path.join(
+                    os.path.dirname(output_path), new_filename
+                )
+                # Save the image to the final output path
+                canvas.save(filename=final_output_path)
+                print(f"Saved to: {final_output_path}")
+
+
+# Example usage
+if __name__ == "__main__":
+    processor = ImageProcessor()
+    processor.set_canvas_size(900, 900)
+    processor.set_background_color("white")
+    processor.resize_image("input_image.jpg", "output_image.jpg", "example")