Making image manipulations thru one class

src/model/inference.py

@@ -5,12 +5,12 @@ Handles detection inference and result storage.
 
 from typing import List, Dict, Optional, Callable
 from pathlib import Path
-from PIL import Image
 import cv2
 import numpy as np
 
 from src.model.yolo_wrapper import YOLOWrapper
 from src.database.db_manager import DatabaseManager
+from src.utils.image import Image
 from src.utils.logger import get_logger
 from src.utils.file_utils import get_relative_path
 
@@ -64,9 +64,9 @@ class InferenceEngine:
                 stored_relative_path = str(Path(image_path).resolve())
 
             # Get image dimensions
-            img = Image.open(image_path)
-            width, height = img.size
-            img.close()
+            img = Image(image_path)
+            width = img.width
+            height = img.height
 
             # Perform detection
             detections = self.yolo.predict(image_path, conf=conf)