import pyaudio
import wave

# Nastavení parametrů nahrávání
FORMAT = pyaudio.paInt16  # Formát zvuku
CHANNELS = 1  # Počet kanálů (mono)
RATE = 44100  # Vzorkovací frekvence (Hz)
CHUNK = 1024  # Velikost bloku dat
RECORD_SECONDS = 5  # Délka nahrávky (v sekundách)
WAVE_OUTPUT_FILENAME = "sample1.wav"  # Název výstupního zvukového souboru

audio = pyaudio.PyAudio()

# Nahrávání zvuku z mikrofonu
stream = audio.open(format=FORMAT, channels=CHANNELS,
                    rate=RATE, input=True,
                    frames_per_buffer=CHUNK)
print("Nahrávám...")

frames = []

for i in range(0, int(RATE / CHUNK * RECORD_SECONDS)):
    data = stream.read(CHUNK)
    frames.append(data)

print("Nahrávání dokončeno")

# Uložení nahrávky do WAV souboru
with wave.open(WAVE_OUTPUT_FILENAME, 'wb') as wf:
    wf.setnchannels(CHANNELS)
    wf.setsampwidth(audio.get_sample_size(FORMAT))
    wf.setframerate(RATE)
    wf.writeframes(b''.join(frames))

stream.stop_stream()
stream.close()
audio.terminate()