Update app.py

app.py

@@ -1,30 +1,65 @@
 import streamlit as st
-from pydub import AudioSegment
-from pydub.utils import make_chunks
-import os
-
-st.title("Простое приложение диктофон")
-
-# Функции для записи и воспроизведения аудио
-def record_audio():
-    # Используем ffmpeg для записи аудио
-    filename = "recorded_audio.wav"
-    os.system(f"ffmpeg -f alsa -i default -t 10 -y {filename}")
-    return filename
-
-def play_audio(filename):
-    audio = AudioSegment.from_wav(filename)
-    st.audio(audio.raw_data, sample_rate=audio.frame_rate)
-
-# Кнопки для записи и воспроизведения
-if st.button("Записать аудио"):
-    filename = record_audio()
-    st.success("Аудио записано!")
-    with open(filename, "rb") as f:
-        st.download_button("Скачать аудио", f, file_name=filename)
-
-if st.button("Воспроизвести аудио"):
-    if os.path.exists("recorded_audio.wav"):
-        play_audio("recorded_audio.wav")
-    else:
-        st.warning("Сначала нужно записать аудио!")
+import numpy as np
+import soundfile as sf
+import tempfile
+
+# Функция для сохранения аудио
+def save_audio_file(audio_bytes):
+    with tempfile.NamedTemporaryFile(delete=False, suffix='.wav') as tmp_file:
+        tmp_file.write(audio_bytes)
+        return tmp_file.name
+
+# HTML и JavaScript для записи аудио
+html_code = """
+<script>
+  let mediaRecorder;
+  let audioChunks = [];
+
+  function startRecording() {
+    navigator.mediaDevices.getUserMedia({ audio: true }).then(stream => {
+      mediaRecorder = new MediaRecorder(stream);
+      mediaRecorder.start();
+
+      mediaRecorder.ondataavailable = event => {
+        audioChunks.push(event.data);
+      };
+
+      mediaRecorder.onstop = () => {
+        const audioBlob = new Blob(audioChunks);
+        const reader = new FileReader();
+        reader.onloadend = function() {
+          const audioData = reader.result.split(',')[1]; // base64
+          const audioBytes = new Uint8Array(atob(audioData).split('').map(c => c.charCodeAt(0)));
+          const xhr = new XMLHttpRequest();
+          xhr.open("POST", "/upload", true);
+          xhr.setRequestHeader("Content-Type", "application/octet-stream");
+          xhr.send(audioBlob);
+          audioChunks = [];
+        };
+        reader.readAsDataURL(audioBlob);
+      };
+    });
+  }
+
+  function stopRecording() {
+    mediaRecorder.stop();
+  }
+</script>
+
+<button onclick="startRecording()">Start Recording</button>
+<button onclick="stopRecording()">Stop Recording</button>
+"""
+
+# Streamlit интерфейс
+st.title("Audio Recorder")
+
+# Встроенный HTML для записи
+st.markdown(html_code, unsafe_allow_html=True)
+
+# Обработка загрузки аудио
+if st.button("Upload Audio"):
+    audio_data = st.file_uploader("Upload your audio file", type=["wav", "mp3"])
+    if audio_data is not None:
+        # Сохранение аудио файла
+        audio_file_path = save_audio_file(audio_data.read())
+        st.success(f"Audio saved at {audio_file_path}")