record.py

import sounddevice as sd
import numpy as np
import scipy.io.wavfile as wav

# Configuration
DURATION = 5  # Recording duration in seconds
SAMPLERATE = 16000  # Whisper expects 16kHz sample rate
OUTPUT_FILENAME = "recorded_audio.wav"  # Output WAV file


def record_audio(duration=DURATION, samplerate=SAMPLERATE, filename=OUTPUT_FILENAME):
    """Records audio from the microphone and saves it as a WAV file."""

    print(f"🎤 Recording for {duration} seconds... Speak now!")

    # Record audio
    audio_data = sd.rec(
        int(duration * samplerate), samplerate=samplerate, channels=1, dtype=np.int16
    )
    sd.wait()  # Wait for recording to complete

    # Save as WAV file
    wav.write(filename, samplerate, audio_data)

    print(f"✅ Recording complete! Audio saved as '{filename}'")


if __name__ == "__main__":
    record_audio()