File size: 577 Bytes
9cc018b
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
import torch
import torchaudio
import soundfile as sf
from pathlib import Path

def save_uploaded_audio(uploaded_file, save_path="user_audio.wav"):
    with open(save_path, "wb") as f:
        f.write(uploaded_file.read())
    return save_path

def record_audio(file_path="recorded.wav", duration=5, sample_rate=16000):
    import sounddevice as sd
    from scipy.io.wavfile import write

    st.info("Recording...")
    audio = sd.rec(int(duration * sample_rate), samplerate=sample_rate, channels=1)
    sd.wait()
    write(file_path, sample_rate, audio)
    return file_path