Update app.py

app.py

@@ -1,66 +1,34 @@
 import streamlit as st
-import pyaudio
+import sounddevice as sd
+import numpy as np
 import wave
-import os
-import tempfile
+from scipy.io.wavfile import write
 
-# Настройки записи
-FORMAT = pyaudio.paInt16
-CHANNELS = 1
-RATE = 44100
-CHUNK = 1024
+# Функция для записи аудио
+def record_audio(duration, samplerate=44100):
+    st.write("🎙️ Запись...")
+    audio_data = sd.rec(int(duration * samplerate), samplerate=samplerate, channels=2, dtype='int16')
+    sd.wait()  # Ожидание завершения записи
+    st.write("✅ Запись завершена!")
+    return audio_data
 
-# Инициализация PyAudio
-audio = pyaudio.PyAudio()
+# Функция для сохранения аудио в WAV файл
+def save_audio(filename, data, samplerate=44100):
+    write(filename, samplerate, data)
+    st.success(f"Аудио сохранено как {filename}")
 
-# Создание интерфейса Streamlit
-st.title("Простой диктофон")
+# Интерфейс Streamlit
+st.title("🎙️ Простой диктофон")
+st.write("Этот диктофон позволяет записывать аудио и сохранять его на вашем устройстве.")
 
-# Ввод продолжительности записи
-record_seconds = st.number_input("Введите продолжительность записи (в секундах):", min_value=1, value=5)
+# Ползунок для выбора длины записи
+duration = st.slider("Длительность записи (секунды)", min_value=1, max_value=300, value=10)
 
 # Кнопка для начала записи
 if st.button("Начать запись"):
-    # Открытие потока для записи
-    stream = audio.open(format=FORMAT, channels=CHANNELS,
-                        rate=RATE, input=True,
-                        frames_per_buffer=CHUNK)
-    st.write("Запись...")
-    frames = []
-
-    # Запись аудио
-    for i in range(0, int(RATE / CHUNK * record_seconds)):
-        data = stream.read(CHUNK)
-        frames.append(data)
-
-    st.write("Запись завершена.")
-
-    # Остановка и закрытие потока
-    stream.stop_stream()
-    stream.close()
-    audio.terminate()
-
-    # Сохранение записанного аудио в файл
-    with tempfile.NamedTemporaryFile(delete=False, suffix=".wav") as temp_audio:
-        with wave.open(temp_audio.name, 'wb') as wf:
-            wf.setnchannels(CHANNELS)
-            wf.setsampwidth(audio.get_sample_size(FORMAT))
-            wf.setframerate(RATE)
-            wf.writeframes(b''.join(frames))
-
-        st.write("Аудио сохранено.")
-
-        # Загрузка аудиофайла
-        with open(temp_audio.name, "rb") as f:
-            audio_bytes = f.read()
-            st.audio(audio_bytes, format='audio/wav')
-
-        # Сохранение файла на сервер
-        file_name = st.text_input("Введите имя файла для сохранения:", value="recorded_audio.wav")
-        if st.button("Сохранить файл"):
-            with open(file_name, "wb") as f:
-                f.write(audio_bytes)
-            st.write(f"Файл сохранен как {file_name}")
-
-        # Удаление временного файла
-        os.remove(temp_audio.name)
+    audio_data = record_audio(duration)
+    
+    # Кнопка для сохранения записи
+    st.audio(np.array(audio_data), format="audio/wav")
+    if st.button("Сохранить запись"):
+        save_audio("recorded_audio.wav", audio_data)