Update app.py

app.py

@@ -138,7 +138,8 @@ def handle_submission():
             st.warning("The agent's response is None. Please try again.")
         elif isinstance(response, Image.Image):
             st.image(response)
-        elif "audio" in response:
+#        elif hasattr(response, 'audio'):
+        elif "audio" in str(response):
             audio_data = base64.b64decode(response.split(",")[1])
             audio = AudioSegment.from_file(io.BytesIO(audio_data))
             st.audio(audio)
@@ -146,6 +147,8 @@ def handle_submission():
             st.audio(response)
         elif isinstance(response, str):
             st.markdown(response)
+        elif isinstance(response, int):
+            st.markdown(response)
         elif "text" in response:
             st.markdown(response)
         else: