Pygrabber only for Windows

modules/ui.py

@@ -27,6 +27,7 @@ from modules.utilities import (
     has_image_extension,
 )
 from modules.video_capture import VideoCapturer
+import platform
 
 ROOT = None
 POPUP = None
@@ -771,39 +772,57 @@ def webcam_preview(root: ctk.CTk, camera_index: int):
 
 def get_available_cameras():
     """Returns a list of available camera names and indices."""
-    try:
-        graph = FilterGraph()
-        devices = graph.get_input_devices()
+    if platform.system() == "Windows":
+        try:
+            graph = FilterGraph()
+            devices = graph.get_input_devices()
 
-        # Create list of indices and names
-        camera_indices = list(range(len(devices)))
-        camera_names = devices
+            # Create list of indices and names
+            camera_indices = list(range(len(devices)))
+            camera_names = devices
 
-        # If no cameras found through DirectShow, try OpenCV fallback
-        if not camera_names:
-            # Try to open camera with index -1 and 0
-            test_indices = [-1, 0]
-            working_cameras = []
+            # If no cameras found through DirectShow, try OpenCV fallback
+            if not camera_names:
+                # Try to open camera with index -1 and 0
+                test_indices = [-1, 0]
+                working_cameras = []
 
-            for idx in test_indices:
-                cap = cv2.VideoCapture(idx)
-                if cap.isOpened():
-                    working_cameras.append(f"Camera {idx}")
-                    cap.release()
+                for idx in test_indices:
+                    cap = cv2.VideoCapture(idx)
+                    if cap.isOpened():
+                        working_cameras.append(f"Camera {idx}")
+                        cap.release()
 
-            if working_cameras:
-                return test_indices[: len(working_cameras)], working_cameras
+                if working_cameras:
+                    return test_indices[: len(working_cameras)], working_cameras
+
+            # If still no cameras found, return empty lists
+            if not camera_names:
+                return [], ["No cameras found"]
+
+            return camera_indices, camera_names
+
+        except Exception as e:
+            print(f"Error detecting cameras: {str(e)}")
+            return [], ["No cameras found"]
+    else:
+        # Unix-like systems (Linux/Mac) camera detection
+        camera_indices = []
+        camera_names = []
+
+        # Test the first 10 indices
+        for i in range(10):
+            cap = cv2.VideoCapture(i)
+            if cap.isOpened():
+                camera_indices.append(i)
+                camera_names.append(f"Camera {i}")
+                cap.release()
 
-        # If still no cameras found, return empty lists
         if not camera_names:
             return [], ["No cameras found"]
 
         return camera_indices, camera_names
 
-    except Exception as e:
-        print(f"Error detecting cameras: {str(e)}")
-        return [], ["No cameras found"]
-
 
 def create_webcam_preview(camera_index: int):
     global preview_label, PREVIEW