Update app.py

app.py

@@ -105,11 +105,12 @@ if prompt := st.chat_input("Message..."):
                 model=selected_model,
                 messages=st.session_state.messages
             )
-            if hasattr(response, "message") and hasattr(response.message, "content") and hasattr(response.message.content, "text"):
-                reply = response.message.content.text
+            if hasattr(response, "message") and hasattr(response.message, "content"):
+                content_items = response.message.content
+                reply = "".join(getattr(item, 'text', '') for item in content_items)
             else:
                 st.write(response)
-                reply = "❗️Couldn't find response.message.content.text in the Cohere response."
+                reply = "❗️Couldn't extract reply from the Cohere response."
             st.markdown(reply)
 
         st.session_state.messages.append({"role": "assistant", "content": reply})