Update audio_capture.py

audio_capture.py

@@ -1,5 +1,5 @@
-# audio_capture.py (Optimized Audio Capture)
-import pyaudio
+# audio_capture.py (Updated with SoundDevice)
+import sounddevice as sd
 import numpy as np
 import wave
 import os
@@ -8,41 +8,42 @@ import config
 
 class AudioRecorder:
     def __init__(self):
-        self.audio = pyaudio.PyAudio()
-        self.stream = None
         self.is_recording = False
         self.sample_rate = config.SAMPLE_RATE
         self.filename = f"meeting_{datetime.now().strftime('%Y%m%d_%H%M%S')}.wav"
         self.audio_buffer = []
+        self.stream = None
         
     def start_recording(self):
         self.is_recording = True
         self.audio_buffer = []
-        self.stream = self.audio.open(
-            format=config.AUDIO_FORMAT,
+        
+        def callback(indata, frames, time, status):
+            if status:
+                print(f"Audio error: {status}")
+            if self.is_recording:
+                self.audio_buffer.append(indata.copy())
+        
+        self.stream = sd.InputStream(
+            samplerate=self.sample_rate,
             channels=config.CHANNELS,
-            rate=self.sample_rate,
-            input=True,
-            frames_per_buffer=config.CHUNK_SIZE,
-            input_device_index=config.INPUT_DEVICE_INDEX
+            callback=callback,
+            dtype='int16'
         )
+        self.stream.start()
         return True
         
     def get_audio_chunk(self):
-        if self.is_recording:
-            try:
-                data = self.stream.read(config.CHUNK_SIZE, exception_on_overflow=False)
-                # Return as numpy array for efficient processing
-                return np.frombuffer(data, dtype=np.int16)
-            except OSError as e:
-                print(f"Audio error: {str(e)}")
-                return None
+        # In this implementation, we don't need to return chunks because
+        # we are using a callback. But to keep the interface, we return the last chunk.
+        if self.is_recording and self.audio_buffer:
+            return self.audio_buffer[-1]
         return None
         
     def stop_recording(self):
         if self.is_recording:
             self.is_recording = False
-            self.stream.stop_stream()
+            self.stream.stop()
             self.stream.close()
             self._save_recording()
             return os.path.join(config.RECORDINGS_DIR, self.filename)
@@ -50,8 +51,9 @@ class AudioRecorder:
         
     def _save_recording(self):
         os.makedirs(config.RECORDINGS_DIR, exist_ok=True)
+        audio_data = np.concatenate(self.audio_buffer, axis=0)
         with wave.open(os.path.join(config.RECORDINGS_DIR, self.filename), 'wb') as wf:
             wf.setnchannels(config.CHANNELS)
-            wf.setsampwidth(self.audio.get_sample_size(config.AUDIO_FORMAT))
+            wf.setsampwidth(2)  # 2 bytes for int16
             wf.setframerate(self.sample_rate)
-            wf.writeframes(b''.join(self.audio_buffer))
+            wf.writeframes(audio_data.tobytes())