From a01e3b2fb83ffd3a16e9f7d451f87869dadf762d Mon Sep 17 00:00:00 2001
From: Ajay Raj <ajay.n.raj@gmail.com>
Date: Mon, 20 Mar 2023 20:22:36 -0700
Subject: [PATCH] few last imports

---
 .../streaming/microphone_input.py             | 51 +++++++++++++++++++
 .../input_device/streaming/telephone_input.py |  9 ++++
 2 files changed, 60 insertions(+)
 create mode 100644 vocode/streaming/input_device/streaming/microphone_input.py
 create mode 100644 vocode/streaming/input_device/streaming/telephone_input.py

diff --git a/vocode/streaming/input_device/streaming/microphone_input.py b/vocode/streaming/input_device/streaming/microphone_input.py
new file mode 100644
index 0000000..f61b261
--- /dev/null
+++ b/vocode/streaming/input_device/streaming/microphone_input.py
@@ -0,0 +1,51 @@
+import sounddevice as sd
+import numpy as np
+from typing import Optional
+import queue
+import wave
+
+from vocode.streaming.input_device.base_input_device import BaseInputDevice
+from vocode.streaming.models.audio_encoding import AudioEncoding
+
+
+class MicrophoneInput(BaseInputDevice):
+    DEFAULT_SAMPLING_RATE = 44100
+    DEFAULT_CHUNK_SIZE = 2048
+
+    def __init__(
+        self,
+        device_info: dict,
+        sampling_rate: int = None,
+        chunk_size: int = DEFAULT_CHUNK_SIZE,
+        microphone_gain: int = 1,
+    ):
+        self.device_info = device_info
+        sampling_rate = sampling_rate or (
+            self.device_info.get("default_samplerate", self.DEFAULT_SAMPLING_RATE)
+        )
+        super().__init__(int(sampling_rate), AudioEncoding.LINEAR16, chunk_size)
+        self.stream = sd.InputStream(
+            dtype=np.int16,
+            channels=1,
+            samplerate=self.sampling_rate,
+            blocksize=self.chunk_size,
+            device=int(self.device_info["index"]),
+            callback=self._stream_callback,
+        )
+        self.stream.start()
+        self.queue = queue.Queue()
+        self.microphone_gain = microphone_gain
+
+    def _stream_callback(self, in_data: np.ndarray[np.int16], *_args):
+        if self.microphone_gain > 1:
+            in_data = in_data * (2 ^ self.microphone_gain)
+        else:
+            in_data = in_data // (2 ^ self.microphone_gain)
+        audio_bytes = in_data.tobytes()
+        self.queue.put_nowait(audio_bytes)
+
+    def get_audio(self) -> Optional[bytes]:
+        try:
+            return self.queue.get_nowait()
+        except queue.Empty:
+            return None
diff --git a/vocode/streaming/input_device/streaming/telephone_input.py b/vocode/streaming/input_device/streaming/telephone_input.py
new file mode 100644
index 0000000..57a22e8
--- /dev/null
+++ b/vocode/streaming/input_device/streaming/telephone_input.py
@@ -0,0 +1,9 @@
+from vocode.streaming.input_device.base_input_device import BaseInputDevice
+from vocode.streaming.models.audio_encoding import AudioEncoding
+
+
+class TelephoneInput(BaseInputDevice):
+    def __init__(self):
+        super().__init__(
+            sampling_rate=8000, audio_encoding=AudioEncoding.MULAW, chunk_size=160
+        )