Skip to content

record_mic

Module for recording a microphone to a file

Recording settings are set internally to match whisper format.

main()

Example of making a simple audio clip

Source code in backend/app/utils/stt/backends/record_mic.py 
56
57
58
59
60
61
62
63
64
65
66
67
68
69
def main():
    """Example of making a simple audio clip"""
    parser = argparse.ArgumentParser(
        formatter_class=argparse.ArgumentDefaultsHelpFormatter)

    parser.add_argument("--record_for", default=20,
                        help="Set recording length (seconds)", type=bool)
    parser.add_argument("--filename", default="output/test.wav",
                        help="location to save audio file")

    args = parser.parse_args()

    record_mic(args.filename, seconds=args.record_for)
    print(f"File saved to {args.filename}")

record_mic(filename, seconds=200)

Simple helper function for recording audio to a file.

Parameters:

Name Type Description Default
filename str

Where to save audio to.

 required
seconds int

How long to record. Defaults to 200.

 200
Source code in backend/app/utils/stt/backends/record_mic.py 
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
def record_mic(filename: str, seconds: int = 200) -> None:
    """Simple helper function for recording audio to a file.

    Args:
        filename (str): Where to save audio to.
        seconds (int, optional): How long to record. Defaults to 200.
    """
    chunk = 1024  # Record in chunks of 1024 samples
    sample_format = pyaudio.paInt16  # 16 bits per sample
    channels = 2
    framerate = 16000  # Record at 44100 samples per second
    audio_interface = pyaudio.PyAudio()  # Create an interface to PortAudio

    print('Recording')

    stream = audio_interface.open(format=sample_format,
                    channels=channels,
                    rate=framerate,
                    frames_per_buffer=chunk,
                    input=True)

    frames = []  # Initialize array to store frames

    # Store data in chunks
    for _ in range(0, int(framerate / chunk * seconds)):
        data = stream.read(chunk)
        frames.append(data)

    # Stop and close the stream
    stream.stop_stream()
    stream.close()
    # Terminate the PortAudio interface
    audio_interface.terminate()

    print('Finished recording')

    # Save the recorded data as a WAV file
    wavefile = wave.open(filename, 'wb')
    wavefile.setnchannels(channels)
    wavefile.setsampwidth(audio_interface.get_sample_size(sample_format))
    wavefile.setframerate(framerate)
    wavefile.writeframes(b''.join(frames))
    wavefile.close()