fix for headless environments

src/utils/interaction.py

@@ -1,20 +1,27 @@
 from dataclasses import dataclass
-
 import cv2
-from screeninfo import get_monitors
-
 from src.logger import logger
 from src.utils.image import ImageUtils
 
-monitor_window = get_monitors()[0]
+# ✅ Safe import for headless environments
+try:
+    from screeninfo import get_monitors
+    monitor_window = get_monitors()[0]
+    SCREEN_WIDTH, SCREEN_HEIGHT = monitor_window.width, monitor_window.height
+except Exception as e:
+    # Headless fallback — e.g., Hugging Face Spaces, servers, CI
+    print(f"[Headless Mode] No monitor detected. Using default size. Error: {e}")
+    SCREEN_WIDTH, SCREEN_HEIGHT = 1920, 1080
 
 
 @dataclass
 class ImageMetrics:
-    # TODO: Move TEXT_SIZE, etc here and find a better class name
-    window_width, window_height = monitor_window.width, monitor_window.height
-    # for positioning image windows
-    window_x, window_y = 0, 0
+    # Window dimensions
+    window_width: int = SCREEN_WIDTH
+    window_height: int = SCREEN_HEIGHT
+    # For positioning image windows
+    window_x: int = 0
+    window_y: int = 0
     reset_pos = [0, 0]
 
 
@@ -26,11 +33,14 @@ class InteractionUtils:
     @staticmethod
     def show(name, origin, pause=1, resize=False, reset_pos=None, config=None):
         image_metrics = InteractionUtils.image_metrics
+
         if origin is None:
             logger.info(f"'{name}' - NoneType image to show!")
             if pause:
                 cv2.destroyAllWindows()
             return
+
+        # Resize if required
         if resize:
             if not config:
                 raise Exception("config not provided for resizing the image to show")
@@ -38,28 +48,21 @@ class InteractionUtils:
         else:
             img = origin
 
-        if not is_window_available(name):
-            cv2.namedWindow(name)
-
-        cv2.imshow(name, img)
-
-        if reset_pos:
-            image_metrics.window_x = reset_pos[0]
-            image_metrics.window_y = reset_pos[1]
-
-        cv2.moveWindow(
-            name,
-            image_metrics.window_x,
-            image_metrics.window_y,
-        )
-
+        # Try to show the image only if display is available
+        try:
+            if not is_window_available(name):
+                cv2.namedWindow(name)
+            cv2.imshow(name, img)
+            cv2.moveWindow(name, image_metrics.window_x, image_metrics.window_y)
+        except cv2.error:
+            # Skip visualization entirely in headless environments
+            logger.info(f"Skipping cv2.imshow for '{name}' (headless mode).")
+
+        # Update next window position
         h, w = img.shape[:2]
-
-        # Set next window position
         margin = 25
         w += margin
         h += margin
-
         w, h = w // 2, h // 2
         if image_metrics.window_x + w > image_metrics.window_width:
             image_metrics.window_x = 0
@@ -71,10 +74,7 @@ class InteractionUtils:
             image_metrics.window_x += w
 
         if pause:
-            logger.info(
-                f"Showing '{name}'\n\t Press Q on image to continue. Press Ctrl + C in terminal to exit"
-            )
-
+            logger.info(f"Showing '{name}' (press Q to continue, Ctrl+C to exit)")
             wait_q()
             InteractionUtils.image_metrics.window_x = 0
             InteractionUtils.image_metrics.window_y = 0
@@ -82,19 +82,19 @@ class InteractionUtils:
 
 @dataclass
 class Stats:
-    # TODO Fill these for stats
-    # Move qbox_vals here?
-    # badThresholds = []
-    # veryBadPoints = []
     files_moved = 0
     files_not_moved = 0
 
 
 def wait_q():
     esc_key = 27
-    while cv2.waitKey(1) & 0xFF not in [ord("q"), esc_key]:
+    try:
+        while cv2.waitKey(1) & 0xFF not in [ord("q"), esc_key]:
+            pass
+        cv2.destroyAllWindows()
+    except cv2.error:
+        # Skip in headless mode
         pass
-    cv2.destroyAllWindows()
 
 
 def is_window_available(name: str) -> bool:
@@ -102,6 +102,5 @@ def is_window_available(name: str) -> bool:
     try:
         cv2.getWindowProperty(name, cv2.WND_PROP_VISIBLE)
         return True
-    except Exception as e:
-        print(e)
+    except Exception:
         return False