def wavmaker()

in TSA-demo/TSA_voice_assistant/waver.py [0:0]


def wavmaker(stream):
    frames = []
    recording = array('h')

    for i in range(0, int(RATE / CHUNK * RECORD_SECONDS)):
        sch = stream.read(CHUNK)
        data = sch
        frames.append(data)
        datab = array('h', sch)
        if byteorder == 'big':
            datab.byteswap()
        recording.extend(datab)

    print("* done recording")

    stream.stop_stream()
    stream.close()
    p.terminate()

    wf = wave.open(WAVE_OUTPUT_FILENAME, 'wb')
    wf.setnchannels(CHANNELS)
    wf.setsampwidth(p.get_sample_size(FORMAT))
    wf.setframerate(RATE)
    wf.writeframes(b''.join(frames))
    wf.close()

    os.system('ffmpeg -i output.wav -ar 16000 -ac 1 trackb.wav -y')

    #print(recording)
    return recording