UI improvements

app.py

@@ -224,6 +224,7 @@ def reset_state():
         st.session_state.option_2 = ""
         st.session_state.option_1_model_name_state = ""
         st.session_state.option_2_model_name_state = ""
+        st.session_state.has_audio = False
 
 def on_option_1_click():
     if st.session_state.transcribed and not st.session_state.option_selected:
@@ -300,7 +301,7 @@ def on_option_none_click():
 
 def on_click_transcribe():
     if st.session_state.has_audio:
-        with st.spinner("💾 Saving and loading results... please wait"):
+        with st.spinner("Transcribing audio... this may take up to 30 seconds"):
             option_1_text, option_2_text = transcribe_audio(
                     )
             st.session_state.option_1 = option_1_text
@@ -312,7 +313,7 @@ def on_click_transcribe():
 
 def on_random_click():
     reset_state()
-    with st.spinner("💾 Saving and loading results... please wait"):
+    with st.spinner("Fetching random audio... please wait"):
         fetch_audio_payload = {"task": "fetch_audio"}
         array, sampling_rate, filepath = send_task(fetch_audio_payload)
         st.session_state.audio = {"data":array,"sample_rate":sampling_rate,"format":"audio/wav"}
@@ -321,6 +322,9 @@ def on_random_click():
         st.session_state.audio_path = filepath
         st.session_state.option_selected = None
 
+def on_reset_click():
+    reset_state()
+
 writer_thread = threading.Thread(target=result_writer_thread)
 writer_thread.start()
 
@@ -350,6 +354,8 @@ def main():
         st.session_state.option_2_model_name_state = ""
     if "user_email" not in st.session_state:
         st.session_state.user_email = ""
+    if "recording" not in st.session_state:
+        st.session_state.recording = True
 
     col1, col2 = st.columns([1, 1])
 
@@ -357,7 +363,7 @@ def main():
         st.markdown("### Record Audio")
         with st.container():
             audio_bytes = audio_recorder(
-                text="🎙️ Click to Record",
+                text="Click microphone to start/stop recording",
                 pause_threshold=3,
                 icon_size="2x",
                 key="audio_recorder",
@@ -375,18 +381,20 @@ def main():
             st.session_state.audio_path = tmp_file.name
             st.session_state.option_selected = None
             st.toast("Audio recorded successfully",icon="🎤")
+            st.session_state.recording = False
 
     with col2:
         st.markdown("### Random Audio Example")
         with st.container():
-            st.button("🎲 Random Audio",on_click=on_random_click,key="random_btn")
+            st.button("🎲 Select Random Audio",on_click=on_random_click,key="random_btn")
+            st.session_state.recording = False
 
     if st.session_state.has_audio:
         st.audio(**st.session_state.audio)
 
 
     with st.container():
-        st.button("📝 Transcribe Audio",on_click=on_click_transcribe,use_container_width=True,key="transcribe_btn")
+        st.button("📝 Transcribe Audio",on_click=on_click_transcribe,use_container_width=True,key="transcribe_btn",disabled=st.session_state.recording)
 
     text_containers = st.columns([1, 1])
     name_containers = st.columns([1, 1])
@@ -419,6 +427,9 @@ def main():
     with c4:
         st.button("Prefer None",on_click=on_option_none_click,key="none_btn")
 
+    with st.container():
+        st.button("New Match",on_click=on_reset_click,key="reset_btn",use_container_width=True)
+
     INSTR = """
     ## Instructions:
     * Record audio to recognise speech (or press 🎲 for random Audio).