Update pages.py

pages.py

@@ -235,13 +235,18 @@ def audio_llm():
     if "messages" not in st.session_state:
         st.session_state.messages = []
     
-    if prompt := st.chat_input():
+    if prompt := st.chat_input(placeholder="Your Instruction"):
         with st.chat_message("user"):
             st.write(prompt)
         st.session_state.messages.append({"role": "user", "content": prompt})
     
         with st.chat_message("assistant"):
             with st.spinner("Thinking..."):
-                stream = generate_response(prompt, st.session_state.audio_base64)
-                response = st.write_stream(stream)
+                try:
+                    stream = generate_response(prompt, st.session_state.audio_base64)
+                    response = st.write_stream(stream)
+                except Exception as e:
+                    response = f"Caught Exception: {repr(e)}. Please contact the administrator to restart this space."
+                    st.write(response)
+                    raise(e)
                 st.session_state.messages.append({"role": "assistant", "content": response})