Add: Recording option for testing CoNeTTE.

.gitignore

@@ -0,0 +1 @@
+record.wav

app.py

@@ -6,6 +6,8 @@ from typing import Any
 
 import streamlit as st
 
+from audiorecorder import audiorecorder
+
 from conette import CoNeTTEModel, conette
 
 
@@ -25,10 +27,22 @@ def main() -> None:
 
     st.warning("Recommanded audio: lasting from **1 to 30s**, sampled at **32 kHz**.")
     audios = st.file_uploader(
-        "Upload audio files here:",
+        "**Upload audio files here:**",
         type=["wav", "flac", "mp3", "ogg", "avi"],
         accept_multiple_files=True,
     )
+    st.write("**OR**")
+    record = audiorecorder(
+        start_prompt="Start recording", stop_prompt="Stop recording", pause_prompt=""
+    )
+
+    record_fpath = "record.wav"
+    if len(record) > 0:
+        record.export(record_fpath, format="wav")
+        st.write(
+            f"Record frame rate: {record.frame_rate}Hz, record duration: {record.duration_seconds:.2f}s"
+        )
+        st.audio(record.export().read())  # type: ignore
 
     with st.expander("Model hyperparameters"):
         task = st.selectbox("Task embedding input", model.tasks, 0)
@@ -110,6 +124,14 @@ def main() -> None:
         for audio_fname, cand in zip(audio_fnames, cands):
             st.success(f"**Output for {audio_fname}:**\n- {format_cand(cand)}")
 
+    if len(record) > 0:
+        outputs = model(
+            record_fpath,
+            **kwargs,
+        )
+        cand = outputs["cands"][0]
+        st.success(f"**Output for {'test'}:**\n- {format_cand(cand)}")
+
 
 if __name__ == "__main__":
     main()

requirements.txt

@@ -1 +1,2 @@
-conette>=0.1.3
+conette>=0.1.3
+streamlit-audiorecorder