diff --git a/modules/ui.py b/modules/ui.py index 494d85a..e83c3d0 100644 --- a/modules/ui.py +++ b/modules/ui.py @@ -6,15 +6,16 @@ from typing import Callable, Tuple, List, Any from types import ModuleType import cv2 from PIL import Image, ImageOps -from pygrabber.dshow_graph import FilterGraph import pyvirtualcam # Import OS-specific modules only when necessary if platform.system() == 'Darwin': # macOS - import objc - from Foundation import NSObject import AVFoundation +if os.name == 'nt': # Windows + from pygrabber.dshow_graph import FilterGraph + + import modules.globals import modules.metadata from modules.face_analyser import get_one_face diff --git a/requirements.txt b/requirements.txt index 915179d..96bb6b4 100644 --- a/requirements.txt +++ b/requirements.txt @@ -23,4 +23,5 @@ tqdm==4.66.4 gfpgan==1.3.8 pyobjc==9.1; sys_platform == 'darwin' pygrabber==0.2 -pyvirtualcam==0.12.0 \ No newline at end of file +pyvirtualcam==0.12.0 +pyobjc-framework-AVFoundation==10.3.1; sys_platform == 'darwin' \ No newline at end of file