Update app.py

app.py

@@ -1,52 +1,47 @@
 import streamlit as st
-import sounddevice as sd
-import numpy as np
+import pyaudio
 import wave
-import io
-from scipy.io.wavfile import write
-
-# Функция для записи аудио
-def record_audio(duration, samplerate=44100):
-    st.write("🎙️ Идёт запись...")
-    audio_data = sd.rec(int(duration * samplerate), samplerate=samplerate, channels=2, dtype='int16')
-    sd.wait()  # Ждём завершения записи
+
+# Параметры записи
+FORMAT = pyaudio.paInt16
+CHANNELS = 2
+RATE = 44100
+CHUNK = 1024
+
+# Функция записи
+def record_audio(duration, output_file="output.wav"):
+    audio = pyaudio.PyAudio()
+    stream = audio.open(format=FORMAT, channels=CHANNELS,
+                        rate=RATE, input=True,
+                        frames_per_buffer=CHUNK)
+    st.write("🎙️ Запись...")
+    frames = []
+
+    for _ in range(0, int(RATE / CHUNK * duration)):
+        data = stream.read(CHUNK)
+        frames.append(data)
+
     st.write("✅ Запись завершена!")
-    return audio_data
 
-# Функция для сохранения аудио в WAV файл
-def save_audio_to_wav(audio_data, samplerate=44100):
-    # Создаём поток в памяти
-    wav_io = io.BytesIO()
-    write(wav_io, samplerate, audio_data)  # Сохраняем в поток
-    wav_io.seek(0)  # Возвращаем курсор в начало
-    return wav_io
+    # Остановка записи
+    stream.stop_stream()
+    stream.close()
+    audio.terminate()
 
-# Интерфейс Streamlit
-st.title("🎙️ Простой диктофон")
-st.write("Этот диктофон позволяет записывать аудио и сохранять его на вашем устройстве.")
+    # Сохранение файла
+    with wave.open(output_file, 'wb') as wf:
+        wf.setnchannels(CHANNELS)
+        wf.setsampwidth(audio.get_sample_size(FORMAT))
+        wf.setframerate(RATE)
+        wf.writeframes(b''.join(frames))
 
-# Ползунок для выбора длины записи
-duration = st.slider("Длительность записи (секунды)", min_value=1, max_value=60, value=5)
+    return output_file
 
-# Переменная для хранения записи
-if "audio_data" not in st.session_state:
-    st.session_state.audio_data = None
+# Streamlit интерфейс
+st.title("🎙️ Простой диктофон на PyAudio")
+duration = st.slider("Длительность записи (секунды)", 1, 60, 5)
 
-# Кнопка для начала записи
 if st.button("Начать запись"):
-    audio_data = record_audio(duration)
-    st.session_state.audio_data = audio_data  # Сохраняем запись в сессии
-
-# Проверяем, есть ли записанные данные
-if st.session_state.audio_data is not None:
-    # Преобразуем запись в WAV и отображаем аудиоплеер
-    audio_wav = save_audio_to_wav(st.session_state.audio_data)
-    st.audio(audio_wav, format="audio/wav")
-
-    # Кнопка для сохранения файла
-    st.download_button(
-        label="Сохранить запись",
-        data=audio_wav,
-        file_name="recorded_audio.wav",
-        mime="audio/wav"
-    )
+    output_file = record_audio(duration)
+    st.audio(output_file, format="audio/wav")
+    st.success(f"Файл сохранён: {output_file}")