use conditional imports and show help errors if modules not found

src/dailyai/transports/daily_transport.py

@@ -15,14 +15,21 @@ from dailyai.pipeline.frames import (
 
 from threading import Event
 
-from daily import (
-    EventHandler,
-    CallClient,
-    Daily,
-    VirtualCameraDevice,
-    VirtualMicrophoneDevice,
-    VirtualSpeakerDevice,
-)
+try:
+    from daily import (
+        EventHandler,
+        CallClient,
+        Daily,
+        VirtualCameraDevice,
+        VirtualMicrophoneDevice,
+        VirtualSpeakerDevice,
+    )
+except ModuleNotFoundError as e:
+    print(f"Exception: {e}")
+    print(
+        "In order to use the Daily transport, you need to `pip install dailyai[daily]`.")
+    raise Exception(f"Missing module: {e}")
+
 
 from dailyai.transports.threaded_transport import ThreadedTransport
 

src/dailyai/transports/local_transport.py

@@ -4,17 +4,18 @@ import tkinter as tk
 
 from dailyai.transports.threaded_transport import ThreadedTransport
 
+try:
+    import pyaudio
+except ModuleNotFoundError as e:
+    print(f"Exception: {e}")
+    print(
+        "In order to use the local transport, you need to `pip install dailyai[local]`. On MacOS, you also need to `brew install portaudio`.")
+    raise Exception(f"Missing module: {e}")
+
 
 class LocalTransport(ThreadedTransport):
     def __init__(self, **kwargs):
         super().__init__(**kwargs)
-        try:
-            global pyaudio
-            import pyaudio
-        except ModuleNotFoundError as e:
-            print(f"Exception: {e}")
-            print("In order to use the local transport, you'll need to `pip install pyaudio`. On MacOS, you'll also need to `brew install portaudio`.")
-            raise Exception(f"Missing module: {e}")
         self._sample_width = kwargs.get("sample_width") or 2
         self._n_channels = kwargs.get("n_channels") or 1
         self._tk_root = kwargs.get("tk_root") or None

src/dailyai/transports/threaded_transport.py

@@ -1,7 +1,6 @@
 from abc import abstractmethod
 import asyncio
 import itertools
-import logging
 import numpy as np
 
 import queue

src/dailyai/transports/websocket_transport.py

@@ -1,7 +1,6 @@
 import asyncio
 import time
 from typing import AsyncGenerator, List
-import websockets
 
 from dailyai.pipeline.frame_processor import FrameProcessor
 from dailyai.pipeline.frames import AudioFrame, ControlFrame, EndFrame, Frame, TTSEndFrame, TTSStartFrame, TextFrame
@@ -10,6 +9,14 @@ from dailyai.serializers.protobuf_serializer import ProtobufFrameSerializer
 from dailyai.transports.abstract_transport import AbstractTransport
 from dailyai.transports.threaded_transport import ThreadedTransport
 
+try:
+    import websockets
+except ModuleNotFoundError as e:
+    print(f"Exception: {e}")
+    print(
+        "In order to use the websocket transport, you need to `pip install dailyai[websocket]`.")
+    raise Exception(f"Missing module: {e}")
+
 
 class WebSocketFrameProcessor(FrameProcessor):
     """This FrameProcessor filters and mutates frames before they're sent over the websocket.