Update app.py

app.py

@@ -2,33 +2,51 @@ import streamlit as st
 import sounddevice as sd
 import numpy as np
 import wave
+import io
 from scipy.io.wavfile import write
 
 # Функция для записи аудио
 def record_audio(duration, samplerate=44100):
-    st.write("🎙️ Запись...")
+    st.write("🎙️ Идёт запись...")
     audio_data = sd.rec(int(duration * samplerate), samplerate=samplerate, channels=2, dtype='int16')
-    sd.wait()  # Ожидание завершения записи
+    sd.wait()  # Ждём завершения записи
     st.write("✅ Запись завершена!")
     return audio_data
 
 # Функция для сохранения аудио в WAV файл
-def save_audio(filename, data, samplerate=44100):
-    write(filename, samplerate, data)
-    st.success(f"Аудио сохранено как {filename}")
+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
 
 # Интерфейс Streamlit
 st.title("🎙️ Простой диктофон")
 st.write("Этот диктофон позволяет записывать аудио и сохранять его на вашем устройстве.")
 
 # Ползунок для выбора длины записи
-duration = st.slider("Длительность записи (секунды)", min_value=1, max_value=300, value=10)
+duration = st.slider("Длительность записи (секунды)", min_value=1, max_value=60, value=5)
+
+# Переменная для хранения записи
+if "audio_data" not in st.session_state:
+    st.session_state.audio_data = None
 
 # Кнопка для начала записи
 if st.button("Начать запись"):
     audio_data = record_audio(duration)
-    
-    # Кнопка для сохранения записи
-    st.audio(np.array(audio_data), format="audio/wav")
-    if st.button("Сохранить запись"):
-        save_audio("recorded_audio.wav", audio_data)
+    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"
+    )