Update utils.py

utils.py

@@ -1,5 +1,6 @@
-import os
 import io
+import os
+import re
 
 import librosa
 import paramiko
@@ -10,6 +11,10 @@ from sshtunnel import SSHTunnelForwarder
 local_port = int(os.getenv('LOCAL_PORT'))
 
 
+class NoAudioException(Exception):
+    pass
+
+
 @st.cache_resource()
 def start_server():
     pkey = paramiko.RSAKey.from_private_key(io.StringIO(os.getenv('PRIVATE_KEY')))
@@ -41,7 +46,17 @@ def load_model():
     return client, model_name
 
 
-def generate_response(text_input):              
+def generate_response(text_input):
+    if not st.session_state.audio_base64:
+        raise NoAudioException("audio is empty.")
+
+    warnings = []
+    if re.search("tool|code|python|java|math|calculate", text_input):
+        warnings.append("WARNING: MERaLiON-AudioLLM is not intended for use in tool calling, math, and coding tasks.")
+
+    if re.search(r'[\u4e00-\u9fff]+', text_input):
+        warnings.append("NOTE: Please try to prompt in English for the best performance.")        
+        
     stream = st.session_state.client.chat.completions.create(
         messages=[{
             "role":
@@ -66,7 +81,7 @@ def generate_response(text_input):
         stream=True,
     )
 
-    return stream
+    return stream, warnings
 
 
 def bytes_to_array(audio_bytes):