Fix import order in line_detection_ai.py

line_detection_ai.py

@@ -1,61 +1,50 @@
-from base import BaseDetector, BaseDetectionPipeline
-from utils import *
-from config import (
-    ImageConfig, 
-    SymbolConfig, 
-    TagConfig, 
-    LineConfig, 
-    PointConfig, 
-    JunctionConfig
-)
-from detectors import (
-    LineDetector, 
-    PointDetector, 
-    JunctionDetector, 
-    SymbolDetector, 
-    TagDetector
-)
-from pathlib import Path
-from storage import StorageFactory
-from common import DetectionResult
-from detection_schema import DetectionContext, JunctionType
-from typing import List, Tuple, Optional, Dict
-import torch
-import numpy as np
-import cv2
+# Standard library imports first
 import os
 import logging
+from typing import List, Dict, Optional, Any
 
+# Configure logging before any other imports
+logging.basicConfig(
+    level=logging.INFO,
+    format='%(asctime)s - %(levelname)s - %(message)s'
+)
 logger = logging.getLogger(__name__)
 
-class DiagramDetectionPipeline:
-    """
-    Pipeline that runs multiple detectors (line, point, junction, etc.) on an image,
-    and keeps a shared DetectionContext in memory.
-    """
-
-    def __init__(self,
-                 tag_detector: Optional[BaseDetector],
-                 symbol_detector: Optional[BaseDetector],
-                 line_detector: Optional[BaseDetector],
-                 point_detector: Optional[BaseDetector],
-                 junction_detector: Optional[BaseDetector],
-                 storage: StorageInterface,
-                 debug_handler: Optional[DebugHandler] = None,
-                 transformer: Optional[CoordinateTransformer] = None):
-        """
-        You can pass None for detectors you don't need.
-        """
-        # super().__init__(storage=storage, debug_handler=debug_handler)
-        self.storage = storage
-        self.debug_handler = debug_handler
-        self.tag_detector = tag_detector
-        self.symbol_detector = symbol_detector
-        self.line_detector = line_detector
-        self.point_detector = point_detector
-        self.junction_detector = junction_detector
-        self.transformer = transformer or CoordinateTransformer()
+# Third-party imports
+import cv2
+import numpy as np
 
+# Local imports
+from base import BaseDetectionPipeline
+from config import (
+    LineConfig, ImageConfig, PointConfig, JunctionConfig,
+    SymbolConfig, TagConfig
+)
+from utils import DebugHandler
+from storage import StorageInterface
+
+# Import detectors after logging is configured
+from detectors import (
+    LineDetector, PointDetector, JunctionDetector,
+    SymbolDetector, TagDetector
+)
+
+class DiagramDetectionPipeline(BaseDetectionPipeline):
+    """Main pipeline for processing P&ID diagrams"""
+    
+    def __init__(
+            self,
+            storage: StorageInterface,
+            debug_handler: Optional[DebugHandler] = None
+    ):
+        super().__init__(storage, debug_handler)
+        # Initialize detectors when needed
+        self._line_detector = None
+        self._point_detector = None
+        self._junction_detector = None
+        self._symbol_detector = None
+        self._tag_detector = None
+        
     def _load_image(self, image_path: str) -> np.ndarray:
         """Load image with validation."""
         image_data = self.storage.load_file(image_path)