Update app.py

app.py

@@ -1,43 +1,66 @@
 import streamlit as st
-import sounddevice as sd
-from scipy.io.wavfile import write
-import time
+import pyaudio
+import wave
+import os
+import tempfile
 
-# Настройки диктофона
-SAMPLE_RATE = 44100  # Частота дискретизации
-CHANNELS = 2  # Количество каналов (стерео)
+# Настройки записи
+FORMAT = pyaudio.paInt16
+CHANNELS = 1
+RATE = 44100
+CHUNK = 1024
+
+# Инициализация PyAudio
+audio = pyaudio.PyAudio()
+
+# Создание интерфейса Streamlit
+st.title("Простой диктофон")
+
+# Ввод продолжительности записи
+record_seconds = st.number_input("Введите продолжительность записи (в секундах):", min_value=1, value=5)
+
+# Кнопка для начала записи
+if st.button("Начать запись"):
+    # Открытие потока для записи
+    stream = audio.open(format=FORMAT, channels=CHANNELS,
+                        rate=RATE, input=True,
+                        frames_per_buffer=CHUNK)
+    st.write("Запись...")
+    frames = []
 
-# Функция для записи аудио
-def record_audio(filename):
     # Запись аудио
-    recording = sd.rec(frames=None, samplerate=SAMPLE_RATE, channels=CHANNELS)
-    st.write("Идет запись...")
-    
-    # Ожидание окончания записи
-    while True:
-        time.sleep(0.1)
-        if st.button("Остановить запись", key=f"stop_{filename}"):
-            break
-    
-    sd.stop()  # Остановка записи
+    for i in range(0, int(RATE / CHUNK * record_seconds)):
+        data = stream.read(CHUNK)
+        frames.append(data)
+
     st.write("Запись завершена.")
-    
-    # Сохранение записи в файл
-    write(filename, SAMPLE_RATE, recording)
-    st.write(f"Файл '{filename}' сохранен.")
 
-# Пользовательский интерфейс Streamlit
-st.title("Диктофон")
+    # Остановка и закрытие потока
+    stream.stop_stream()
+    stream.close()
+    audio.terminate()
 
-if st.button("Начать запись"):
-    # Генерация уникального имени файла
-    filename = f"recording_{int(time.time())}.wav"
-    record_audio(filename)
+    # Сохранение записанного аудио в файл
+    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')
 
-    # Воспроизведение записи
-    st.audio(filename)
+        # Сохранение файла на сервер
+        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}")
 
-# Отображение списка сохраненных записей
-st.subheader("Сохраненные записи:")
-for filename in st.session_state.get("recordings", []):
-    st.audio(filename)
+        # Удаление временного файла
+        os.remove(temp_audio.name)