feat: set back add_image_to_folder to yolo

src/data_models/image_manager.py

@@ -98,32 +98,3 @@ class ImageManager:
     def close_connection(self):
         """Close the connection."""
         self.session.close()
-
-
-def add_image_to_db(result, image_manager, park_id):
-    """
-    Adds an image to the database.
-
-    Args:
-        result: YOLO result object containing the image details.
-        image_manager (ImageManager): Instance of ImageManager to interact with the database.
-        park_id (int): The ID of the associated park.
-
-    Returns:
-        int: The ID of the added image.
-    """
-    image_name = result.path.split("/")[-1]
-    current_datetime = datetime.now()
-    created_at = current_datetime.strftime("%Y-%m-%d %H:%M:%S")
-
-    try:
-        print(f"Adding image '{image_name}' to the database...")
-        image = image_manager.add_image(
-            name=image_name, created_at=created_at, park_id=park_id
-        )
-        image_id = image_manager.get_image_id(image_name)
-        print(f"Image '{image_name}' added with ID: {image_id}")
-        return image_id
-    except Exception as e:
-        print(f"Error adding image '{image_name}' to the database: {e}")
-        return None

src/yolo_predictions.py

@@ -1,9 +1,44 @@
+from datetime import datetime
+
 from data_models.bbox_manager import BoundingBoxManager, add_bboxes_to_db
-from data_models.image_manager import ImageManager, add_image_to_db
-from data_models.park_manager import ParkManager
+from data_models.image_manager import ImageManager
 from ultralytics import YOLO
 
 
+model = None
+bbox_manager = BoundingBoxManager()
+image_manager = ImageManager()
+
+
+def add_image_to_db(result, image_manager, park_id):
+    """
+    Adds an image to the database.
+
+    Args:
+        result: YOLO result object containing the image details.
+        image_manager (ImageManager): Instance of ImageManager to interact with the database.
+        park_id (int): The ID of the associated park.
+
+    Returns:
+        int: The ID of the added image.
+    """
+    image_name = result.path.split("/")[-1]
+    current_datetime = datetime.now()
+    created_at = current_datetime.strftime("%Y-%m-%d %H:%M:%S")
+
+    try:
+        print(f"Adding image '{image_name}' to the database...")
+        image = image_manager.add_image(
+            name=image_name, created_at=created_at, park_id=park_id
+        )
+        image_id = image_manager.get_image_id(image_name)
+        print(f"Image '{image_name}' added with ID: {image_id}")
+        return image_id
+    except Exception as e:
+        print(f"Error adding image '{image_name}' to the database: {e}")
+        return None
+
+
 def load_model(model_file):
     """
     Load the YOLO model.