ImagePRO-Python 0.1.0__tar.gz → 0.1.1__tar.gz

{imagepro_python-0.1.0/src/ImagePRO_Python.egg-info → imagepro_python-0.1.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ImagePRO-Python
-Version: 0.1.0
+Version: 0.1.1
 Summary: clean, modular, and easy-to-use Python library for image processing tasks.
 Author-email: Parsa Safaie <parsasafaie.2568@proton.me>
 License: MIT

{imagepro_python-0.1.0 → imagepro_python-0.1.1}/pyproject.toml

@@ -7,7 +7,7 @@ where = ["src"]
 
 [project]
 name = "ImagePRO-Python"
-version = "0.1.0"
+version = "0.1.1"
 description = "clean, modular, and easy-to-use Python library for image processing tasks."
 readme = "README.md"
 requires-python = "==3.10.*"

{imagepro_python-0.1.0 → imagepro_python-0.1.1}/src/ImagePRO/human_analysis/body_analysis/body_pose_estimation.py

@@ -8,7 +8,7 @@ import mediapipe as mp
 parent_dir = Path(__file__).resolve().parent.parent.parent
 sys.path.append(str(parent_dir))
 
-from io_handler import IOHandler
+from utils.io_handler import IOHandler
 
 mp_pose = mp.solutions.pose
 

{imagepro_python-0.1.0 → imagepro_python-0.1.1}/src/ImagePRO/human_analysis/body_analysis/hand_tracking.py

@@ -8,7 +8,7 @@ import mediapipe as mp
 parent_dir = Path(__file__).resolve().parent.parent.parent
 sys.path.append(str(parent_dir))
 
-from io_handler import IOHandler
+from utils.io_handler import IOHandler
 
 mp_hands = mp.solutions.hands
 

{imagepro_python-0.1.0 → imagepro_python-0.1.1}/src/ImagePRO/human_analysis/face_analysis/eye_status_analysis.py

@@ -8,7 +8,7 @@ import mediapipe as mp
 parent_dir = Path(__file__).resolve().parent.parent.parent
 sys.path.append(str(parent_dir))
 
-from io_handler import IOHandler
+from utils.io_handler import IOHandler
 from human_analysis.face_analysis.face_mesh_analysis import analyze_face_mesh
 
 mp_face_mesh = mp.solutions.face_mesh

{imagepro_python-0.1.0 → imagepro_python-0.1.1}/src/ImagePRO/human_analysis/face_analysis/face_detection.py

@@ -8,7 +8,7 @@ import numpy as np
 parent_dir = Path(__file__).resolve().parent.parent.parent
 sys.path.append(str(parent_dir))
 
-from io_handler import IOHandler
+from utils.io_handler import IOHandler
 from human_analysis.face_analysis.face_mesh_analysis import analyze_face_mesh
 
 

{imagepro_python-0.1.0 → imagepro_python-0.1.1}/src/ImagePRO/human_analysis/face_analysis/face_mesh_analysis.py

@@ -8,7 +8,7 @@ import mediapipe as mp
 parent_dir = Path(__file__).resolve().parent.parent.parent
 sys.path.append(str(parent_dir))
 
-from io_handler import IOHandler
+from utils.io_handler import IOHandler
 
 mp_face_mesh = mp.solutions.face_mesh
 mp_drawing_utils = mp.solutions.drawing_utils

{imagepro_python-0.1.0 → imagepro_python-0.1.1}/src/ImagePRO/human_analysis/face_analysis/head_pose_estimation.py

@@ -8,7 +8,7 @@ import mediapipe as mp
 parent_dir = Path(__file__).resolve().parent.parent.parent
 sys.path.append(str(parent_dir))
 
-from io_handler import IOHandler
+from utils.io_handler import IOHandler
 from human_analysis.face_analysis.face_mesh_analysis import analyze_face_mesh
 
 mp_face_mesh = mp.solutions.face_mesh

{imagepro_python-0.1.0 → imagepro_python-0.1.1}/src/ImagePRO/pre_processing/blur.py

@@ -8,7 +8,7 @@ import numpy as np
 parent_dir = Path(__file__).resolve().parent.parent
 sys.path.append(str(parent_dir))
 
-from io_handler import IOHandler
+from utils.io_handler import IOHandler
 
 
 def apply_average_blur(kernel_size=(5, 5), image_path=None, np_image=None, result_path=None):

{imagepro_python-0.1.0 → imagepro_python-0.1.1}/src/ImagePRO/pre_processing/contrast.py

@@ -9,7 +9,7 @@ parent_dir = Path(__file__).resolve().parent.parent
 sys.path.append(str(parent_dir))
 
 from pre_processing.grayscale import convert_to_grayscale
-from io_handler import IOHandler
+from utils.io_handler import IOHandler
 
 
 def apply_clahe_contrast(

{imagepro_python-0.1.0 → imagepro_python-0.1.1}/src/ImagePRO/pre_processing/crop.py

@@ -5,7 +5,7 @@ from pathlib import Path
 parent_dir = Path(__file__).resolve().parent.parent
 sys.path.append(str(parent_dir))
 
-from io_handler import IOHandler
+from utils.io_handler import IOHandler
 
 
 def crop_image(start_point, end_point, image_path=None, np_image=None, result_path=None):

{imagepro_python-0.1.0 → imagepro_python-0.1.1}/src/ImagePRO/pre_processing/grayscale.py

@@ -7,7 +7,7 @@ import cv2
 parent_dir = Path(__file__).resolve().parent.parent
 sys.path.append(str(parent_dir))
 
-from io_handler import IOHandler
+from utils.io_handler import IOHandler
 
 
 def convert_to_grayscale(image_path=None, np_image=None, result_path=None):

{imagepro_python-0.1.0 → imagepro_python-0.1.1}/src/ImagePRO/pre_processing/resize.py

@@ -7,7 +7,7 @@ import cv2
 parent_dir = Path(__file__).resolve().parent.parent
 sys.path.append(str(parent_dir))
 
-from io_handler import IOHandler
+from utils.io_handler import IOHandler
 
 
 def resize_image(new_size, image_path=None, np_image=None, result_path=None):

{imagepro_python-0.1.0 → imagepro_python-0.1.1}/src/ImagePRO/pre_processing/rotate.py

@@ -7,7 +7,7 @@ import cv2
 parent_dir = Path(__file__).resolve().parent.parent
 sys.path.append(str(parent_dir))
 
-from io_handler import IOHandler
+from utils.io_handler import IOHandler
 
 
 def rotate_image_90(image_path=None, np_image=None, result_path=None):

{imagepro_python-0.1.0 → imagepro_python-0.1.1}/src/ImagePRO/pre_processing/sharpen.py

@@ -8,8 +8,8 @@ import numpy as np
 parent_dir = Path(__file__).resolve().parent.parent
 sys.path.append(str(parent_dir))
 
-from io_handler import IOHandler
-from blur import apply_average_blur
+from utils.io_handler import IOHandler
+from pre_processing.blur import apply_average_blur
 
 
 def apply_laplacian_sharpening(

imagepro_python-0.1.1/src/ImagePRO/utils/io_handler.py

@@ -0,0 +1,178 @@
+# io_handler.py
+
+import os
+import json
+import csv
+from pathlib import Path
+
+import cv2
+import numpy as np
+
+
+class IOHandler:
+    @staticmethod
+    def load_image(image_path=None, np_image=None):
+        """
+        Load an image from disk or use an existing NumPy array.
+
+        Args:
+            image_path (str, optional): Path to the image file.
+            np_image (np.ndarray, optional): Image as a NumPy array.
+
+        Returns:
+            np.ndarray: Loaded image.
+
+        Raises:
+            TypeError: Invalid input types.
+            FileNotFoundError: File not found.
+            ValueError: Both inputs are None or image loading failed.
+        """
+        if image_path is not None:
+            if not isinstance(image_path, str):
+                raise TypeError("'image_path' must be a string.")
+            if not os.path.isfile(image_path):
+                raise FileNotFoundError(f"File not found: {image_path}")
+            img = cv2.imread(image_path)
+            if img is None:
+                raise ValueError(f"Unable to load image: {image_path}")
+            return img
+
+        if np_image is not None:
+            if not isinstance(np_image, np.ndarray):
+                raise TypeError("'np_image' must be a NumPy array.")
+            return np_image
+
+        raise ValueError("Provide at least 'image_path' or 'np_image'.")
+
+    @staticmethod
+    def save_image(np_image, result_path=None):
+        """
+        Save a NumPy image array (or list of arrays) to disk or return it.
+
+        Args:
+            np_image (np.ndarray or list[np.ndarray]): Image(s) to save.
+            result_path (str, optional): Save path.
+
+        Returns:
+            str or np.ndarray or list[np.ndarray]: 
+                Save message if path given, otherwise returns input.
+
+        Raises:
+            TypeError: Invalid input types.
+            IOError: Saving failed.
+        """
+        if not isinstance(np_image, (np.ndarray, list)):
+            raise TypeError("'np_image' must be a NumPy array or list of arrays.")
+
+        if isinstance(np_image, list) and not all(isinstance(i, np.ndarray) for i in np_image):
+            raise TypeError("All items in the list must be NumPy arrays.")
+
+        if result_path is not None and not isinstance(result_path, str):
+            raise TypeError("'result_path' must be a string or None.")
+
+        if result_path:
+            if isinstance(np_image, np.ndarray):
+                if not cv2.imwrite(result_path, np_image):
+                    raise IOError(f"Failed to save image: {result_path}")
+                return f"Image saved at {result_path}"
+
+            for i, img in enumerate(np_image):
+                suffix = f"_{i}" if i > 0 else ""
+                path = result_path.replace(".jpg", f"{suffix}.jpg")
+                if not cv2.imwrite(path, img):
+                    raise IOError(f"Failed to save image: {path}")
+            return f"Images saved using base path {result_path}"
+
+        return np_image
+
+    @staticmethod
+    def save_csv(data, result_path=None):
+        """
+        Save a list of lists as a CSV file or return it.
+
+        Args:
+            data (list[list]): Tabular data.
+            result_path (str, optional): Save path.
+
+        Returns:
+            str or list[list]: Save message or original data.
+
+        Raises:
+            TypeError: Invalid input types.
+        """
+        if not isinstance(data, list) or not all(isinstance(row, list) for row in data):
+            raise TypeError("'data' must be a list of lists.")
+
+        if result_path:
+            if not isinstance(result_path, str):
+                raise TypeError("'result_path' must be a string or None.")
+            with open(result_path, "w", newline='') as f:
+                writer = csv.writer(f)
+                writer.writerows(data)
+            return f"CSV saved at {result_path}"
+
+        return data
+
+    @staticmethod
+    def save_json(data, result_path=None):
+        """
+        Save data as a JSON file or return it.
+
+        Args:
+            data (Any): JSON-serializable data.
+            result_path (str, optional): Save path.
+
+        Returns:
+            str or Any: Save message or original data.
+
+        Raises:
+            TypeError: Invalid path type.
+        """
+        if result_path:
+            if not isinstance(result_path, str):
+                raise TypeError("'result_path' must be a string or None.")
+            with open(result_path, "w") as f:
+                json.dump(data, f, indent=4)
+            return f"JSON saved at {result_path}"
+
+        return data
+
+    @staticmethod
+    def save(data, result_path=None, file_type=None):
+        """
+        Save data to disk automatically based on file type or extension.
+
+        Args:
+            data (Any): Data to save.
+            result_path (str, optional): Save path.
+            file_type (str, optional): 'image', 'csv', or 'json'.
+
+        Returns:
+            str or Any: Save message or original data.
+
+        Raises:
+            ValueError: Unsupported file type or extension.
+            TypeError: Invalid input types.
+        """
+        if file_type is None:
+            if result_path is None:
+                return data
+
+            ext = Path(result_path).suffix.lower()
+            if ext in [".jpg", ".jpeg", ".png"]:
+                file_type = "image"
+            elif ext == ".csv":
+                file_type = "csv"
+            elif ext == ".json":
+                file_type = "json"
+            else:
+                raise ValueError(f"Unsupported file extension: {ext}")
+
+        if file_type == "image":
+            return IOHandler.save_image(data, result_path)
+        elif file_type == "csv":
+            return IOHandler.save_csv(data, result_path)
+        elif file_type == "json":
+            return IOHandler.save_json(data, result_path)
+
+        raise ValueError(f"Unsupported file type: {file_type}")

{imagepro_python-0.1.0 → imagepro_python-0.1.1/src/ImagePRO_Python.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ImagePRO-Python
-Version: 0.1.0
+Version: 0.1.1
 Summary: clean, modular, and easy-to-use Python library for image processing tasks.
 Author-email: Parsa Safaie <parsasafaie.2568@proton.me>
 License: MIT

{imagepro_python-0.1.0 → imagepro_python-0.1.1}/src/ImagePRO_Python.egg-info/SOURCES.txt

@@ -18,6 +18,8 @@ src/ImagePRO/pre_processing/grayscale.py
 src/ImagePRO/pre_processing/resize.py
 src/ImagePRO/pre_processing/rotate.py
 src/ImagePRO/pre_processing/sharpen.py
+src/ImagePRO/utils/__init__.py
+src/ImagePRO/utils/io_handler.py
 src/ImagePRO_Python.egg-info/PKG-INFO
 src/ImagePRO_Python.egg-info/SOURCES.txt
 src/ImagePRO_Python.egg-info/dependency_links.txt