add new examples

src/content/common.py

@@ -69,6 +69,42 @@ MODEL_NAMES = OrderedDict({})
 
 
 AUDIO_SAMPLES_W_INSTRUCT = {
+    "female_pilot#1": {
+        "apperance": "Female Pilot Interview: Transcription",
+        "instructions": [
+            "Please transcribe the speech"
+        ]
+    },
+    "female_pilot#2": {
+        "apperance": "Female Pilot Interview: Aircraft name",
+        "instructions": [
+            "What does 大力士 mean in the conversation"
+        ]
+    },
+    "female_pilot#3": {
+        "apperance": "Female Pilot Interview: Air Force Personnel Count",
+        "instructions": [
+            "How many air force personnel are there?"
+        ]
+    },
+    "female_pilot#4": {
+        "apperance": "Female Pilot Interview: Air Force Personnel Name",
+        "instructions": [
+            "Can you tell me the names of the two pilots?"
+        ]
+    },
+    "female_pilot#5": {
+        "apperance": "Female Pilot Interview: Pilot Seat Restriction",
+        "instructions": [
+            "What is the concern of having a big butt for pilot?"
+        ]
+    },
+    "female_pilot#6": {
+        "apperance": "Female Pilot Interview: Conversation Mood",
+        "instructions": [
+            "What is the mood of the conversation?"
+        ]
+    },
     "7_ASR_IMDA_PART3_30_ASR_v2_2269": {
         "apperance": "7. Automatic Speech Recognition task: conversation in Singapore accent",
         "instructions": [
@@ -415,8 +451,8 @@ def header_section(component_name, description="", concise_description="", icon=
 def sidebar_fragment():
     with st.container(height=256, border=False):
         st.page_link("pages/playground.py", disabled=st.session_state.disprompt, label="🚀 Playground")
-        st.page_link("pages/agent.py", disabled=st.session_state.disprompt, label="👥 Chatbot")
-        st.page_link("pages/voice_chat.py", disabled=st.session_state.disprompt, label="🗣️ Voice Chat")
+        st.page_link("pages/agent.py", disabled=st.session_state.disprompt, label="👥 Cascade System")
+        st.page_link("pages/voice_chat.py", disabled=st.session_state.disprompt, label="🗣️ End-to-End Voice Chat")
     
     st.divider()
 
@@ -446,7 +482,8 @@ def successful_example_section(audio_sample_names, audio_array_state, audio_base
         key='select')
     
     if sample_name and st.session_state.on_select:
-        audio_bytes = open(f"audio_samples/{sample_name}.wav", "rb").read()
+        file_name = sample_name.split("#")[0]
+        audio_bytes = open(f"audio_samples/{file_name}.wav", "rb").read()
         st.session_state.update(
             on_select=False,
             new_prompt=AUDIO_SAMPLES_W_INSTRUCT[sample_name]["instructions"][0]

src/content/playground.py

@@ -24,7 +24,7 @@ QUICK_ACTIONS = [
     },
     {
         "name": "**Transcript**",
-        "instruction": "Please transcribe this speech.",
+        "instruction": "Please transcribe the speech",
         "width": 9.5,
     }
 ]

src/tunnel.py

@@ -6,6 +6,10 @@ import streamlit as st
 from sshtunnel import SSHTunnelForwarder
 
 
+DEFAULT_LOCAL_PORTS = "8000 8001"
+DEFAULT_REMOTE_PORTS = "8000 8001"
+
+
 @st.cache_resource()
 def start_server():
     server = SSHTunnelManager()
@@ -23,11 +27,11 @@ class SSHTunnelManager:
             ssh_pkey=pkey,  
             local_bind_addresses=[
                 ("127.0.0.1", int(port)) 
-                for port in os.getenv('LOCAL_PORTS').split(" ")
+                for port in os.getenv('LOCAL_PORTS', DEFAULT_LOCAL_PORTS).split(" ")
             ],
             remote_bind_addresses=[
-                ("127.0.0.1", 8000),
-                ("127.0.0.1", 8001),
+                ("127.0.0.1", int(port)) 
+                for port in os.getenv('REMOTE_PORTS', DEFAULT_REMOTE_PORTS).split(" ")
             ]
         )